greendao3.0数据库升级

国外大神子作用于升级
package com.example.green.greendao.upData;

import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;
import android.support.annotation.NonNull;
import android.text.TextUtils;
import android.util.Log;

import org.greenrobot.greendao.AbstractDao;
import org.greenrobot.greendao.database.Database;
import org.greenrobot.greendao.database.StandardDatabase;
import org.greenrobot.greendao.internal.DaoConfig;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class MigrationHelper {
	public static boolean DEBUG = true;
	private static String TAG = "MigrationHelper2";
	private static final String SQLITE_MASTER = "sqlite_master";
	private static final String SQLITE_TEMP_MASTER = "sqlite_temp_master";

	public static void migrate(SQLiteDatabase db, Class<? extends AbstractDao<?, ?>>... daoClasses) {
		printLog("【The Old Database Version】" + db.getVersion());
		Database database = new StandardDatabase(db);
		migrate(database, daoClasses);
	}

	//TODO 迁移数据库
	public static void migrate(Database database, Class<? extends AbstractDao<?, ?>>... daoClasses) {
		printLog("【Generate temp table】start");
		generateTempTables(database, daoClasses);
		printLog("【Generate temp table】complete");

		dropAllTables(database, true, daoClasses);
		createAllTables(database, false, daoClasses);

		printLog("【Restore data】start");
		restoreData(database, daoClasses);
		printLog("【Restore data】complete");
	}

	//TODO 生成临时表
	private static void generateTempTables(Database db, Class<? extends AbstractDao<?, ?>>... daoClasses) {
		for (int i = 0; i < daoClasses.length; i++) {
			String tempTableName = null;

			DaoConfig daoConfig = new DaoConfig(db, daoClasses[i]);
			String tableName = daoConfig.tablename;
			if (!isTableExists(db, false, tableName)) {
				printLog("【New Table】" + tableName);
				continue;
			}
			tempTableName = daoConfig.tablename.concat("_TEMP");
			StringBuilder dropTableStringBuilder = new StringBuilder();
			dropTableStringBuilder.append("DROP TABLE IF EXISTS ").append(tempTableName).append(";");
			db.execSQL(dropTableStringBuilder.toString());

			StringBuilder insertTableStringBuilder = new StringBuilder();
			insertTableStringBuilder.append("CREATE TEMPORARY TABLE ").append(tempTableName);
			insertTableStringBuilder.append(" AS SELECT * FROM ").append(tableName).append(";");
			db.execSQL(insertTableStringBuilder.toString());
			printLog("【Table】" + tableName + "\n ---Columns-->" + getColumnsStr(daoConfig));
			printLog("【Generate temp table】" + tempTableName);
		}
	}

	//TODO 当前表是否存在
	private static boolean isTableExists(Database db, boolean isTemp, String tableName) {
		if (db == null || TextUtils.isEmpty(tableName)) {
			return false;
		}
		String dbName = isTemp ? SQLITE_TEMP_MASTER : SQLITE_MASTER;
		String sql = "SELECT COUNT(*) FROM " + dbName + " WHERE type = ? AND name = ?";
		Cursor cursor = null;
		int count = 0;
		try {
			cursor = db.rawQuery(sql, new String[]{"table", tableName});
			if (cursor == null || !cursor.moveToFirst()) {
				return false;
			}
			count = cursor.getInt(0);
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (cursor != null)
				cursor.close();
		}
		return count > 0;
	}


	//TODO 列数量
	private static String getColumnsStr(DaoConfig daoConfig) {
		if (daoConfig == null) {
			return "no columns";
		}
		StringBuilder builder = new StringBuilder();
		for (int i = 0; i < daoConfig.allColumns.length; i++) {
			builder.append(daoConfig.allColumns[i]);
			builder.append(",");
		}
		if (builder.length() > 0) {
			builder.deleteCharAt(builder.length() - 1);
		}
		return builder.toString();
	}


	//TODO 删除所有表
	private static void dropAllTables(Database db, boolean ifExists, @NonNull Class<? extends AbstractDao<?, ?>>... daoClasses) {
		reflectMethod(db, "dropTable", ifExists, daoClasses);
		printLog("【Drop all table】");
	}

	//TODO 创建所有表
	private static void createAllTables(Database db, boolean ifNotExists, @NonNull Class<? extends AbstractDao<?, ?>>... daoClasses) {
		reflectMethod(db, "createTable", ifNotExists, daoClasses);
		printLog("【Create all table】");
	}

	/**
	 * dao class already define the sql exec method, so just invoke it
	 */
	private static void reflectMethod(Database db, String methodName, boolean isExists, @NonNull Class<? extends AbstractDao<?, ?>>... daoClasses) {
		if (daoClasses.length < 1) {
			return;
		}
		try {
			for (Class cls : daoClasses) {
				Method method = cls.getDeclaredMethod(methodName, Database.class, boolean.class);
				method.invoke(null, db, isExists);
			}
		} catch (NoSuchMethodException e) {
			e.printStackTrace();
		} catch (InvocationTargetException e) {
			e.printStackTrace();
		} catch (IllegalAccessException e) {
			e.printStackTrace();
		}
	}

	//TODO 恢复数据
	private static void restoreData(Database db, Class<? extends AbstractDao<?, ?>>... daoClasses) {
		for (int i = 0; i < daoClasses.length; i++) {
			DaoConfig daoConfig = new DaoConfig(db, daoClasses[i]);
			String tableName = daoConfig.tablename;
			String tempTableName = daoConfig.tablename.concat("_TEMP");

			if (!isTableExists(db, true, tempTableName)) {
				continue;
			}

			// get all columns from tempTable, take careful to use the columns list
			List<String> columns = getColumns(db, tempTableName);
			ArrayList<String> properties = new ArrayList<>(columns.size());
			for (int j = 0; j < daoConfig.properties.length; j++) {
				String columnName = daoConfig.properties[j].columnName;
				if (columns.contains(columnName)) {
					properties.add(columnName);
				}
			}
			if (properties.size() > 0) {
				final String columnSQL = TextUtils.join(",", properties);

				StringBuilder insertTableStringBuilder = new StringBuilder();
				insertTableStringBuilder.append("INSERT INTO ").append(tableName).append(" (");
				insertTableStringBuilder.append(columnSQL);
				insertTableStringBuilder.append(") SELECT ");
				insertTableStringBuilder.append(columnSQL);
				insertTableStringBuilder.append(" FROM ").append(tempTableName).append(";");
				db.execSQL(insertTableStringBuilder.toString());
				printLog("【Restore data】 to " + tableName);
			}
			StringBuilder dropTableStringBuilder = new StringBuilder();
			dropTableStringBuilder.append("DROP TABLE ").append(tempTableName);
			db.execSQL(dropTableStringBuilder.toString());
			printLog("【Drop temp table】" + tempTableName);
		}
	}

	private static List<String> getColumns(Database db, String tableName) {
		List<String> columns = null;
		Cursor cursor = null;
		try {
			cursor = db.rawQuery("SELECT * FROM " + tableName + " limit 0", null);
			if (null != cursor && cursor.getColumnCount() > 0) {
				columns = Arrays.asList(cursor.getColumnNames());
			}
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (cursor != null)
				cursor.close();
			if (null == columns)
				columns = new ArrayList<>();
		}
		return columns;
	}

	private static void printLog(String info) {
		if (DEBUG) {
			Log.d(TAG, info);
		}
	}

}

继承与自己生成的DaoMaster类内部类DaoMaster.DevOpenHelper

package com.example.green.greendao.upData;

import android.content.Context;
import android.database.sqlite.SQLiteDatabase;
import android.util.Log;

import com.example.green.greendao.dao.DaoMaster;
import com.example.green.greendao.dao.UserInfoDao;

import org.greenrobot.greendao.database.Database;

/**
 * Created by wbxu on 2017/6/19.
 * 自定义 MySQLiteOpenHelper集成 DaoMaster.OpenHelper 重写更新数据库的方法
 * 当app下的build.gradle 的schemaVersion数据库的版本号改变时,,创建数据库会调用onUpgrade更细数据库的方法
 */

public class MySQLiteOpenHelper extends DaoMaster.DevOpenHelper {
	/**
	 * @param context 上下文
	 * @param name    原来定义的数据库的名字   新旧数据库一致
	 * @param factory 可以null
	 */
	public MySQLiteOpenHelper(Context context, String name, SQLiteDatabase.CursorFactory factory) {
		super(context, name, factory);
	}

	/**
	 * @param db
	 * @param oldVersion
	 * @param newVersion 更新数据库的时候自己调用
	 */
	@Override
	public void onUpgrade(Database db, int oldVersion, int newVersion) {
		Log.d("flag111", "-----调用了");
		Log.i("flag111", "Upgrading schema from version " + oldVersion + " to " + newVersion + " by dropping all tables");

		//具体的数据转移在MigrationHelper2类中
		/**
		 *  将db传入     将gen目录下的所有的Dao.类传入
		 */
		if (newVersion > oldVersion) {
			MigrationHelper.migrate(db, UserInfoDao.class);
		}
	}
}

初始化数据库时MySQLiteOpenHelper是刚自己创建的类

	/**
	 *      * 设置greenDao
	 *     
	 */
	private void setDatabase() {
		// 通过 DaoMaster 的内部类 DevOpenHelper,你可以得到一个便利的 SQLiteOpenHelper 对象。
		// 可能你已经注意到了,你并不需要去编写「CREATE TABLE」这样的 SQL 语句,因为 greenDAO 已经帮你做了。
		// 注意:默认的 DaoMaster.DevOpenHelper 会在数据库升级时,删除所有的表,意味着这将导致数据的丢失。
		// 所以,在正式的项目中,你还应该做一层封装,来实现数据库的安全升级。
		// 此处sport-db表示数据库名称 可以任意填写
		mHelper = new MySQLiteOpenHelper(this, "test_greendao", null);
		db = mHelper.getWritableDatabase();
		// 注意:该数据库连接属于 DaoMaster,所以多个 Session 指的是相同的数据库连接。
		mDaoMaster = new DaoMaster(db);
		mDaoSession = mDaoMaster.newSession();
	}

	public DaoSession getDaoSession() {
		return mDaoSession;
	}

	public SQLiteDatabase getDb() {
		return db;
	}

升级时更改版本号自动调用MySQLiteOpenHelper的onUpgrade方法

greendao {
    schemaVersion 1
    daoPackage 'com.example.green.greendao.dao'
    targetGenDir 'src/main/java'
}

混淆配置

greendao

-keep class org.greenrobot.greendao.**{*;}
-keep class com.example.green.greendao.upData.**{*;}
-keep public interface org.greenrobot.greendao.**
-keepclassmembers class * extends org.greenrobot.greendao.AbstractDao {
public static java.lang.String TABLENAME;
}
-keep class **$Properties
-keep class net.sqlcipher.database.**{*;}
-keep public interface net.sqlcipher.database.**
-dontwarn net.sqlcipher.database.**
-dontwarn org.greenrobot.greendao.**

升级反射配置

-keep public class * extends android.database.sqlite.SQLiteOpenHelper{*;}
-keep public class * extends android.app.Application{*;}
-keep public class * extends org.greenrobot.greendao.AbstractDao{*;}

 

 

 

 

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值