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 java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Modifier;
import mobi.cangol.mobile.logging.Log;
public class DatabaseUtils {
private DatabaseUtils() {
}
/**
* 创建表
*
* @param db
* @param clazz
*/
public static void createTable(SQLiteDatabase db, Class clazz) {
if (clazz.isAnnotationPresent(DatabaseTable.class)) {
StringBuilder sql = new StringBuilder("CREATE TABLE IF NOT EXISTS ");
DatabaseTable table = clazz.getAnnotation(DatabaseTable.class);
String tableName = "".equals(table.value()) ? clazz.getSimpleName() : table.value();
sql.append(tableName).append('(');
Field[] fields = clazz.getDeclaredFields();
String filedName = null;
boolean isFirst = true;
for (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;
}
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)) {
StringBuilder sql = new StringBuilder("DROP TABLE IF EXISTS ");
DatabaseTable table = clazz.getAnnotation(DatabaseTable.class);
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 clazz
* @return
*/
public static String getIdColumnName(Class clazz) {
String columnName = null;
for (Field field : clazz.getDeclaredFields()) {
field.setAccessible(true);
if (field.isEnumConstant() || Modifier.isFinal(field.getModifiers()) || Modifier.isTransient(field.getModifiers())) {
continue;
}
if (field.isAnnotationPresent(DatabaseField.class)) {
DatabaseField dbField = field.getAnnotation(DatabaseField.class);
if (dbField.primaryKey() == true) {
columnName = "".equals(dbField.value()) ? field.getName() : dbField.value();
break;
}
}
}
return columnName;
}
/**
* 获取主键的值
*
* @param obj
* @return
* @throws IllegalAccessException
* @throws IllegalArgumentException
*/
public static Object getIdValue(Object obj) throws IllegalAccessException, IllegalArgumentException {
Object value = null;
for (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)) {
DatabaseField dbField = field.getAnnotation(DatabaseField.class);
if (dbField.primaryKey() == true) {
value = field.get(obj);
break;
}
}
}
return value;
}
/**
* 获取键值对象
*
* @param object
* @return
* @throws IllegalAccessException
* @throws IllegalArgumentException
*/
public static ContentValues getContentValues(Object object) throws IllegalAccessException, IllegalArgumentException {
ContentValues v = new ContentValues();
for (Field field : object.getClass().getDeclaredFields()) {
field.setAccessible(true);
if (field.isAnnotationPresent(DatabaseField.class)) {
DatabaseField dbField = field.getAnnotation(DatabaseField.class);
if (!dbField.primaryKey()) {
String filedName = "".equals(dbField.value()) ? field.getName() : dbField.value();
v.put(filedName, String.valueOf(field.get(object)));
}
}
}
return v;
}
/**
* 查询记录的值，赋值给obj
*
* @param obj
* @param cursor
* @param
* @return
* @throws InstantiationException
* @throws IllegalAccessException
*/
public static T cursorToObject(T obj, Cursor cursor) throws InstantiationException, IllegalAccessException {
Field[] fields = obj.getClass().getDeclaredFields();
String columnName = null;
for (Field field : fields) {
field.setAccessible(true);
if (field.isEnumConstant() || Modifier.isFinal(field.getModifiers()) || Modifier.isTransient(field.getModifiers())) {
continue;
}
if (field.isAnnotationPresent(DatabaseField.class)) {
DatabaseField dbField = field.getAnnotation(DatabaseField.class);
columnName = "".equals(dbField.value()) ? field.getName() : dbField.value();
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 cursorToObject(Class clazz, Cursor cursor) throws InstantiationException, IllegalAccessException, NoSuchMethodException, InvocationTargetException {
Constructor constructor = clazz.getDeclaredConstructor();
constructor.setAccessible(true);
T obj = (T) constructor.newInstance();
return cursorToObject(obj, cursor);
}
/**
* 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 ? true : false);
} else if (field.getType() == Float.class || field.getType() == float.class) {
field.set(t, cursor.getFloat(cursor.getColumnIndex(columnName)));
}
} catch (IllegalArgumentException e) {
Log.e(e.getMessage());
} catch (IllegalAccessException e) {
Log.e(e.getMessage());
}
}
}