mobi.cangol.mobile.db.DatabaseUtils Maven / Gradle / Ivy
/**
* Copyright (c) 2013 Cangol
*
* Licensed under the Apache License, Version 2.0 (the "License")
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package mobi.cangol.mobile.db;
import android.content.ContentValues;
import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;
import android.text.TextUtils;
import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Modifier;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import mobi.cangol.mobile.logging.Log;
public class DatabaseUtils {
private DatabaseUtils() {
}
/**
* 创建表索引
*
* @param db
* @param clazz
*/
public static void createIndex(SQLiteDatabase db, Class clazz, String indexName, String... fieldNames) {
if (clazz.isAnnotationPresent(DatabaseTable.class)) {
final DatabaseTable table = clazz.getAnnotation(DatabaseTable.class);
final String tableName = "".equals(table.value()) ? clazz.getSimpleName() : table.value();
final StringBuilder sql = new StringBuilder("CREATE INDEX ");
sql.append(indexName).append(" on ").append(tableName).append('(');
Field field = null;
String columnName = null;
for (int i = 0; i < fieldNames.length; i++) {
try {
field = clazz.getDeclaredField(fieldNames[i]);
field.setAccessible(true);
if (field.isEnumConstant() || Modifier.isFinal(field.getModifiers()) || Modifier.isTransient(field.getModifiers())) {
continue;
} else if (field.isAnnotationPresent(DatabaseField.class)) {
final DatabaseField dbField = field.getAnnotation(DatabaseField.class);
columnName = "".equals(dbField.value()) ? field.getName() : dbField.value();
sql.append(columnName);
if (i < fieldNames.length - 1)
sql.append(',');
}
} catch (NoSuchFieldException e) {
Log.e(e.getMessage());
}
}
sql.append(')');
db.execSQL(sql.toString());
} else {
throw new IllegalStateException(clazz + " not DatabaseTable Annotation");
}
}
/**
* 创建表
*
* @param db
* @param clazz
*/
public static void createTable(SQLiteDatabase db, Class clazz) {
if (clazz.isAnnotationPresent(DatabaseTable.class)) {
final StringBuilder sql = new StringBuilder("CREATE TABLE IF NOT EXISTS ");
final DatabaseTable table = clazz.getAnnotation(DatabaseTable.class);
final String tableName = "".equals(table.value()) ? clazz.getSimpleName() : table.value();
sql.append(tableName).append('(');
final Field[] fields = clazz.getDeclaredFields();
String filedName = null;
boolean isFirst = true;
for (final Field field : fields) {
field.setAccessible(true);
if (field.isEnumConstant() || Modifier.isFinal(field.getModifiers()) || Modifier.isTransient(field.getModifiers())) {
continue;
}
if (field.isAnnotationPresent(DatabaseField.class)) {
if (!isFirst) {
sql.append(',');
} else {
isFirst = false;
}
final DatabaseField dbField = field.getAnnotation(DatabaseField.class);
filedName = "".equals(dbField.value()) ? field.getName() : dbField.value();
sql.append(filedName);
sql.append(' ').append(getDbType(field.getType()));
if (dbField.primaryKey()) {
sql.append(" PRIMARY KEY AUTOINCREMENT");
}
if (dbField.unique()) {
sql.append(" UNIQUE");
}
if (dbField.notNull()) {
sql.append(" NOT NULL");
}
}
}
sql.append(')');
db.execSQL(sql.toString());
} else {
throw new IllegalStateException(clazz + " not DatabaseTable Annotation");
}
}
/**
* 获取sqlite对应的数据类型
*
* @param clazz
* @return
*/
public static String getDbType(Class clazz) {
if (clazz == String.class
|| clazz == Character.class || clazz == char.class
|| clazz == Boolean.class || clazz == boolean.class) {
return "TEXT";
} else if (clazz == Integer.class || clazz == int.class
|| clazz == Long.class || clazz == long.class
|| clazz == Short.class || clazz == short.class) {
return "INTEGER";
} else if (clazz == Double.class || clazz == double.class) {
return "Double";
} else if (clazz == Float.class || clazz == float.class) {
return "FLOAT";
} else {
return "BLOB";
}
}
/**
* 删除表
*
* @param db
* @param clazz
*/
public static void dropTable(SQLiteDatabase db, Class clazz) {
if (clazz.isAnnotationPresent(DatabaseTable.class)) {
final StringBuilder sql = new StringBuilder("DROP TABLE IF EXISTS ");
final DatabaseTable table = clazz.getAnnotation(DatabaseTable.class);
final String tableName = "".equals(table.value()) ? clazz.getSimpleName() : table.value();
sql.append(tableName);
db.execSQL(sql.toString());
} else {
throw new IllegalStateException(clazz + " not DatabaseTable Annotation");
}
}
/**
* 删除表
*
* @param db
* @param table
*/
public static void dropTable(SQLiteDatabase db, String table) {
if (table != null && !"".equals(table.trim())) {
final StringBuilder sql = new StringBuilder("DROP TABLE IF EXISTS ");
sql.append(table);
db.execSQL(sql.toString());
} else {
throw new IllegalStateException(table + " not DatabaseTable Annotation");
}
}
/**
* 表添加列
*
* @param db
* @param clazz
*/
public static void addColumn(SQLiteDatabase db, Class clazz, String... columns) {
Log.d("addColumn ");
if (clazz.isAnnotationPresent(DatabaseTable.class)) {
final DatabaseTable table = clazz.getAnnotation(DatabaseTable.class);
final String tableName = "".equals(table.value()) ? clazz.getSimpleName() : table.value();
final Map map = getColumnNames(clazz);
for (int i = 0; i < columns.length; i++) {
if (!TextUtils.isEmpty(columns[i]) && map.containsKey(columns[i])) {
final StringBuilder sql = new StringBuilder("ALTER TABLE ")
.append(tableName)
.append(" ADD COLUMN ")
.append(columns[i])
.append(" ")
.append(getDbType(map.get(columns[i]).getType()))
.append(";");
db.execSQL(sql.toString());
Log.d("" + sql.toString());
} else {
throw new IllegalStateException("column " + columns[i] + " is exist!");
}
}
} else {
throw new IllegalStateException(clazz + " not DatabaseTable Annotation");
}
}
/**
* 获取所有要数据库化的列名
*
* @param clazz
* @return
*/
public static Map getColumnNames(Class clazz) {
final Map map = new HashMap<>();
for (final Field field : clazz.getDeclaredFields()) {
field.setAccessible(true);
if (field.isAnnotationPresent(DatabaseField.class)) {
final DatabaseField dbField = field.getAnnotation(DatabaseField.class);
map.put("".equals(dbField.value()) ? field.getName() : dbField.value(), field);
}
}
return map;
}
public static String getIdColumnName(Class clazz) {
String columnName = null;
for (final Field field : clazz.getDeclaredFields()) {
field.setAccessible(true);
if (field.isEnumConstant() || Modifier.isFinal(field.getModifiers()) || Modifier.isTransient(field.getModifiers())) {
continue;
}
if (field.isAnnotationPresent(DatabaseField.class)) {
final DatabaseField dbField = field.getAnnotation(DatabaseField.class);
if (dbField.primaryKey()) {
columnName = "".equals(dbField.value()) ? field.getName() : dbField.value();
break;
}
}
}
return columnName;
}
/**
* 获取主键的值
*
* @param obj
* @return
* @throws IllegalAccessException
*/
public static Object getIdValue(Object obj) throws IllegalAccessException {
Object value = null;
for (final Field field : obj.getClass().getDeclaredFields()) {
field.setAccessible(true);
if (field.isEnumConstant() || Modifier.isFinal(field.getModifiers()) || Modifier.isTransient(field.getModifiers())) {
continue;
}
if (field.isAnnotationPresent(DatabaseField.class)) {
final DatabaseField dbField = field.getAnnotation(DatabaseField.class);
if (dbField.primaryKey()) {
value = field.get(obj);
break;
}
}
}
return value;
}
/**
* 获取键值对象
*
* @param object
* @return
* @throws IllegalAccessException
*/
public static ContentValues getContentValues(Object object) throws IllegalAccessException {
final ContentValues v = new ContentValues();
String filedName = null;
for (final Field field : object.getClass().getDeclaredFields()) {
field.setAccessible(true);
if (field.isAnnotationPresent(DatabaseField.class)) {
final DatabaseField dbField = field.getAnnotation(DatabaseField.class);
if (!dbField.primaryKey()) {
filedName = "".equals(dbField.value()) ? field.getName() : dbField.value();
v.put(filedName, String.valueOf(field.get(object)));
}
}
}
return v;
}
/**
* 获取键值对象
*
* @param object
* @param columns
* @return
* @throws IllegalAccessException
* @throws IllegalArgumentException
*/
public static ContentValues getContentValues(Object object, String[] columns) throws IllegalAccessException {
final ContentValues v = new ContentValues();
String filedName = null;
final Set set = (columns == null) ? new HashSet() : new HashSet<>(Arrays.asList(columns));
for (final Field field : object.getClass().getDeclaredFields()) {
field.setAccessible(true);
if (field.isAnnotationPresent(DatabaseField.class)) {
final DatabaseField dbField = field.getAnnotation(DatabaseField.class);
filedName = "".equals(dbField.value()) ? field.getName() : dbField.value();
if (!dbField.primaryKey() && (set.isEmpty() || set.contains(filedName))) {
v.put(filedName, String.valueOf(field.get(object)));
}
}
}
return v;
}
/**
* 查询记录的值，赋值给obj
*
* @param obj
* @param cursor
* @param
* @return
*/
public static T cursorToObject(T obj, Cursor cursor, String[] columns) {
final Field[] fields = obj.getClass().getDeclaredFields();
final Set set = (columns == null) ? new HashSet() : new HashSet<>(Arrays.asList(columns));
String columnName = null;
for (final Field field : fields) {
field.setAccessible(true);
if (field.isEnumConstant() || Modifier.isFinal(field.getModifiers()) || Modifier.isTransient(field.getModifiers())) {
continue;
}
if (field.isAnnotationPresent(DatabaseField.class)) {
final DatabaseField dbField = field.getAnnotation(DatabaseField.class);
columnName = "".equals(dbField.value()) ? field.getName() : dbField.value();
if (set.isEmpty() || set.contains(columnName))
setFieldValue(obj, field, columnName, cursor);
}
}
return obj;
}
/**
* 查询记录的值，赋值给clazz的实例obj
*
* @param clazz
* @param cursor
* @param
* @return
* @throws InstantiationException
* @throws IllegalAccessException
* @throws NoSuchMethodException
* @throws InvocationTargetException
*/
public static T cursorToClassObject(Class clazz, Cursor cursor, String[] columns) throws InstantiationException, IllegalAccessException, NoSuchMethodException, InvocationTargetException {
final Constructor constructor = clazz.getDeclaredConstructor();
constructor.setAccessible(true);
final T obj = (T) constructor.newInstance();
return cursorToObject(obj, cursor, columns);
}
/**
* field 赋值
*
* @param t
* @param field
* @param columnName
* @param cursor
* @param
*/
public static void setFieldValue(T t, Field field, String columnName, Cursor cursor) {
try {
if (field.getType() == String.class) {
field.set(t, cursor.getString(cursor.getColumnIndex(columnName)));
} else if (field.getType() == Integer.class || field.getType() == int.class) {
field.set(t, cursor.getInt(cursor.getColumnIndex(columnName)));
} else if (field.getType() == Long.class || field.getType() == long.class) {
field.set(t, cursor.getLong(cursor.getColumnIndex(columnName)));
} else if (field.getType() == Double.class || field.getType() == double.class) {
field.set(t, cursor.getDouble(cursor.getColumnIndex(columnName)));
} else if (field.getType() == Boolean.class || field.getType() == boolean.class) {
field.set(t, cursor.getInt(cursor.getColumnIndex(columnName)) == 1);
} else if (field.getType() == Float.class || field.getType() == float.class) {
field.set(t, cursor.getFloat(cursor.getColumnIndex(columnName)));
}
} catch (Exception e) {
Log.e(e.getMessage());
}
}
}