a

554325746@qq.com

2019-05-29 1bef2d413cd9eae19fa7aff0524f73aa3f94b113

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package com.basic.security.dao;
 
import android.content.Context;
import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase;
import android.util.Pair;
 
import com.basic.security.base.BaseApplication;
import com.basic.security.utils.Constants;
 
import java.io.File;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
 
import static android.content.Context.MODE_PRIVATE;
 
public class DatabaseManager {
 
    public static final Lock databaseLock = new ReentrantLock();
    public static final Lock databaseExecSqlLock = new ReentrantLock();
    public static Map<String, SQLiteDatabase> databaseMap = new HashMap<>();
 
    public static SQLiteDatabase getDatabase() {
        return getDatabase(Constants.databaseName);
    }
 
    public static void intiTables() {
        DatabaseManager.execSQL("CREATE TABLE IF NOT EXISTS alarm (\"matchPersonListStr\" TEXT, property TEXT, \"alarmLargePicture\" TEXT,\"alarmPerson\" TEXT,\"alarmPicture\" TEXT,\"alarmVideo\" TEXT,\"alarmTime\" TEXT,\"alarmAddress\" TEXT,\"alarmType\" TEXT,\"id\" TEXT, \"createTime\" TEXT, \"closeAlarm\" TEXT, \"mute\" TEXT, PRIMARY KEY(id))");
        DatabaseManager.execSQL("CREATE UNIQUE INDEX IF NOT EXISTS idx_alarm ON alarm(id)");
        DatabaseManager.execSQL("CREATE TABLE IF NOT EXISTS setting (\"id\" TEXT, \"name\" TEXT, PRIMARY KEY(id))");
        DatabaseManager.execSQL("CREATE UNIQUE INDEX IF NOT EXISTS idx_setting ON setting (id)");
    }
 
    public static SQLiteDatabase getDatabase(String databaseName) {
        SQLiteDatabase database = databaseMap.get(databaseName);
        try {
            databaseLock.lock();
            if (database == null) {
                database = BaseApplication.getApplication().openOrCreateDatabase(Constants.databaseName, MODE_PRIVATE, null);
                databaseMap.put(databaseName, database);
                intiTables();
                setCustomDatabaseFiles(BaseApplication.getApplication());
                initTableColumnNames();
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            databaseLock.unlock();
        }
        return database;
    }
 
    public static void execSQL(String sql) {
        try {
            databaseExecSqlLock.lock();
            getDatabase().execSQL(sql);
        } finally {
            databaseExecSqlLock.unlock();
        }
    }
 
    public static void setCustomDatabaseFiles(Context context) {
        try {
            Class<?> debugDB = Class.forName("com.amitshekhar.DebugDB");
            Class[] argTypes = new Class[]{HashMap.class};
            Method setCustomDatabaseFiles = debugDB.getMethod("setCustomDatabaseFiles", argTypes);
            HashMap<String, Pair<File, String>> customDatabaseFiles = new HashMap<>();
            customDatabaseFiles.put("base_database",new Pair<>(new File(Constants.databaseName), ""));
            setCustomDatabaseFiles.invoke(null, customDatabaseFiles);
        } catch (Exception ignore) {
            ignore.printStackTrace();
        }
    }
 
    private static void initTableColumnNames() {
        SQLiteDatabase mDataBase = DatabaseManager.getDatabase();
        Cursor c = mDataBase.rawQuery("SELECT name FROM sqlite_master WHERE type='table'", null);
        List<String> tableNameList = new ArrayList<>();
        if (c.moveToFirst()) {
            while (!c.isAfterLast()) {
                tableNameList.add(c.getString(0));
                c.moveToNext();
            }
        }
        c.close();
        for (String tableName : tableNameList) {
            Cursor dbCursor = mDataBase.query(tableName, null, null, null, null, null, null);
            String[] columnNames = dbCursor.getColumnNames();
            SqliteManager.tableColumnNames.put(tableName, Arrays.asList(columnNames));
            dbCursor.close();
        }
    }
 
}