Ammar Githam
4 years ago
3 changed files with 235 additions and 254 deletions
-
3app/build.gradle
-
478app/src/main/java/awais/instagrabber/db/AppDatabase.kt
-
6app/src/main/java/awais/instagrabber/db/Converters.kt
@ -1,270 +1,252 @@ |
|||
package awais.instagrabber.db; |
|||
|
|||
import android.content.ContentValues; |
|||
import android.content.Context; |
|||
import android.database.Cursor; |
|||
import android.database.sqlite.SQLiteDatabase; |
|||
import android.util.Log; |
|||
import android.util.Pair; |
|||
|
|||
import androidx.annotation.NonNull; |
|||
import androidx.room.Database; |
|||
import androidx.room.Room; |
|||
import androidx.room.RoomDatabase; |
|||
import androidx.room.TypeConverters; |
|||
import androidx.room.migration.Migration; |
|||
import androidx.sqlite.db.SupportSQLiteDatabase; |
|||
|
|||
import java.time.Instant; |
|||
import java.time.LocalDateTime; |
|||
import java.time.ZoneId; |
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
|
|||
import awais.instagrabber.db.dao.AccountDao; |
|||
import awais.instagrabber.db.dao.DMLastNotifiedDao; |
|||
import awais.instagrabber.db.dao.FavoriteDao; |
|||
import awais.instagrabber.db.dao.RecentSearchDao; |
|||
import awais.instagrabber.db.entities.Account; |
|||
import awais.instagrabber.db.entities.DMLastNotified; |
|||
import awais.instagrabber.db.entities.Favorite; |
|||
import awais.instagrabber.db.entities.RecentSearch; |
|||
import awais.instagrabber.models.enums.FavoriteType; |
|||
import awais.instagrabber.utils.Utils; |
|||
|
|||
@Database(entities = {Account.class, Favorite.class, DMLastNotified.class, RecentSearch.class}, |
|||
version = 6) |
|||
@TypeConverters({Converters.class}) |
|||
public abstract class AppDatabase extends RoomDatabase { |
|||
private static final String TAG = AppDatabase.class.getSimpleName(); |
|||
|
|||
private static AppDatabase INSTANCE; |
|||
|
|||
public abstract AccountDao accountDao(); |
|||
|
|||
public abstract FavoriteDao favoriteDao(); |
|||
|
|||
public abstract DMLastNotifiedDao dmLastNotifiedDao(); |
|||
|
|||
public abstract RecentSearchDao recentSearchDao(); |
|||
|
|||
public static AppDatabase getDatabase(final Context context) { |
|||
if (INSTANCE == null) { |
|||
synchronized (AppDatabase.class) { |
|||
if (INSTANCE == null) { |
|||
INSTANCE = Room.databaseBuilder(context.getApplicationContext(), AppDatabase.class, "cookiebox.db") |
|||
.addMigrations(MIGRATION_1_2, MIGRATION_2_3, MIGRATION_3_4, MIGRATION_4_5, MIGRATION_5_6) |
|||
.build(); |
|||
package awais.instagrabber.db |
|||
|
|||
import android.content.ContentValues |
|||
import android.content.Context |
|||
import android.database.sqlite.SQLiteDatabase |
|||
import android.util.Log |
|||
import androidx.room.Database |
|||
import androidx.room.Room |
|||
import androidx.room.RoomDatabase |
|||
import androidx.room.TypeConverters |
|||
import androidx.room.migration.Migration |
|||
import androidx.sqlite.db.SupportSQLiteDatabase |
|||
import awais.instagrabber.db.dao.AccountDao |
|||
import awais.instagrabber.db.dao.DMLastNotifiedDao |
|||
import awais.instagrabber.db.dao.FavoriteDao |
|||
import awais.instagrabber.db.dao.RecentSearchDao |
|||
import awais.instagrabber.db.entities.Account |
|||
import awais.instagrabber.db.entities.DMLastNotified |
|||
import awais.instagrabber.db.entities.Favorite |
|||
import awais.instagrabber.db.entities.RecentSearch |
|||
import awais.instagrabber.utils.Utils |
|||
import awais.instagrabber.utils.extensions.TAG |
|||
import java.time.Instant |
|||
import java.time.LocalDateTime |
|||
import java.time.ZoneId |
|||
import java.util.* |
|||
|
|||
@Database(entities = [Account::class, Favorite::class, DMLastNotified::class, RecentSearch::class], version = 6) |
|||
@TypeConverters(Converters::class) |
|||
abstract class AppDatabase : RoomDatabase() { |
|||
abstract fun accountDao(): AccountDao |
|||
abstract fun favoriteDao(): FavoriteDao |
|||
abstract fun dmLastNotifiedDao(): DMLastNotifiedDao |
|||
abstract fun recentSearchDao(): RecentSearchDao |
|||
|
|||
companion object { |
|||
private lateinit var INSTANCE: AppDatabase |
|||
|
|||
fun getDatabase(context: Context): AppDatabase { |
|||
if (!this::INSTANCE.isInitialized) { |
|||
synchronized(AppDatabase::class.java) { |
|||
if (!this::INSTANCE.isInitialized) { |
|||
INSTANCE = Room.databaseBuilder(context.applicationContext, AppDatabase::class.java, "cookiebox.db") |
|||
.addMigrations(MIGRATION_1_2, MIGRATION_2_3, MIGRATION_3_4, MIGRATION_4_5, MIGRATION_5_6) |
|||
.build() |
|||
} |
|||
} |
|||
} |
|||
return INSTANCE |
|||
} |
|||
return INSTANCE; |
|||
} |
|||
|
|||
static final Migration MIGRATION_1_2 = new Migration(1, 2) { |
|||
@Override |
|||
public void migrate(@NonNull SupportSQLiteDatabase db) { |
|||
db.execSQL("ALTER TABLE cookies ADD " + Account.COL_FULL_NAME + " TEXT"); |
|||
db.execSQL("ALTER TABLE cookies ADD " + Account.COL_PROFILE_PIC + " TEXT"); |
|||
private val MIGRATION_1_2: Migration = object : Migration(1, 2) { |
|||
override fun migrate(db: SupportSQLiteDatabase) { |
|||
db.execSQL("ALTER TABLE cookies ADD " + Account.COL_FULL_NAME + " TEXT") |
|||
db.execSQL("ALTER TABLE cookies ADD " + Account.COL_PROFILE_PIC + " TEXT") |
|||
} |
|||
} |
|||
}; |
|||
|
|||
static final Migration MIGRATION_2_3 = new Migration(2, 3) { |
|||
@Override |
|||
public void migrate(@NonNull SupportSQLiteDatabase db) { |
|||
final List<Favorite> oldFavorites = backupOldFavorites(db); |
|||
// recreate with new columns (as there will be no doubt about the `query_display` column being present or not in the future versions) |
|||
db.execSQL("DROP TABLE " + Favorite.TABLE_NAME); |
|||
db.execSQL("CREATE TABLE " + Favorite.TABLE_NAME + " (" |
|||
+ Favorite.COL_ID + " INTEGER PRIMARY KEY," |
|||
+ Favorite.COL_QUERY + " TEXT," |
|||
+ Favorite.COL_TYPE + " TEXT," |
|||
+ Favorite.COL_DISPLAY_NAME + " TEXT," |
|||
+ Favorite.COL_PIC_URL + " TEXT," |
|||
+ Favorite.COL_DATE_ADDED + " INTEGER)"); |
|||
// add the old favorites back |
|||
for (final Favorite oldFavorite : oldFavorites) { |
|||
insertOrUpdateFavorite(db, oldFavorite); |
|||
private val MIGRATION_2_3: Migration = object : Migration(2, 3) { |
|||
override fun migrate(db: SupportSQLiteDatabase) { |
|||
val oldFavorites = backupOldFavorites(db) |
|||
// recreate with new columns (as there will be no doubt about the `query_display` column being present or not in the future versions) |
|||
db.execSQL("DROP TABLE " + Favorite.TABLE_NAME) |
|||
db.execSQL("CREATE TABLE " + Favorite.TABLE_NAME + " (" |
|||
+ Favorite.COL_ID + " INTEGER PRIMARY KEY," |
|||
+ Favorite.COL_QUERY + " TEXT," |
|||
+ Favorite.COL_TYPE + " TEXT," |
|||
+ Favorite.COL_DISPLAY_NAME + " TEXT," |
|||
+ Favorite.COL_PIC_URL + " TEXT," |
|||
+ Favorite.COL_DATE_ADDED + " INTEGER)") |
|||
// add the old favorites back |
|||
for (oldFavorite in oldFavorites) { |
|||
insertOrUpdateFavorite(db, oldFavorite) |
|||
} |
|||
} |
|||
} |
|||
}; |
|||
|
|||
static final Migration MIGRATION_3_4 = new Migration(3, 4) { |
|||
@Override |
|||
public void migrate(@NonNull SupportSQLiteDatabase db) { |
|||
// Required when migrating to Room. |
|||
// The original table primary keys were not 'NOT NULL', so the migration to Room were failing without the below migration. |
|||
// Taking this opportunity to rename cookies table to accounts |
|||
|
|||
// Create new table with name 'accounts' |
|||
db.execSQL("CREATE TABLE " + Account.TABLE_NAME + " (" |
|||
+ Account.COL_ID + " INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT," |
|||
+ Account.COL_UID + " TEXT," |
|||
+ Account.COL_USERNAME + " TEXT," |
|||
+ Account.COL_COOKIE + " TEXT," |
|||
+ Account.COL_FULL_NAME + " TEXT," |
|||
+ Account.COL_PROFILE_PIC + " TEXT)"); |
|||
// Insert all data from table 'cookies' to 'accounts' |
|||
db.execSQL("INSERT INTO " + Account.TABLE_NAME + " (" |
|||
+ Account.COL_UID + "," |
|||
+ Account.COL_USERNAME + "," |
|||
+ Account.COL_COOKIE + "," |
|||
+ Account.COL_FULL_NAME + "," |
|||
+ Account.COL_PROFILE_PIC + ") " |
|||
+ "SELECT " |
|||
+ Account.COL_UID + "," |
|||
+ Account.COL_USERNAME + "," |
|||
+ Account.COL_COOKIE + "," |
|||
+ Account.COL_FULL_NAME + "," |
|||
+ Account.COL_PROFILE_PIC |
|||
+ " FROM cookies"); |
|||
// Drop old cookies table |
|||
db.execSQL("DROP TABLE cookies"); |
|||
|
|||
// Create favorite backup table |
|||
db.execSQL("CREATE TABLE " + Favorite.TABLE_NAME + "_backup (" |
|||
+ Favorite.COL_ID + " INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT," |
|||
+ Favorite.COL_QUERY + " TEXT," |
|||
+ Favorite.COL_TYPE + " TEXT," |
|||
+ Favorite.COL_DISPLAY_NAME + " TEXT," |
|||
+ Favorite.COL_PIC_URL + " TEXT," |
|||
+ Favorite.COL_DATE_ADDED + " INTEGER)"); |
|||
// Insert all data from table 'favorite' to 'favorite_backup' |
|||
db.execSQL("INSERT INTO " + Favorite.TABLE_NAME + "_backup (" |
|||
+ Favorite.COL_QUERY + "," |
|||
+ Favorite.COL_TYPE + "," |
|||
+ Favorite.COL_DISPLAY_NAME + "," |
|||
+ Favorite.COL_PIC_URL + "," |
|||
+ Favorite.COL_DATE_ADDED + ") " |
|||
+ "SELECT " |
|||
+ Favorite.COL_QUERY + "," |
|||
+ Favorite.COL_TYPE + "," |
|||
+ Favorite.COL_DISPLAY_NAME + "," |
|||
+ Favorite.COL_PIC_URL + "," |
|||
+ Favorite.COL_DATE_ADDED |
|||
+ " FROM " + Favorite.TABLE_NAME); |
|||
// Drop favorites |
|||
db.execSQL("DROP TABLE " + Favorite.TABLE_NAME); |
|||
// Rename favorite_backup to favorites |
|||
db.execSQL("ALTER TABLE " + Favorite.TABLE_NAME + "_backup RENAME TO " + Favorite.TABLE_NAME); |
|||
private val MIGRATION_3_4: Migration = object : Migration(3, 4) { |
|||
override fun migrate(db: SupportSQLiteDatabase) { |
|||
// Required when migrating to Room. |
|||
// The original table primary keys were not 'NOT NULL', so the migration to Room were failing without the below migration. |
|||
// Taking this opportunity to rename cookies table to accounts |
|||
|
|||
// Create new table with name 'accounts' |
|||
db.execSQL("CREATE TABLE " + Account.TABLE_NAME + " (" |
|||
+ Account.COL_ID + " INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT," |
|||
+ Account.COL_UID + " TEXT," |
|||
+ Account.COL_USERNAME + " TEXT," |
|||
+ Account.COL_COOKIE + " TEXT," |
|||
+ Account.COL_FULL_NAME + " TEXT," |
|||
+ Account.COL_PROFILE_PIC + " TEXT)") |
|||
// Insert all data from table 'cookies' to 'accounts' |
|||
db.execSQL("INSERT INTO " + Account.TABLE_NAME + " (" |
|||
+ Account.COL_UID + "," |
|||
+ Account.COL_USERNAME + "," |
|||
+ Account.COL_COOKIE + "," |
|||
+ Account.COL_FULL_NAME + "," |
|||
+ Account.COL_PROFILE_PIC + ") " |
|||
+ "SELECT " |
|||
+ Account.COL_UID + "," |
|||
+ Account.COL_USERNAME + "," |
|||
+ Account.COL_COOKIE + "," |
|||
+ Account.COL_FULL_NAME + "," |
|||
+ Account.COL_PROFILE_PIC |
|||
+ " FROM cookies") |
|||
// Drop old cookies table |
|||
db.execSQL("DROP TABLE cookies") |
|||
|
|||
// Create favorite backup table |
|||
db.execSQL("CREATE TABLE " + Favorite.TABLE_NAME + "_backup (" |
|||
+ Favorite.COL_ID + " INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT," |
|||
+ Favorite.COL_QUERY + " TEXT," |
|||
+ Favorite.COL_TYPE + " TEXT," |
|||
+ Favorite.COL_DISPLAY_NAME + " TEXT," |
|||
+ Favorite.COL_PIC_URL + " TEXT," |
|||
+ Favorite.COL_DATE_ADDED + " INTEGER)") |
|||
// Insert all data from table 'favorite' to 'favorite_backup' |
|||
db.execSQL("INSERT INTO " + Favorite.TABLE_NAME + "_backup (" |
|||
+ Favorite.COL_QUERY + "," |
|||
+ Favorite.COL_TYPE + "," |
|||
+ Favorite.COL_DISPLAY_NAME + "," |
|||
+ Favorite.COL_PIC_URL + "," |
|||
+ Favorite.COL_DATE_ADDED + ") " |
|||
+ "SELECT " |
|||
+ Favorite.COL_QUERY + "," |
|||
+ Favorite.COL_TYPE + "," |
|||
+ Favorite.COL_DISPLAY_NAME + "," |
|||
+ Favorite.COL_PIC_URL + "," |
|||
+ Favorite.COL_DATE_ADDED |
|||
+ " FROM " + Favorite.TABLE_NAME) |
|||
// Drop favorites |
|||
db.execSQL("DROP TABLE " + Favorite.TABLE_NAME) |
|||
// Rename favorite_backup to favorites |
|||
db.execSQL("ALTER TABLE " + Favorite.TABLE_NAME + "_backup RENAME TO " + Favorite.TABLE_NAME) |
|||
} |
|||
} |
|||
}; |
|||
|
|||
static final Migration MIGRATION_4_5 = new Migration(4, 5) { |
|||
@Override |
|||
public void migrate(@NonNull final SupportSQLiteDatabase database) { |
|||
database.execSQL("CREATE TABLE IF NOT EXISTS `dm_last_notified` (" + |
|||
"`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, " + |
|||
"`thread_id` TEXT, " + |
|||
"`last_notified_msg_ts` INTEGER, " + |
|||
"`last_notified_at` INTEGER)"); |
|||
database.execSQL("CREATE UNIQUE INDEX IF NOT EXISTS `index_dm_last_notified_thread_id` ON `dm_last_notified` (`thread_id`)"); |
|||
private val MIGRATION_4_5: Migration = object : Migration(4, 5) { |
|||
override fun migrate(database: SupportSQLiteDatabase) { |
|||
database.execSQL("CREATE TABLE IF NOT EXISTS `dm_last_notified` (" + |
|||
"`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, " + |
|||
"`thread_id` TEXT, " + |
|||
"`last_notified_msg_ts` INTEGER, " + |
|||
"`last_notified_at` INTEGER)") |
|||
database.execSQL("CREATE UNIQUE INDEX IF NOT EXISTS `index_dm_last_notified_thread_id` ON `dm_last_notified` (`thread_id`)") |
|||
} |
|||
} |
|||
}; |
|||
|
|||
static final Migration MIGRATION_5_6 = new Migration(5, 6) { |
|||
@Override |
|||
public void migrate(@NonNull final SupportSQLiteDatabase database) { |
|||
database.execSQL("CREATE TABLE IF NOT EXISTS `recent_searches` (" + |
|||
"`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, " + |
|||
"`ig_id` TEXT NOT NULL, " + |
|||
"`name` TEXT NOT NULL, " + |
|||
"`username` TEXT, " + |
|||
"`pic_url` TEXT, " + |
|||
"`type` TEXT NOT NULL, " + |
|||
"`last_searched_on` INTEGER NOT NULL)"); |
|||
database.execSQL("CREATE UNIQUE INDEX IF NOT EXISTS `index_recent_searches_ig_id_type` ON `recent_searches` (`ig_id`, `type`)"); |
|||
private val MIGRATION_5_6: Migration = object : Migration(5, 6) { |
|||
override fun migrate(database: SupportSQLiteDatabase) { |
|||
database.execSQL("CREATE TABLE IF NOT EXISTS `recent_searches` (" + |
|||
"`id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, " + |
|||
"`ig_id` TEXT NOT NULL, " + |
|||
"`name` TEXT NOT NULL, " + |
|||
"`username` TEXT, " + |
|||
"`pic_url` TEXT, " + |
|||
"`type` TEXT NOT NULL, " + |
|||
"`last_searched_on` INTEGER NOT NULL)") |
|||
database.execSQL("CREATE UNIQUE INDEX IF NOT EXISTS `index_recent_searches_ig_id_type` ON `recent_searches` (`ig_id`, `type`)") |
|||
} |
|||
} |
|||
}; |
|||
|
|||
@NonNull |
|||
private static List<Favorite> backupOldFavorites(@NonNull final SupportSQLiteDatabase db) { |
|||
// check if old favorites table had the column query_display |
|||
final boolean queryDisplayExists = checkColumnExists(db, Favorite.TABLE_NAME, "query_display"); |
|||
Log.d(TAG, "backupOldFavorites: queryDisplayExists: " + queryDisplayExists); |
|||
final List<Favorite> oldModels = new ArrayList<>(); |
|||
final String sql = "SELECT " |
|||
+ "query_text," |
|||
+ "date_added" |
|||
+ (queryDisplayExists ? ",query_display" : "") |
|||
+ " FROM " + Favorite.TABLE_NAME; |
|||
try (final Cursor cursor = db.query(sql)) { |
|||
if (cursor != null && cursor.moveToFirst()) { |
|||
do { |
|||
try { |
|||
final String queryText = cursor.getString(cursor.getColumnIndex("query_text")); |
|||
final Pair<FavoriteType, String> favoriteTypeQueryPair = Utils.migrateOldFavQuery(queryText); |
|||
if (favoriteTypeQueryPair == null) continue; |
|||
final FavoriteType type = favoriteTypeQueryPair.first; |
|||
final String query = favoriteTypeQueryPair.second; |
|||
final long epochMillis = cursor.getLong(cursor.getColumnIndex("date_added")); |
|||
final LocalDateTime localDateTime = LocalDateTime.ofInstant( |
|||
Instant.ofEpochMilli(epochMillis), |
|||
ZoneId.systemDefault() |
|||
); |
|||
oldModels.add(new Favorite( |
|||
0, |
|||
query, |
|||
type, |
|||
queryDisplayExists ? cursor.getString(cursor.getColumnIndex("query_display")) : null, |
|||
null, |
|||
localDateTime |
|||
)); |
|||
} catch (Exception e) { |
|||
Log.e(TAG, "onUpgrade", e); |
|||
private fun backupOldFavorites(db: SupportSQLiteDatabase): List<Favorite> { |
|||
// check if old favorites table had the column query_display |
|||
val queryDisplayExists = checkColumnExists(db, Favorite.TABLE_NAME, "query_display") |
|||
Log.d(TAG, "backupOldFavorites: queryDisplayExists: $queryDisplayExists") |
|||
val oldModels: MutableList<Favorite> = ArrayList() |
|||
val sql = ("SELECT " |
|||
+ "query_text," |
|||
+ "date_added" |
|||
+ (if (queryDisplayExists) ",query_display" else "") |
|||
+ " FROM " + Favorite.TABLE_NAME) |
|||
try { |
|||
db.query(sql).use { cursor -> |
|||
if (cursor != null && cursor.moveToFirst()) { |
|||
do { |
|||
try { |
|||
val queryText = cursor.getString(cursor.getColumnIndex("query_text")) |
|||
val favoriteTypeQueryPair = Utils.migrateOldFavQuery(queryText) ?: continue |
|||
val type = favoriteTypeQueryPair.first |
|||
val query = favoriteTypeQueryPair.second |
|||
val epochMillis = cursor.getLong(cursor.getColumnIndex("date_added")) |
|||
val localDateTime = LocalDateTime.ofInstant( |
|||
Instant.ofEpochMilli(epochMillis), |
|||
ZoneId.systemDefault() |
|||
) |
|||
oldModels.add(Favorite( |
|||
0, |
|||
query, |
|||
type, |
|||
if (queryDisplayExists) cursor.getString(cursor.getColumnIndex("query_display")) else null, |
|||
null, |
|||
localDateTime |
|||
)) |
|||
} catch (e: Exception) { |
|||
Log.e(TAG, "onUpgrade", e) |
|||
} |
|||
} while (cursor.moveToNext()) |
|||
} |
|||
} while (cursor.moveToNext()); |
|||
} |
|||
} catch (e: Exception) { |
|||
Log.e(TAG, "onUpgrade", e) |
|||
} |
|||
} catch (Exception e) { |
|||
Log.e(TAG, "onUpgrade", e); |
|||
Log.d(TAG, "backupOldFavorites: oldModels:$oldModels") |
|||
return oldModels |
|||
} |
|||
Log.d(TAG, "backupOldFavorites: oldModels:" + oldModels); |
|||
return oldModels; |
|||
} |
|||
|
|||
private static synchronized void insertOrUpdateFavorite(@NonNull final SupportSQLiteDatabase db, @NonNull final Favorite model) { |
|||
final ContentValues values = new ContentValues(); |
|||
values.put(Favorite.COL_QUERY, model.getQuery()); |
|||
values.put(Favorite.COL_TYPE, model.getType().toString()); |
|||
values.put(Favorite.COL_DISPLAY_NAME, model.getDisplayName()); |
|||
values.put(Favorite.COL_PIC_URL, model.getPicUrl()); |
|||
values.put(Favorite.COL_DATE_ADDED, model.getDateAdded().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli()); |
|||
int rows; |
|||
if (model.getId() >= 1) { |
|||
rows = db.update(Favorite.TABLE_NAME, |
|||
SQLiteDatabase.CONFLICT_IGNORE, |
|||
values, |
|||
Favorite.COL_ID + "=?", |
|||
new String[]{String.valueOf(model.getId())}); |
|||
} else { |
|||
rows = db.update(Favorite.TABLE_NAME, |
|||
SQLiteDatabase.CONFLICT_IGNORE, |
|||
values, |
|||
Favorite.COL_QUERY + "=?" + " AND " + Favorite.COL_TYPE + "=?", |
|||
new String[]{model.getQuery(), model.getType().toString()}); |
|||
} |
|||
if (rows != 1) { |
|||
db.insert(Favorite.TABLE_NAME, SQLiteDatabase.CONFLICT_IGNORE, values); |
|||
@Synchronized |
|||
private fun insertOrUpdateFavorite(db: SupportSQLiteDatabase, model: Favorite) { |
|||
val values = ContentValues() |
|||
values.put(Favorite.COL_QUERY, model.query) |
|||
values.put(Favorite.COL_TYPE, model.type.toString()) |
|||
values.put(Favorite.COL_DISPLAY_NAME, model.displayName) |
|||
values.put(Favorite.COL_PIC_URL, model.picUrl) |
|||
values.put(Favorite.COL_DATE_ADDED, model.dateAdded!!.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli()) |
|||
val rows: Int = if (model.id >= 1) { |
|||
db.update(Favorite.TABLE_NAME, |
|||
SQLiteDatabase.CONFLICT_IGNORE, |
|||
values, |
|||
Favorite.COL_ID + "=?", arrayOf(model.id.toString())) |
|||
} else { |
|||
db.update(Favorite.TABLE_NAME, |
|||
SQLiteDatabase.CONFLICT_IGNORE, |
|||
values, |
|||
Favorite.COL_QUERY + "=?" + " AND " + Favorite.COL_TYPE + "=?", arrayOf(model.query, model.type.toString())) |
|||
} |
|||
if (rows != 1) { |
|||
db.insert(Favorite.TABLE_NAME, SQLiteDatabase.CONFLICT_IGNORE, values) |
|||
} |
|||
} |
|||
} |
|||
|
|||
private static boolean checkColumnExists(@NonNull final SupportSQLiteDatabase db, |
|||
@NonNull final String tableName, |
|||
@NonNull final String columnName) { |
|||
boolean exists = false; |
|||
try (Cursor cursor = db.query("PRAGMA table_info(" + tableName + ")")) { |
|||
if (cursor.moveToFirst()) { |
|||
do { |
|||
final String currentColumn = cursor.getString(cursor.getColumnIndex("name")); |
|||
if (currentColumn.equals(columnName)) { |
|||
exists = true; |
|||
@Suppress("SameParameterValue") |
|||
private fun checkColumnExists( |
|||
db: SupportSQLiteDatabase, |
|||
tableName: String, |
|||
columnName: String, |
|||
): Boolean { |
|||
var exists = false |
|||
try { |
|||
db.query("PRAGMA table_info($tableName)").use { cursor -> |
|||
if (cursor.moveToFirst()) { |
|||
do { |
|||
val currentColumn = cursor.getString(cursor.getColumnIndex("name")) |
|||
if (currentColumn == columnName) { |
|||
exists = true |
|||
} |
|||
} while (cursor.moveToNext()) |
|||
} |
|||
} while (cursor.moveToNext()); |
|||
|
|||
} |
|||
} catch (ex: Exception) { |
|||
Log.e(TAG, "checkColumnExists", ex) |
|||
} |
|||
} catch (Exception ex) { |
|||
Log.e(TAG, "checkColumnExists", ex); |
|||
return exists |
|||
} |
|||
return exists; |
|||
} |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue