forked from OtterMind/Chat2DB
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathindex.ts
More file actions
158 lines (145 loc) · 4.51 KB
/
Copy pathindex.ts
File metadata and controls
158 lines (145 loc) · 4.51 KB
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
import { tableList } from './table';
// 创建数据库的方法
export const createDB = (dbName: string, version: number) => {
return new Promise((resolve, reject) => {
const request = window.indexedDB.open(dbName, version);
request.onerror = (event: any) => {
reject(event.target.error);
};
request.onsuccess = (event: any) => {
resolve(event.target.result);
};
request.onupgradeneeded = (event: any) => {
const db = event.target.result; // 数据库对象
// 创建存储库
tableList.forEach((item: any) => {
const { tableDetails } = item;
const objectStore = db.createObjectStore(tableDetails.name, tableDetails.primaryKey);
tableDetails.column.forEach((i: any) => {
if (i.isIndex) {
objectStore.createIndex(i.name, i.keyPath, i.options);
}
});
});
};
});
};
type TableType = 'workspaceConsoleDDL';
type DBType = 'chat2db';
// 添加数据
export const addData = (db: DBType, tableName: TableType, data: any) => {
return new Promise((resolve, reject) => {
const transaction = window._indexedDB[db].transaction(tableName, 'readwrite');
const objectStore = transaction.objectStore(tableName);
const request = objectStore.add(data);
request.onsuccess = () => {
resolve(true);
};
request.onerror = (error) => {
reject(error);
};
});
};
// 通过索引删除数据
export const deleteDataByIndex = (db: DBType, tableName: TableType, indexName, indexValue) => {
return new Promise((resolve, reject) => {
const transaction = window._indexedDB[db].transaction(tableName, 'readwrite');
const objectStore = transaction.objectStore(tableName);
const request = objectStore.index(indexName).delete(indexValue);
request.onsuccess = () => {
resolve(true);
};
request.onerror = () => {
reject(false);
};
});
};
// 通过主键删除数据
export const deleteData = (db: DBType, tableName: TableType, key: any) => {
return new Promise((resolve, reject) => {
const transaction = window._indexedDB[db].transaction(tableName, 'readwrite');
const objectStore = transaction.objectStore(tableName);
const request = objectStore.delete(key);
request.onsuccess = () => {
resolve(true);
};
request.onerror = () => {
reject(false);
};
});
};
// 通过索引查询数据,支持传入多个索引
export const getDataByIndex = (db: DBType, tableName: TableType, indexName: string, indexValue: any) => {
return new Promise((resolve, reject) => {
const transaction = window._indexedDB[db].transaction(tableName, 'readwrite');
const objectStore = transaction.objectStore(tableName);
const request = objectStore.index(indexName).get(indexValue);
request.onsuccess = () => {
resolve(request.result);
};
request.onerror = () => {
reject(false);
};
});
};
// 通过游标查询数据，支持传入多个条件
export const getDataByCursor = (db: DBType, tableName: TableType, condition: {[key in string]: any}
) => {
return new Promise((resolve, reject) => {
const transaction = window._indexedDB[db].transaction(tableName, 'readwrite');
const objectStore = transaction.objectStore(tableName);
const request = objectStore.openCursor();
const result: any[] = [];
request.onsuccess = (event: any) => {
const cursor = event.target.result;
if (cursor) {
let flag = true;
Object.keys(condition).forEach((key) => {
if (cursor.value[key] !== condition[key]) {
flag = false;
}
});
if (flag) {
result.push(cursor.value);
}
cursor.continue();
} else {
resolve(result);
}
};
request.onerror = () => {
reject(false);
};
});
};
// 修改数据
export const updateData = (db: DBType, tableName: TableType, data: any) => {
return new Promise((resolve, reject) => {
const transaction = window._indexedDB[db].transaction(tableName, 'readwrite');
const objectStore = transaction.objectStore(tableName);
const request = objectStore.put(data);
request.onsuccess = () => {
resolve(true);
};
request.onerror = () => {
reject(false);
};
});
};
// 关闭数据库
export const closeDB = (db: DBType) => {
return new Promise((resolve) => {
window._indexedDB[db].close();
resolve(true);
});
};
export default {
createDB,
addData,
deleteDataByIndex,
deleteData,
getDataByIndex,
getDataByCursor,
updateData,
closeDB,
};