新增数据库操作工具类，加入获取连接、事务相关方法
This commit is contained in:
zengfantian
parent
f35d1b9b19
commit
c5543527f8
|
|
@ -0,0 +1,240 @@
|
|||
package com.zftlive.android.tools.db;
|
||||
|
||||
import java.sql.SQLException;
|
||||
import java.sql.Savepoint;
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import android.content.Context;
|
||||
import android.database.sqlite.SQLiteDatabase;
|
||||
import android.util.Log;
|
||||
|
||||
import com.j256.ormlite.android.AndroidDatabaseConnection;
|
||||
import com.j256.ormlite.android.apptools.OpenHelperManager;
|
||||
import com.j256.ormlite.android.apptools.OrmLiteSqliteOpenHelper;
|
||||
import com.j256.ormlite.dao.Dao;
|
||||
import com.j256.ormlite.support.ConnectionSource;
|
||||
import com.j256.ormlite.support.DatabaseConnection;
|
||||
import com.j256.ormlite.table.TableUtils;
|
||||
import com.zftlive.android.base.BaseEntity;
|
||||
|
||||
/**
|
||||
* 数据库访问帮助类
|
||||
* @author 曾繁添
|
||||
* @version 1.0
|
||||
*
|
||||
*/
|
||||
public class DatabaseHelper extends OrmLiteSqliteOpenHelper {
|
||||
|
||||
private static String databaseName = "default_database";
|
||||
private static int databaseVersion = 1;
|
||||
private static DatabaseHelper instance = null;
|
||||
private static String TAG = DatabaseHelper.class.getSimpleName();
|
||||
private Map<String, Dao> daoMap = new HashMap<String, Dao>();
|
||||
private static List<Class<? extends BaseEntity>> table = new ArrayList<Class<? extends BaseEntity>>();
|
||||
|
||||
/**
|
||||
* 对外实例化对象不采用该构造方法
|
||||
* @param context
|
||||
*/
|
||||
@Deprecated
|
||||
public DatabaseHelper(Context context) {
|
||||
super(context, databaseName, null, databaseVersion);
|
||||
}
|
||||
|
||||
/**
|
||||
* 实例化对象
|
||||
* @param dbName 数据库名称
|
||||
* @param version 数据库版本
|
||||
* @return
|
||||
*/
|
||||
public static synchronized DatabaseHelper gainInstance(Context context,String dbName, int version) {
|
||||
if (instance == null) {
|
||||
databaseName = dbName;
|
||||
databaseVersion = version;
|
||||
//会隐式调用public DatabaseHelper(Context context){}
|
||||
instance = OpenHelperManager.getHelper(context, DatabaseHelper.class);
|
||||
}
|
||||
return instance;
|
||||
}
|
||||
|
||||
/**
|
||||
* 注册表实体
|
||||
* @param entity 表实体
|
||||
*/
|
||||
public static void registerTables(Class<? extends BaseEntity> entity){
|
||||
table.add(entity);
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建表
|
||||
* @param entity 实体
|
||||
*/
|
||||
public void createTable(Class<? extends BaseEntity> entity) {
|
||||
try {
|
||||
TableUtils.createTableIfNotExists(getConnectionSource(), entity);
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG, "Unable to create table-->"+entity.getSimpleName(),e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 清除表数据
|
||||
* @param entity 实体
|
||||
*/
|
||||
public void clearTableData(Class<? extends BaseEntity> entity) {
|
||||
try {
|
||||
TableUtils.clearTable(getConnectionSource(), entity);
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG, "Unable to clear table data for -->"+entity.getSimpleName(),e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除表
|
||||
* @param entity 实体
|
||||
*/
|
||||
public void dropTable(Class<? extends BaseEntity> entity) {
|
||||
try {
|
||||
TableUtils.dropTable(getConnectionSource(), entity, true);
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG, "Unable to drop table-->"+entity.getSimpleName(),e);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 获取数据库连接
|
||||
* @return 数据库连接
|
||||
* @throws SQLException
|
||||
*/
|
||||
public DatabaseConnection getConnection() throws SQLException{
|
||||
DatabaseConnection connection = null;
|
||||
if(isOpen()){
|
||||
connection = getConnectionSource().getReadWriteConnection();
|
||||
}else{
|
||||
connection = new AndroidDatabaseConnection(getWritableDatabase(), true);
|
||||
}
|
||||
return connection;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 关闭数据库连接
|
||||
*/
|
||||
public void closeConnection(DatabaseConnection connection) {
|
||||
try {
|
||||
if (connection != null) {
|
||||
if(!connection.isClosed()){
|
||||
connection.close();
|
||||
}
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG, "关闭数据库失败，原因："+e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 提交事务
|
||||
* @param connection 数据库连接
|
||||
* @param savePoint 事务点
|
||||
*/
|
||||
public void commit(DatabaseConnection connection,Savepoint savePoint){
|
||||
try {
|
||||
connection.commit(savePoint);
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG, "提交事务失败，原因："+e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 回滚事务
|
||||
* @param connection 数据库连接
|
||||
* @param savePoint 事务点
|
||||
*/
|
||||
public void rollback(DatabaseConnection connection,Savepoint savePoint){
|
||||
try {
|
||||
connection.rollback(savePoint);
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG, "回滚事务失败，原因："+e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取Dao
|
||||
* @throws SQLException
|
||||
*/
|
||||
public synchronized <D extends Dao<T, ?>, T> D getDao(Class<T> clazz) throws SQLException {
|
||||
Dao dao = null;
|
||||
try {
|
||||
String keyClassName = clazz.getSimpleName();
|
||||
if (daoMap.containsKey(keyClassName)) {
|
||||
dao = daoMap.get(keyClassName);
|
||||
}
|
||||
if (dao == null) {
|
||||
dao = super.getDao(clazz);
|
||||
daoMap.put(keyClassName, dao);
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG, "Unable to getDao", e);
|
||||
throw e;
|
||||
}
|
||||
return (D) dao;
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放数据库连接
|
||||
*/
|
||||
public void releaseAll() {
|
||||
if (instance != null) {
|
||||
OpenHelperManager.releaseHelper();
|
||||
instance = null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放资源
|
||||
*/
|
||||
@Override
|
||||
public void close() {
|
||||
super.close();
|
||||
for (String key : daoMap.keySet()) {
|
||||
Dao dao = daoMap.get(key);
|
||||
dao = null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 创建数据库时执行，初始化表
|
||||
*/
|
||||
@Override
|
||||
public void onCreate(SQLiteDatabase database,ConnectionSource connectionSource) {
|
||||
try {
|
||||
for (Class<? extends BaseEntity> entity : table) {
|
||||
TableUtils.createTableIfNotExists(connectionSource, entity);
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG, "Unable to create datbases", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新数据库时执行， 历史数据会被清空，残留这个bug，待改进
|
||||
*/
|
||||
@Override
|
||||
public void onUpgrade(SQLiteDatabase database,ConnectionSource connectionSource, int oldVersion, int newVersion) {
|
||||
try {
|
||||
//先清表
|
||||
for (Class<? extends BaseEntity> entity : table) {
|
||||
TableUtils.dropTable(connectionSource, entity, true);
|
||||
}
|
||||
//重建表
|
||||
onCreate(database, connectionSource);
|
||||
} catch (SQLException e) {
|
||||
Log.e(TAG,"Unable to upgrade database from version " + oldVersion+ " to new " + newVersion, e);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue