net.nordist.lloydproof.CorrectionStorage.java Source code

Introduction

Here is the source code for net.nordist.lloydproof.CorrectionStorage.java

Source

// Copyright  2012 by Brent J. Nordquist. Some Rights Reserved.
// This work is licensed under http://creativecommons.org/licenses/by-sa/3.0/

package net.nordist.lloydproof;

import android.content.ContentValues;
import android.content.Context;
import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;
import android.database.sqlite.SQLiteOpenHelper;
import android.util.Log;
import java.util.Iterator;
import java.util.List;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

public class CorrectionStorage extends SQLiteOpenHelper {
    private static final String TAG = "CorrectionStorage";
    private static final String TABLE_NAME = "corrections";
    private static final String DATABASE_NAME = "corrections.db";
    private static final int DATABASE_VERSION = 1;

    private SQLiteDatabase writeDB;
    private SQLiteDatabase readDB;

    public CorrectionStorage(Context context) {
        super(context, DATABASE_NAME, null, DATABASE_VERSION);
    }

    @Override
    public void onCreate(SQLiteDatabase database) {
        Log.w(TAG, "Creating database v" + DATABASE_VERSION);
        database.execSQL("CREATE TABLE " + TABLE_NAME + " (id integer PRIMARY KEY AUTOINCREMENT,"
                + " current_text text NOT NULL DEFAULT '');");
    }

    @Override
    public void onUpgrade(SQLiteDatabase database, int oldVersion, int newVersion) {
        Log.w(TAG, "Upgrading database from v" + oldVersion + " to v" + newVersion
                + ", which will destroy all old data");
        database.execSQL("DROP TABLE IF EXISTS " + TABLE_NAME + ";");
        onCreate(database);
    }

    public int save(String currentText) {
        ContentValues values = new ContentValues();
        values.put("current_text", currentText);
        openWriteDB();
        int savedId = (int) writeDB.insert(TABLE_NAME, null, values);
        if (savedId > 0) {
            Log.d(TAG, "saved correction as id=" + savedId);
        } else {
            Log.e(TAG, "error saving correction");
        }
        return savedId;
    }

    public int count() {
        openReadDB();
        Cursor cursor = readDB.query(TABLE_NAME, new String[] { "COUNT(1)" }, null, null, null, null, null);
        if (cursor.getCount() > 0) {
            cursor.moveToFirst();
            return cursor.getInt(0);
        } else {
            return 0;
        }
    }

    public void deleteAll() {
        openWriteDB();
        writeDB.delete(TABLE_NAME, null, null);
        Log.d(TAG, "deleted all rows");
    }

    public int deleteById(int id) { // NOPMD - id makes sense
        openWriteDB();
        String idString = Integer.toString(id);
        int deletedCount = writeDB.delete(TABLE_NAME, "id=?", new String[] { idString });
        if (deletedCount > 0) {
            Log.d(TAG, "deleted correction with id=" + idString);
        } else {
            Log.e(TAG, "error deleting correction");
        }
        return deletedCount;
    }

    public int deleteByIdList(List<Integer> idList) {
        openWriteDB();
        int deletedCount = 0;
        Iterator<Integer> i = idList.iterator();
        while (i.hasNext()) {
            deletedCount += deleteById(i.next().intValue());
        }
        return deletedCount;
    }

    private JSONObject getNextAsJSONObject(Cursor cursor) throws JSONException {
        JSONObject jobject = new JSONObject();
        jobject.put("sync_id", cursor.getInt(0));
        jobject.put("current_text", cursor.getString(1));
        return jobject;
    }

    public JSONArray getAllAsJSONArray() throws JSONException {
        openReadDB();
        Cursor cursor = readDB.query(TABLE_NAME, new String[] { "id", "current_text" }, null, null, null, null,
                null);
        JSONArray jarray = new JSONArray();
        while (cursor.moveToNext()) {
            jarray.put(getNextAsJSONObject(cursor));
        }
        return jarray;
    }

    public void close() {
        if (writeDB != null) {
            writeDB.close();
            writeDB = null;
        }
        if (readDB != null) {
            readDB.close();
            readDB = null;
        }
    }

    protected void finalize() throws Throwable {
        try {
            close();
        } finally {
            super.finalize();
        }
    }

    private void openWriteDB() {
        if (writeDB == null) {
            writeDB = getWritableDatabase();
        }
    }

    private void openReadDB() {
        if (readDB == null) {
            readDB = getReadableDatabase();
        }
    }
}