SessionDB.java

package org.droidplanner.android.droneshare.data;

import android.content.ContentValues;
import android.content.Context;
import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;
import android.database.sqlite.SQLiteOpenHelper;
import android.net.Uri;
import android.support.annotation.Nullable;

import org.droidplanner.android.droneshare.data.SessionContract.SessionData;

import java.util.ArrayList;
import java.util.List;

import timber.log.Timber;

/**
 * Created by fhuya on 12/30/14.
 */
public class SessionDB extends SQLiteOpenHelper {

    public SessionDB(Context context) {
        super(context, SessionContract.DB_NAME, null, SessionContract.DB_VERSION);
    }

    @Override
    public void onCreate(SQLiteDatabase db) {
        Timber.i("Creating session database.");
        db.execSQL(SessionContract.getSqlCreateEntries());
    }

    @Override
    public void onUpgrade(SQLiteDatabase db, int oldVersion, int newVersion) {
        if (oldVersion == 1) {
            SessionContract.migrateFromV1(db);
        } else {
            Timber.w("Unrecognized database version %d for %s.", oldVersion, SessionContract.DB_NAME);
        }
    }

    /**
     * Return a unique id for the session that can be used to close it
     * @param startTimeInMillis
     * @param connectionType
     */
    public long startSession(long startTimeInMillis, String connectionType, @Nullable Uri tlogLoggingUri){
        //Get the data repository in write mode.
        SQLiteDatabase db = getWritableDatabase();

        ContentValues values = new ContentValues();
        values.put(SessionContract.SessionData.COLUMN_NAME_START_TIME, startTimeInMillis);
        values.put(SessionData.COLUMN_NAME_CONNECTION_TYPE, connectionType);
        values.put(SessionData.COLUMN_NAME_LABEL, SessionData.getSessionLabel(startTimeInMillis));
        if(tlogLoggingUri != null) {
            values.put(SessionData.COLUMN_NAME_TLOG_LOGGING_URI, tlogLoggingUri.toString());
        }

        return db.insert(SessionData.TABLE_NAME, null, values);
    }

    public void endSessions(long endTimeInMillis, long... rowIds){
        if(rowIds == null || rowIds.length == 0)
            return;

        SQLiteDatabase db = getWritableDatabase();

        ContentValues values = new ContentValues();
        values.put(SessionData.COLUMN_NAME_END_TIME, endTimeInMillis);

        boolean isFirst = true;
        StringBuilder selection = new StringBuilder();
        String[] selectionArgs = new String[rowIds.length];
        int argIndex = 0;
        for(long rowId : rowIds){
            if(!isFirst) {
                selection.append(" OR ");
            }
            else {
                isFirst = false;
            }
            selection.append(SessionData._ID).append(" LIKE ?");

            selectionArgs[argIndex++] = String.valueOf(rowId);
        }

        db.update(SessionData.TABLE_NAME, values, selection.toString(), selectionArgs);
    }

    public SessionData getSessionData(long sessionId){
        SQLiteDatabase db = getReadableDatabase();

        String[] projection = {SessionData.COLUMN_NAME_START_TIME,
            SessionData.COLUMN_NAME_END_TIME, SessionData.COLUMN_NAME_CONNECTION_TYPE,
            SessionData.COLUMN_NAME_TLOG_LOGGING_URI, SessionData.COLUMN_NAME_LABEL};

        String selection = SessionData._ID + " LIKE ?";
        String[] selectionArgs = {String.valueOf(sessionId)};

        Cursor cursor = db.query(SessionData.TABLE_NAME, projection, selection, selectionArgs, null,
            null, null);
        SessionData sessionData = null;
        if(cursor.moveToFirst()){
            long startTime = cursor.getLong(cursor.getColumnIndex(SessionData.COLUMN_NAME_START_TIME));
            long endTime = cursor.getLong(cursor.getColumnIndex(SessionData.COLUMN_NAME_END_TIME));
            String connectionTypeLabel = cursor.getString(cursor.getColumnIndex(SessionData.COLUMN_NAME_CONNECTION_TYPE));
            String tlogEncodedUri = cursor.getString(cursor.getColumnIndex(SessionData.COLUMN_NAME_TLOG_LOGGING_URI));
            Uri tlogLoggingUri = Uri.parse(tlogEncodedUri);
            String sessionLabel = cursor.getString(cursor.getColumnIndex(SessionData.COLUMN_NAME_LABEL));

            sessionData = new SessionData(sessionId, startTime, endTime, connectionTypeLabel, tlogLoggingUri, sessionLabel);
        }

        cursor.close();
        return sessionData;
    }

    public long[] getOpenedSessions(){
        SQLiteDatabase db = getReadableDatabase();

        String[] projection = {SessionData._ID};
        String selection = SessionData.COLUMN_NAME_END_TIME + "IS NULL";

        Cursor cursor = db.query(SessionData.TABLE_NAME, projection, selection, null, null, null, null);
        long[] sessionIds = new long[cursor.getCount()];
        int index = 0;
        for(boolean hasNext = cursor.moveToFirst(); hasNext; hasNext = cursor.moveToNext()){
            sessionIds[index++] = cursor.getLong(cursor.getColumnIndex(SessionData._ID));
        }

        cursor.close();
        return sessionIds;
    }

    public List<SessionData> getCompletedSessions(boolean tlogLogged){
        SQLiteDatabase db = getReadableDatabase();

        String[] projection = {SessionData._ID, SessionData.COLUMN_NAME_START_TIME,
            SessionData.COLUMN_NAME_END_TIME, SessionData.COLUMN_NAME_CONNECTION_TYPE,
            SessionData.COLUMN_NAME_TLOG_LOGGING_URI, SessionData.COLUMN_NAME_LABEL};

        String selection = SessionData.COLUMN_NAME_END_TIME + " IS NOT NULL";
        if(tlogLogged){
            selection += " AND " + SessionData.COLUMN_NAME_TLOG_LOGGING_URI + " IS NOT NULL";
        }
        String orderBy = SessionData.COLUMN_NAME_START_TIME + " ASC";

        Cursor cursor = db.query(SessionData.TABLE_NAME, projection, selection, null, null, null, orderBy);
        List<SessionData> sessionDataList = new ArrayList<>(cursor.getCount());
        for(boolean hasNext = cursor.moveToFirst(); hasNext; hasNext = cursor.moveToNext()){
            long id = cursor.getLong(cursor.getColumnIndex(SessionData._ID));
            long startTime = cursor.getLong(cursor.getColumnIndex(SessionData.COLUMN_NAME_START_TIME));
            long endTime = cursor.getLong(cursor.getColumnIndex(SessionData.COLUMN_NAME_END_TIME));
            String connectionTypeLabel = cursor.getString(cursor.getColumnIndex(SessionData.COLUMN_NAME_CONNECTION_TYPE));
            String tlogEncodedUri = cursor.getString(cursor.getColumnIndex(SessionData.COLUMN_NAME_TLOG_LOGGING_URI));
            Uri tlogLoggingUri = Uri.parse(tlogEncodedUri);
            String sessionLabel = cursor.getString(cursor.getColumnIndex(SessionData.COLUMN_NAME_LABEL));

            sessionDataList.add(new SessionData(id, startTime, endTime, connectionTypeLabel, tlogLoggingUri, sessionLabel));
        }

        cursor.close();
        return sessionDataList;
    }

    public void removeSessionData(long id) {
        SQLiteDatabase db = getWritableDatabase();

        String whereClause = SessionData._ID + " LIKE ?";
        String[] whereArgs = {String.valueOf(id)};

        db.delete(SessionData.TABLE_NAME, whereClause, whereArgs);
    }

    public void renameSession(long id, String label) {
        SQLiteDatabase db = getWritableDatabase();

        ContentValues values = new ContentValues();
        values.put(SessionData.COLUMN_NAME_LABEL, label);

        String whereClause = SessionData._ID + " LIKE ?";
        String[] whereArgs = {String.valueOf(id)};

        db.update(SessionData.TABLE_NAME, values, whereClause, whereArgs);
    }

    public void cleanupOpenedSessions(long endTimeInMillis){
        SQLiteDatabase db = getWritableDatabase();

        ContentValues values = new ContentValues();
        values.put(SessionData.COLUMN_NAME_END_TIME, endTimeInMillis);

        String selection = SessionData.COLUMN_NAME_END_TIME + " IS NULL";
        db.update(SessionData.TABLE_NAME, values, selection, null);
    }
}