@cloudcome/utils-uni 1.5.1 → 1.6.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/database.cjs CHANGED
@@ -1,20 +1,4 @@
1
1
  "use strict";
2
- var __typeError = (msg) => {
3
- throw TypeError(msg);
4
- };
5
- var __accessCheck = (obj, member, msg) => member.has(obj) || __typeError("Cannot " + msg);
6
- var __privateGet = (obj, member, getter) => (__accessCheck(obj, member, "read from private field"), getter ? getter.call(obj) : member.get(obj));
7
- var __privateAdd = (obj, member, value) => member.has(obj) ? __typeError("Cannot add the same private member more than once") : member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
8
- var __privateSet = (obj, member, value, setter) => (__accessCheck(obj, member, "write to private field"), setter ? setter.call(obj, value) : member.set(obj, value), value);
9
- var __privateWrapper = (obj, member, setter, getter) => ({
10
- set _(value) {
11
- __privateSet(obj, member, value, setter);
12
- },
13
- get _() {
14
- return __privateGet(obj, member, getter);
15
- }
16
- });
17
- var _host, _isTransaction, _hasWhere, _hasWhereId, _hasSelect, _hasOrder, _hasSkip, _hasLimit;
18
2
  Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
19
3
  const error = require("@cloudcome/utils-core/error");
20
4
  const object = require("@cloudcome/utils-core/object");
@@ -24,97 +8,114 @@ const db0 = uniCloud.database();
24
8
  const dbCmd = db0.command;
25
9
  const dbAgg = db0.command.aggregate;
26
10
  class Db {
11
+ #host;
12
+ /**
13
+ * 是否为事务环境
14
+ * - 查询条件只能是 id
15
+ * - 不能聚合操作
16
+ */
17
+ #isTransaction;
27
18
  /**
28
19
  * 构造函数,初始化数据库集合引用
29
- * @param collection 集合名称
20
+ * @param collection 数据表名称
30
21
  * @param _mockDatabase 模拟数据库,用于单元测试
31
22
  */
32
23
  constructor(options) {
33
- __privateAdd(this, _host);
34
- /**
35
- * 是否为事务环境
36
- * - 查询条件只能是 id
37
- * - 不能聚合操作
38
- */
39
- __privateAdd(this, _isTransaction);
40
- __privateAdd(this, _hasWhere, 0);
41
- __privateAdd(this, _hasWhereId, 0);
42
- __privateAdd(this, _hasSelect, 0);
43
- __privateAdd(this, _hasOrder, 0);
44
- __privateAdd(this, _hasSkip, 0);
45
- __privateAdd(this, _hasLimit, 0);
46
- __privateSet(this, _host, options._mockDatabase || options.transaction || db0.collection(options.collection));
47
- __privateSet(this, _isTransaction, !!options.transaction);
24
+ this.#host = options._mockDatabase || options.transaction || db0.collection(options.table);
25
+ this.#isTransaction = !!options.transaction;
48
26
  }
49
27
  /**
50
28
  * 创建聚合操作实例
51
29
  * @returns 聚合操作实例
52
30
  */
53
31
  aggregate() {
54
- if (!__privateGet(this, _isTransaction)) throw new Error("db.aggregate() 不支持事务模式");
55
- return __privateGet(this, _host).aggregate();
32
+ if (!this.#isTransaction) throw new Error("db.aggregate() 不支持事务模式");
33
+ return this.#host.aggregate();
56
34
  }
57
- /**
58
- * 设置查询条件
59
- * @param where 查询条件对象
60
- * @returns 当前Db实例,支持链式执行
61
- */
62
- where(where) {
63
- if (__privateGet(this, _hasWhere)) throw new Error("db.where() 方法只能执行一次");
35
+ #hasWhere = void 0;
36
+ #hasWhereId = void 0;
37
+ #where(where, from) {
38
+ if (this.#hasWhere) throw new Error(`已调用过一次 db.${_toWhereMethod(this.#hasWhere)} 了`);
64
39
  const whereKeys = Object.keys(where);
65
40
  const isWhereId = whereKeys.length === 1 && whereKeys[0] === "_id";
66
- __privateWrapper(this, _hasWhere)._++;
41
+ if (isWhereId && this.#hasLimit) {
42
+ throw new Error(`db.${_toWhereIdMethod(from)} 方法不能与 db.limit() 方法同时调用`);
43
+ }
44
+ this.#hasWhere = from;
67
45
  if (isWhereId) {
68
- __privateWrapper(this, _hasWhereId)._++;
69
- __privateSet(this, _host, __privateGet(this, _host).doc(where._id));
46
+ this.#hasWhereId = from;
47
+ this.#host = this.#host.doc(where._id);
70
48
  } else {
71
- __privateSet(this, _host, __privateGet(this, _host).where(where));
49
+ this.#host = this.#host.where(where);
72
50
  }
73
51
  return this;
74
52
  }
53
+ /**
54
+ * 设置查询条件
55
+ * @param where 查询条件对象
56
+ * @returns 当前Db实例,支持链式调用
57
+ */
58
+ where(where) {
59
+ return this.#where(where, "where");
60
+ }
61
+ /**
62
+ * 根据ID设置查询条件
63
+ * @param id 记录ID
64
+ * @returns 当前Db实例,支持链式调用
65
+ */
66
+ whereId(id) {
67
+ return this.#where({ _id: id }, "whereId");
68
+ }
69
+ #hasSelect = 0;
75
70
  /**
76
71
  * 指定要返回的字段
77
72
  * @param fields 要返回的字段对象,true表示返回,false表示不返回
78
- * @returns 当前Db实例,支持链式执行
73
+ * @returns 当前Db实例,支持链式调用
79
74
  */
80
75
  select(fields) {
81
- if (__privateGet(this, _hasSelect)) throw new Error("db.select() 方法只能执行一次");
82
- __privateWrapper(this, _hasSelect)._++;
83
- __privateSet(this, _host, __privateGet(this, _host).field(fields));
76
+ if (this.#hasSelect) throw new Error("db.select() 方法只能调用一次");
77
+ this.#hasSelect++;
78
+ this.#host = this.#host.field(fields);
84
79
  return this;
85
80
  }
81
+ #hasOrder = 0;
86
82
  /**
87
83
  * 设置排序规则
88
84
  * @param order 排序规则对象,key为字段名,value为"asc"或"desc"
89
- * @returns 当前Db实例,支持链式执行
85
+ * @returns 当前Db实例,支持链式调用
90
86
  */
91
87
  order(order) {
92
- __privateWrapper(this, _hasOrder)._++;
88
+ this.#hasOrder++;
93
89
  object.objectEach(order, (val, key) => {
94
- __privateSet(this, _host, __privateGet(this, _host).orderBy(key, val));
90
+ this.#host = this.#host.orderBy(key, val);
95
91
  });
96
92
  return this;
97
93
  }
94
+ #hasSkip = 0;
98
95
  /**
99
96
  * 跳过指定数量的记录
100
97
  * @param skip 要跳过的记录数
101
- * @returns 当前Db实例,支持链式执行
98
+ * @returns 当前Db实例,支持链式调用
102
99
  */
103
100
  skip(skip) {
104
- if (__privateGet(this, _hasSkip)) throw new Error("db.skip() 方法只能执行一次");
105
- __privateWrapper(this, _hasSkip)._++;
106
- __privateSet(this, _host, __privateGet(this, _host).skip(skip));
101
+ if (this.#hasSkip) throw new Error("db.skip() 方法只能调用一次");
102
+ this.#hasSkip++;
103
+ this.#host = this.#host.skip(skip);
107
104
  return this;
108
105
  }
106
+ #hasLimit = 0;
109
107
  /**
110
108
  * 限制返回的记录数量
111
109
  * @param limit 最大返回记录数
112
- * @returns 当前Db实例,支持链式执行
110
+ * @returns 当前Db实例,支持链式调用
113
111
  */
114
112
  limit(limit) {
115
- if (__privateGet(this, _hasLimit)) throw new Error("db.limit() 方法只能执行一次");
116
- __privateWrapper(this, _hasLimit)._++;
117
- __privateSet(this, _host, __privateGet(this, _host).limit(limit));
113
+ if (this.#hasLimit) throw new Error("db.limit() 方法只能调用一次");
114
+ if (this.#hasWhereId) {
115
+ throw new Error(`db.limit() 方法不能与 ${_toWhereIdMethod(this.#hasWhereId)} 方法同时调用`);
116
+ }
117
+ this.#hasLimit++;
118
+ this.#host = this.#host.limit(limit);
118
119
  return this;
119
120
  }
120
121
  /**
@@ -123,26 +124,28 @@ class Db {
123
124
  * @returns 创建结果
124
125
  */
125
126
  async create(data) {
126
- if (__privateGet(this, _hasWhere)) throw new Error("db.create() 方法不支持 where 条件");
127
- if (__privateGet(this, _hasSelect)) throw new Error("db.create() 方法不支持 select 条件");
128
- if (__privateGet(this, _hasOrder)) throw new Error("db.create() 方法不支持 order 条件");
129
- if (__privateGet(this, _hasSkip)) throw new Error("db.create() 方法不支持 skip 条件");
130
- if (__privateGet(this, _hasLimit)) throw new Error("db.create() 方法不支持 limit 条件");
131
- const res = await __privateGet(this, _host).add(data);
132
- return parseDatabaseOutput(res);
127
+ if (this.#hasWhere) throw new Error("db.create() 方法不支持 where 条件");
128
+ if (this.#hasSelect) throw new Error("db.create() 方法不支持 select 条件");
129
+ if (this.#hasOrder) throw new Error("db.create() 方法不支持 order 条件");
130
+ if (this.#hasSkip) throw new Error("db.create() 方法不支持 skip 条件");
131
+ if (this.#hasLimit) throw new Error("db.create() 方法不支持 limit 条件");
132
+ const res = await this.#host.add(data);
133
+ const { id } = parseDatabaseOutput(res);
134
+ return id;
133
135
  }
134
136
  /**
135
137
  * 执行查询操作
136
138
  * @returns 查询结果
137
139
  */
138
140
  async query() {
139
- const res = await __privateGet(this, _host).get();
140
- return parseDatabaseOutput(res);
141
+ const res = await this.#host.get();
142
+ const { data } = parseDatabaseOutput(res);
143
+ return data;
141
144
  }
142
145
  async queryOne(ignoreMiss = false) {
143
- if (__privateGet(this, _hasLimit)) throw new Error("db.queryOne() 方法不支持 limit 条件");
144
- this.limit(1);
145
- const { data } = await this.query();
146
+ if (this.#hasLimit) throw new Error("db.queryOne() 方法不支持 limit 条件");
147
+ if (!this.#hasWhereId) this.limit(1);
148
+ const data = await this.query();
146
149
  const res = data.at(0);
147
150
  if (!ignoreMiss && !res) throw new Error("未找到匹配记录");
148
151
  return res;
@@ -152,12 +155,13 @@ class Db {
152
155
  * @returns 记录总数
153
156
  */
154
157
  async count() {
155
- if (__privateGet(this, _hasSelect)) throw new Error("db.count() 方法不支持 select 条件");
156
- if (__privateGet(this, _hasOrder)) throw new Error("db.count() 方法不支持 order 条件");
157
- if (__privateGet(this, _hasSkip)) throw new Error("db.count() 方法不支持 skip 条件");
158
- if (__privateGet(this, _hasLimit)) throw new Error("db.count() 方法不支持 limit 条件");
159
- const res = await __privateGet(this, _host).count();
160
- return parseDatabaseOutput(res);
158
+ if (this.#hasSelect) throw new Error("db.count() 方法不支持 select 条件");
159
+ if (this.#hasOrder) throw new Error("db.count() 方法不支持 order 条件");
160
+ if (this.#hasSkip) throw new Error("db.count() 方法不支持 skip 条件");
161
+ if (this.#hasLimit) throw new Error("db.count() 方法不支持 limit 条件");
162
+ const res = await this.#host.count();
163
+ const { total } = parseDatabaseOutput(res);
164
+ return total;
161
165
  }
162
166
  /**
163
167
  * 更新记录
@@ -165,46 +169,40 @@ class Db {
165
169
  * @returns 更新结果
166
170
  */
167
171
  async update(data) {
168
- if (!__privateGet(this, _hasWhere)) throw new Error("设置 where 条件后才能执行 db.update() 方法");
169
- if (__privateGet(this, _hasSelect)) throw new Error("db.update() 方法不支持 select 条件");
170
- if (__privateGet(this, _hasOrder)) throw new Error("db.update() 方法不支持 order 条件");
171
- if (__privateGet(this, _hasSkip)) throw new Error("db.update() 方法不支持 skip 条件");
172
- if (__privateGet(this, _hasLimit)) throw new Error("db.update() 方法不支持 limit 条件");
173
- if (__privateGet(this, _isTransaction) && !__privateGet(this, _hasWhereId)) throw new Error("事务模式下 db.update() 的 where 条件必须是 _id");
174
- const res = await __privateGet(this, _host).update(data);
175
- return parseDatabaseOutput(res);
172
+ if (!this.#hasWhere) throw new Error("设置 where 条件后才能执行 db.update() 方法");
173
+ if (this.#hasSelect) throw new Error("db.update() 方法不支持 select 条件");
174
+ if (this.#hasOrder) throw new Error("db.update() 方法不支持 order 条件");
175
+ if (this.#hasSkip) throw new Error("db.update() 方法不支持 skip 条件");
176
+ if (this.#hasLimit) throw new Error("db.update() 方法不支持 limit 条件");
177
+ if (this.#isTransaction && !this.#hasWhereId) throw new Error("事务模式下 db.update() 的 where 条件必须是 _id");
178
+ const res = await this.#host.update(data);
179
+ const { updated } = parseDatabaseOutput(res);
180
+ return updated;
176
181
  }
177
182
  /**
178
183
  * 删除记录
179
184
  * @returns 删除结果
180
185
  */
181
186
  async remove() {
182
- if (!__privateGet(this, _hasWhere)) throw new Error("设置 where 条件后才能执行 db.remove() 方法");
183
- if (__privateGet(this, _hasSelect)) throw new Error("db.remove() 方法不支持 select 条件");
184
- if (__privateGet(this, _hasOrder)) throw new Error("db.remove() 方法不支持 order 条件");
185
- if (__privateGet(this, _hasSkip)) throw new Error("db.remove() 方法不支持 skip 条件");
186
- if (__privateGet(this, _hasLimit)) throw new Error("db.remove() 方法不支持 limit 条件");
187
- if (__privateGet(this, _isTransaction) && !__privateGet(this, _hasWhereId)) throw new Error("事务模式下 db.remove() 的 where 条件必须是 _id");
188
- const res = await __privateGet(this, _host).remove();
189
- return parseDatabaseOutput(res);
187
+ if (!this.#hasWhere) throw new Error("设置 where 条件后才能执行 db.remove() 方法");
188
+ if (this.#hasSelect) throw new Error("db.remove() 方法不支持 select 条件");
189
+ if (this.#hasOrder) throw new Error("db.remove() 方法不支持 order 条件");
190
+ if (this.#hasSkip) throw new Error("db.remove() 方法不支持 skip 条件");
191
+ if (this.#hasLimit) throw new Error("db.remove() 方法不支持 limit 条件");
192
+ if (this.#isTransaction && !this.#hasWhereId) throw new Error("事务模式下 db.remove() 的 where 条件必须是 _id");
193
+ const res = await this.#host.remove();
194
+ const { deleted } = parseDatabaseOutput(res);
195
+ return deleted;
190
196
  }
191
197
  }
192
- _host = new WeakMap();
193
- _isTransaction = new WeakMap();
194
- _hasWhere = new WeakMap();
195
- _hasWhereId = new WeakMap();
196
- _hasSelect = new WeakMap();
197
- _hasOrder = new WeakMap();
198
- _hasSkip = new WeakMap();
199
- _hasLimit = new WeakMap();
200
198
  const db = {
201
199
  /**
202
200
  * 获取指定名称的数据库集合实例
203
- * @param collection 集合名称
201
+ * @param table 数据表名称
204
202
  * @returns Db类实例,用于执行数据库操作
205
203
  */
206
- collection(collection) {
207
- return new Db({ collection });
204
+ table(table) {
205
+ return new Db({ table });
208
206
  }
209
207
  };
210
208
  function parseDatabaseOutput(res) {
@@ -217,27 +215,45 @@ function parseDatabaseOutput(res) {
217
215
  }
218
216
  return res;
219
217
  }
218
+ function _toWhereMethod(whereFrom) {
219
+ return whereFrom === "where" ? "where({...})" : "whereId(id)";
220
+ }
221
+ function _toWhereIdMethod(whereFrom) {
222
+ return whereFrom === "where" ? "where({ _id })" : "whereId(id)";
223
+ }
220
224
  async function dbUpsert(options) {
221
- const { collection, where, create, update, onBeforeCreate, onAfterCreate, onBeforeUpdate, onAfterUpdate, _mockDb } = options;
222
- const _db = () => (_mockDb == null ? void 0 : _mockDb(collection)) || db.collection(collection);
223
- const found = await _db().where(where).limit(1).queryOne(true);
225
+ const {
226
+ collection,
227
+ where,
228
+ select = {},
229
+ create,
230
+ update,
231
+ onBeforeCreate,
232
+ onAfterCreate,
233
+ onBeforeUpdate,
234
+ onAfterUpdate,
235
+ _mockDb
236
+ } = options;
237
+ if ("_id" in select) throw new Error("select 条件不能包含 _id 字段");
238
+ const _db = () => _mockDb?.(collection) || db.table(collection);
239
+ const found = await _db().where(where).select(select || {}).limit(1).queryOne(true);
224
240
  if (found) {
225
- await (onBeforeUpdate == null ? void 0 : onBeforeUpdate(found));
241
+ await onBeforeUpdate?.(found);
226
242
  const updateData = type.isFunction(update) ? update(found) : update;
227
- const updated = await _db().where({ _id: found._id }).update(updateData);
228
- onAfterUpdate == null ? void 0 : onAfterUpdate();
243
+ const updated = await _db().whereId(found._id).update(updateData);
244
+ onAfterUpdate?.();
229
245
  return updated;
230
246
  }
231
- await (onBeforeCreate == null ? void 0 : onBeforeCreate());
232
- const created = await _db().create(create);
233
- await (onAfterCreate == null ? void 0 : onAfterCreate(created.id));
234
- return created;
247
+ await onBeforeCreate?.();
248
+ const createdId = await _db().create(create);
249
+ await onAfterCreate?.(createdId);
250
+ return createdId;
235
251
  }
236
252
  async function dbTransaction(transact, _mockDatabase) {
237
253
  const db2 = _mockDatabase || uniCloud.database();
238
254
  const [err1, transaction] = await _try.tryFlatten(db2.startTransaction());
239
255
  if (err1) throw err1;
240
- const ta = new Db({ collection: "", transaction });
256
+ const ta = new Db({ table: "", transaction });
241
257
  const [err2, result] = await _try.tryFlatten(async () => {
242
258
  const result2 = await transact(ta);
243
259
  await transaction.commit();
@@ -1 +1 @@
1
- {"version":3,"file":"database.cjs","sources":["../src/database/db.ts","../src/database/fns.ts"],"sourcesContent":["import { errorAssign } from '@cloudcome/utils-core/error';\nimport { objectEach, objectOmit } from '@cloudcome/utils-core/object';\nimport type { AnyObject, LowercaseStartString } from '@cloudcome/utils-core/types';\nimport type { UniClientDatabaseOutput, UniCloudDatabaseOutput } from './types';\n\ntype DbSelectPositive = {\n [key: LowercaseStartString]: true;\n};\ntype DbSelectNegative = { _id?: false };\nexport type DbSelect = Omit<DbSelectPositive, '_id'> & DbSelectNegative;\nexport type DbOrder = Record<string, 'asc' | 'desc'>;\nexport type DbRecord<S extends Record<string, boolean>> = {\n [K in keyof S as S[K] extends true ? K : never]: unknown;\n};\n\nconst db0 = uniCloud.database();\n/**\n * 数据库操作符命令\n */\nexport const dbCmd = db0.command;\n\n/**\n * 数据库聚合操作符命令\n */\nexport const dbAgg = db0.command.aggregate;\n\nclass Aggregate {\n #db: UniCloud.CollectionReference;\n\n constructor(db: UniCloud.CollectionReference) {\n this.#db = db;\n }\n\n start() {\n return this.#db.aggregate();\n }\n}\n\nexport type DbOptions = {\n /**\n * 集合名称\n */\n collection: string;\n\n /**\n * 事务对象,用于事务操作\n */\n // biome-ignore lint/suspicious/noExplicitAny: <explanation>\n transaction?: any;\n\n /**\n * 模拟数据库,用于单元测试\n */\n // biome-ignore lint/suspicious/noExplicitAny: <explanation>\n _mockDatabase?: any;\n};\n\nexport class Db {\n #host: UniCloud.CollectionReference;\n\n /**\n * 是否为事务环境\n * - 查询条件只能是 id\n * - 不能聚合操作\n */\n #isTransaction: boolean;\n\n /**\n * 构造函数,初始化数据库集合引用\n * @param collection 集合名称\n * @param _mockDatabase 模拟数据库,用于单元测试\n */\n constructor(options: DbOptions) {\n this.#host = options._mockDatabase || options.transaction || db0.collection(options.collection);\n this.#isTransaction = !!options.transaction;\n }\n\n /**\n * 创建聚合操作实例\n * @returns 聚合操作实例\n */\n aggregate() {\n if (!this.#isTransaction) throw new Error('db.aggregate() 不支持事务模式');\n\n return this.#host.aggregate();\n }\n\n #hasWhere = 0;\n #hasWhereId = 0;\n\n /**\n * 设置查询条件\n * @param where 查询条件对象\n * @returns 当前Db实例,支持链式执行\n */\n where(where: AnyObject) {\n if (this.#hasWhere) throw new Error('db.where() 方法只能执行一次');\n\n const whereKeys = Object.keys(where);\n const isWhereId = whereKeys.length === 1 && whereKeys[0] === '_id';\n this.#hasWhere++;\n\n if (isWhereId) {\n this.#hasWhereId++;\n // @ts-ignore\n this.#host = this.#host.doc(where._id);\n } else {\n // @ts-ignore\n this.#host = this.#host.where(where);\n }\n\n return this;\n }\n\n #hasSelect = 0;\n\n /**\n * 指定要返回的字段\n * @param fields 要返回的字段对象,true表示返回,false表示不返回\n * @returns 当前Db实例,支持链式执行\n */\n select(fields: DbSelect) {\n if (this.#hasSelect) throw new Error('db.select() 方法只能执行一次');\n\n this.#hasSelect++;\n // @ts-ignore\n this.#host = this.#host.field(fields);\n return this;\n }\n\n #hasOrder = 0;\n\n /**\n * 设置排序规则\n * @param order 排序规则对象,key为字段名,value为\"asc\"或\"desc\"\n * @returns 当前Db实例,支持链式执行\n */\n order(order: DbOrder) {\n this.#hasOrder++;\n objectEach(order, (val, key) => {\n // @ts-ignore\n this.#host = this.#host.orderBy(key, val);\n });\n\n return this;\n }\n\n #hasSkip = 0;\n\n /**\n * 跳过指定数量的记录\n * @param skip 要跳过的记录数\n * @returns 当前Db实例,支持链式执行\n */\n skip(skip: number) {\n if (this.#hasSkip) throw new Error('db.skip() 方法只能执行一次');\n\n this.#hasSkip++;\n // @ts-ignore\n this.#host = this.#host.skip(skip);\n return this;\n }\n\n #hasLimit = 0;\n\n /**\n * 限制返回的记录数量\n * @param limit 最大返回记录数\n * @returns 当前Db实例,支持链式执行\n */\n limit(limit: number) {\n if (this.#hasLimit) throw new Error('db.limit() 方法只能执行一次');\n\n this.#hasLimit++;\n // @ts-ignore\n this.#host = this.#host.limit(limit);\n return this;\n }\n\n /**\n * 创建新记录\n * @param data 要创建的数据\n * @returns 创建结果\n */\n async create(data: AnyObject) {\n if (this.#hasWhere) throw new Error('db.create() 方法不支持 where 条件');\n if (this.#hasSelect) throw new Error('db.create() 方法不支持 select 条件');\n if (this.#hasOrder) throw new Error('db.create() 方法不支持 order 条件');\n if (this.#hasSkip) throw new Error('db.create() 方法不支持 skip 条件');\n if (this.#hasLimit) throw new Error('db.create() 方法不支持 limit 条件');\n\n const res = await this.#host.add(data);\n return parseDatabaseOutput<{ id: string }>(res);\n }\n\n /**\n * 执行查询操作\n * @returns 查询结果\n */\n async query<T>() {\n const res = await this.#host.get();\n return parseDatabaseOutput<{ data: T[] }>(res);\n }\n\n /**\n * 只查询一条,自动添加 limit(1) 条件\n * @param ignoreMiss 是否忽略没有匹配到记录\n * @returns 查询结果\n */\n async queryOne<T>(): Promise<T>;\n async queryOne<T>(ignoreMiss: false): Promise<T>;\n async queryOne<T>(ignoreMiss: true): Promise<T | undefined>;\n async queryOne<T>(ignoreMiss = false): Promise<T | undefined> {\n if (this.#hasLimit) throw new Error('db.queryOne() 方法不支持 limit 条件');\n\n this.limit(1);\n const { data } = await this.query<T>();\n const res = data.at(0);\n\n if (!ignoreMiss && !res) throw new Error('未找到匹配记录');\n return res;\n }\n\n /**\n * 获取匹配记录的数量\n * @returns 记录总数\n */\n async count() {\n if (this.#hasSelect) throw new Error('db.count() 方法不支持 select 条件');\n if (this.#hasOrder) throw new Error('db.count() 方法不支持 order 条件');\n if (this.#hasSkip) throw new Error('db.count() 方法不支持 skip 条件');\n if (this.#hasLimit) throw new Error('db.count() 方法不支持 limit 条件');\n\n const res = await this.#host.count();\n return parseDatabaseOutput<{ total: number }>(res);\n }\n\n /**\n * 更新记录\n * @param data 要更新的数据\n * @returns 更新结果\n */\n async update(data: AnyObject) {\n if (!this.#hasWhere) throw new Error('设置 where 条件后才能执行 db.update() 方法');\n if (this.#hasSelect) throw new Error('db.update() 方法不支持 select 条件');\n if (this.#hasOrder) throw new Error('db.update() 方法不支持 order 条件');\n if (this.#hasSkip) throw new Error('db.update() 方法不支持 skip 条件');\n if (this.#hasLimit) throw new Error('db.update() 方法不支持 limit 条件');\n\n if (this.#isTransaction && !this.#hasWhereId) throw new Error('事务模式下 db.update() 的 where 条件必须是 _id');\n\n const res = await this.#host.update(data);\n return parseDatabaseOutput<{ updated: number }>(res);\n }\n\n /**\n * 删除记录\n * @returns 删除结果\n */\n async remove() {\n if (!this.#hasWhere) throw new Error('设置 where 条件后才能执行 db.remove() 方法');\n if (this.#hasSelect) throw new Error('db.remove() 方法不支持 select 条件');\n if (this.#hasOrder) throw new Error('db.remove() 方法不支持 order 条件');\n if (this.#hasSkip) throw new Error('db.remove() 方法不支持 skip 条件');\n if (this.#hasLimit) throw new Error('db.remove() 方法不支持 limit 条件');\n\n if (this.#isTransaction && !this.#hasWhereId) throw new Error('事务模式下 db.remove() 的 where 条件必须是 _id');\n\n const res = await this.#host.remove();\n return parseDatabaseOutput<{ deleted: number }>(res);\n }\n}\n\n/**\n * 数据库操作对象\n */\nexport const db = {\n /**\n * 获取指定名称的数据库集合实例\n * @param collection 集合名称\n * @returns Db类实例,用于执行数据库操作\n */\n collection(collection: string) {\n return new Db({ collection });\n },\n};\n\n/**\n * 解析数据库执行结果\n * @param res 客户端、云端响应结果\n * @returns 处理后的结果\n */\nexport function parseDatabaseOutput<T>(res: UniClientDatabaseOutput<T> | UniCloudDatabaseOutput<T>) {\n const keys = Object.keys(res as AnyObject);\n // 客户端 { result: {errCode: 0, errMsg: 'ok'} & 数据 }\n const isClient = keys.length === 1 && keys[0] === 'result';\n\n if (isClient) {\n const { result } = res as UniClientDatabaseOutput<T>;\n if (!result.errCode) return objectOmit(result, ['errCode', 'errMsg', 'code', 'message']);\n throw errorAssign(new Error(result.errMsg), result);\n }\n\n // 云端 数据\n return res as T;\n}\n","import { tryFlatten } from '@cloudcome/utils-core/try';\nimport { isFunction } from '@cloudcome/utils-core/type';\nimport type { AnyObject, MaybeCallable } from '@cloudcome/utils-core/types';\nimport { Db, type DbSelect, db } from './db';\n\n/**\n * 数据库 upsert 操作的配置选项\n *\n * @template W - 查询条件类型\n * @template S - 查询返回字段类型\n * @template C - 创建数据类型\n * @template U - 更新数据类型\n * @template R - 查询结果类型\n */\nexport type DbUpsertOptions<\n W extends AnyObject,\n S extends DbSelect,\n C extends AnyObject,\n U extends AnyObject,\n R extends AnyObject,\n> = {\n /** 集合名称 */\n collection: string;\n\n /** 查询条件 */\n where: W;\n\n /** 查询返回字段 */\n select?: S;\n\n /** 创建数据 */\n create: C;\n\n /**\n * 更新数据,可以是对象或根据查询结果生成更新对象的函数\n * @param row 查询到的文档数据,仅在传入函数时可用\n */\n update: U | ((row: R) => U);\n\n /** 创建前回调函数 */\n onBeforeCreate?: () => unknown;\n\n /**\n * 创建后回调函数\n * @param id 创建的文档ID\n */\n onAfterCreate?: (id: string) => unknown;\n\n /**\n * 更新前回调函数\n * @param row 查询到的文档\n */\n onBeforeUpdate?: (row: R) => unknown;\n\n /** 更新后回调函数 */\n onAfterUpdate?: () => unknown;\n\n /** 用于测试的模拟数据库实例 */\n // biome-ignore lint/suspicious/noExplicitAny: <explanation>\n _mockDb?: (collection: string) => any;\n};\n\nexport async function dbUpsert<\n W extends AnyObject,\n S extends DbSelect,\n C extends AnyObject,\n U extends AnyObject,\n R extends AnyObject,\n>(options: DbUpsertOptions<W, S, C, U, R>) {\n const { collection, where, create, update, onBeforeCreate, onAfterCreate, onBeforeUpdate, onAfterUpdate, _mockDb } =\n options;\n const _db = () => (_mockDb?.(collection) || db.collection(collection)) as Db;\n const found = await _db().where(where).limit(1).queryOne<R>(true);\n\n if (found) {\n await onBeforeUpdate?.(found);\n const updateData = isFunction(update) ? update(found) : update;\n const updated = await _db().where({ _id: found._id }).update(updateData);\n onAfterUpdate?.();\n\n return updated;\n }\n\n await onBeforeCreate?.();\n const created = await _db().create(create);\n await onAfterCreate?.(created.id);\n\n return created;\n}\n\ntype _TransactionDb = {\n startTransaction: () => Promise<_Transaction>;\n};\n\ntype _Transaction = {\n commit: () => Promise<unknown>;\n rollback: () => Promise<unknown>;\n};\n\n/**\n * 在数据库事务中执行操作\n *\n * @template T - 事务操作返回值类型\n * @param transact - 事务执行函数,接收事务数据库实例作为参数\n * @param _mockDatabase - 用于测试的模拟数据库实例\n * @returns 事务操作的返回结果\n *\n * @example\n * ```typescript\n * const result = await dbTransaction(async (ta) => {\n * const user = await ta.collection('users').create({ name: 'John' });\n * const order = await ta.collection('orders').create({ userId: user.id, amount: 100 });\n * return { user, order };\n * });\n * ```\n */\nexport async function dbTransaction<T>(transact: (ta: Db) => Promise<T>, _mockDatabase?: _TransactionDb) {\n const db = (_mockDatabase || uniCloud.database()) as _TransactionDb;\n\n const [err1, transaction] = await tryFlatten(db.startTransaction());\n if (err1) throw err1;\n\n const ta = new Db({ collection: '', transaction });\n const [err2, result] = await tryFlatten(async () => {\n const result = await transact(ta);\n await transaction.commit();\n return result;\n });\n\n if (err2) {\n await tryFlatten(transaction.rollback());\n throw err2;\n }\n\n return result;\n}\n"],"names":["objectEach","objectOmit","errorAssign","isFunction","db","tryFlatten","result"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAeA,MAAM,MAAM,SAAS,SAAS;AAIvB,MAAM,QAAQ,IAAI;AAKZ,MAAA,QAAQ,IAAI,QAAQ;AAiC1B,MAAM,GAAG;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAed,YAAY,SAAoB;AAdhC;AAOA;AAAA;AAAA;AAAA;AAAA;AAAA;AAsBA,kCAAY;AACZ,oCAAc;AA0Bd,mCAAa;AAgBb,kCAAY;AAiBZ,iCAAW;AAgBX,kCAAY;AA1FL,uBAAA,OAAQ,QAAQ,iBAAiB,QAAQ,eAAe,IAAI,WAAW,QAAQ,UAAU;AACzF,uBAAA,gBAAiB,CAAC,CAAC,QAAQ;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOlC,YAAY;AACV,QAAI,CAAC,mBAAK,gBAAsB,OAAA,IAAI,MAAM,wBAAwB;AAE3D,WAAA,mBAAK,OAAM,UAAU;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAW9B,MAAM,OAAkB;AACtB,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,qBAAqB;AAEnD,UAAA,YAAY,OAAO,KAAK,KAAK;AACnC,UAAM,YAAY,UAAU,WAAW,KAAK,UAAU,CAAC,MAAM;AACxD,2BAAA,WAAA;AAEL,QAAI,WAAW;AACR,6BAAA,aAAA;AAEL,yBAAK,OAAQ,mBAAK,OAAM,IAAI,MAAM,GAAG;AAAA,IAAA,OAChC;AAEL,yBAAK,OAAQ,mBAAK,OAAM,MAAM,KAAK;AAAA,IAAA;AAG9B,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUT,OAAO,QAAkB;AACvB,QAAI,mBAAK,YAAkB,OAAA,IAAI,MAAM,sBAAsB;AAEtD,2BAAA,YAAA;AAEL,uBAAK,OAAQ,mBAAK,OAAM,MAAM,MAAM;AAC7B,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUT,MAAM,OAAgB;AACf,2BAAA,WAAA;AACMA,WAAAA,WAAA,OAAO,CAAC,KAAK,QAAQ;AAE9B,yBAAK,OAAQ,mBAAK,OAAM,QAAQ,KAAK,GAAG;AAAA,IAAA,CACzC;AAEM,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUT,KAAK,MAAc;AACjB,QAAI,mBAAK,UAAgB,OAAA,IAAI,MAAM,oBAAoB;AAElD,2BAAA,UAAA;AAEL,uBAAK,OAAQ,mBAAK,OAAM,KAAK,IAAI;AAC1B,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUT,MAAM,OAAe;AACnB,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,qBAAqB;AAEpD,2BAAA,WAAA;AAEL,uBAAK,OAAQ,mBAAK,OAAM,MAAM,KAAK;AAC5B,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQT,MAAM,OAAO,MAAiB;AAC5B,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,4BAA4B;AAChE,QAAI,mBAAK,YAAkB,OAAA,IAAI,MAAM,6BAA6B;AAClE,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,4BAA4B;AAChE,QAAI,mBAAK,UAAgB,OAAA,IAAI,MAAM,2BAA2B;AAC9D,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,4BAA4B;AAEhE,UAAM,MAAM,MAAM,mBAAK,OAAM,IAAI,IAAI;AACrC,WAAO,oBAAoC,GAAG;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOhD,MAAM,QAAW;AACf,UAAM,MAAM,MAAM,mBAAK,OAAM,IAAI;AACjC,WAAO,oBAAmC,GAAG;AAAA,EAAA;AAAA,EAW/C,MAAM,SAAY,aAAa,OAA+B;AAC5D,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,8BAA8B;AAElE,SAAK,MAAM,CAAC;AACZ,UAAM,EAAE,KAAA,IAAS,MAAM,KAAK,MAAS;AAC/B,UAAA,MAAM,KAAK,GAAG,CAAC;AAErB,QAAI,CAAC,cAAc,CAAC,IAAW,OAAA,IAAI,MAAM,SAAS;AAC3C,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOT,MAAM,QAAQ;AACZ,QAAI,mBAAK,YAAkB,OAAA,IAAI,MAAM,4BAA4B;AACjE,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,2BAA2B;AAC/D,QAAI,mBAAK,UAAgB,OAAA,IAAI,MAAM,0BAA0B;AAC7D,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,2BAA2B;AAE/D,UAAM,MAAM,MAAM,mBAAK,OAAM,MAAM;AACnC,WAAO,oBAAuC,GAAG;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQnD,MAAM,OAAO,MAAiB;AAC5B,QAAI,CAAC,mBAAK,WAAiB,OAAA,IAAI,MAAM,iCAAiC;AACtE,QAAI,mBAAK,YAAkB,OAAA,IAAI,MAAM,6BAA6B;AAClE,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,4BAA4B;AAChE,QAAI,mBAAK,UAAgB,OAAA,IAAI,MAAM,2BAA2B;AAC9D,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,4BAA4B;AAE5D,QAAA,mBAAK,mBAAkB,CAAC,mBAAK,aAAmB,OAAA,IAAI,MAAM,qCAAqC;AAEnG,UAAM,MAAM,MAAM,mBAAK,OAAM,OAAO,IAAI;AACxC,WAAO,oBAAyC,GAAG;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOrD,MAAM,SAAS;AACb,QAAI,CAAC,mBAAK,WAAiB,OAAA,IAAI,MAAM,iCAAiC;AACtE,QAAI,mBAAK,YAAkB,OAAA,IAAI,MAAM,6BAA6B;AAClE,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,4BAA4B;AAChE,QAAI,mBAAK,UAAgB,OAAA,IAAI,MAAM,2BAA2B;AAC9D,QAAI,mBAAK,WAAiB,OAAA,IAAI,MAAM,4BAA4B;AAE5D,QAAA,mBAAK,mBAAkB,CAAC,mBAAK,aAAmB,OAAA,IAAI,MAAM,qCAAqC;AAEnG,UAAM,MAAM,MAAM,mBAAK,OAAM,OAAO;AACpC,WAAO,oBAAyC,GAAG;AAAA,EAAA;AAEvD;AArNE;AAOA;AAsBA;AACA;AA0BA;AAgBA;AAiBA;AAgBA;AAiHK,MAAM,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAMhB,WAAW,YAAoB;AAC7B,WAAO,IAAI,GAAG,EAAE,YAAY;AAAA,EAAA;AAEhC;AAOO,SAAS,oBAAuB,KAA6D;AAC5F,QAAA,OAAO,OAAO,KAAK,GAAgB;AAEzC,QAAM,WAAW,KAAK,WAAW,KAAK,KAAK,CAAC,MAAM;AAElD,MAAI,UAAU;AACN,UAAA,EAAE,WAAW;AACf,QAAA,CAAC,OAAO,QAAgB,QAAAC,kBAAW,QAAQ,CAAC,WAAW,UAAU,QAAQ,SAAS,CAAC;AACvF,UAAMC,MAAAA,YAAY,IAAI,MAAM,OAAO,MAAM,GAAG,MAAM;AAAA,EAAA;AAI7C,SAAA;AACT;ACnPA,eAAsB,SAMpB,SAAyC;AACnC,QAAA,EAAE,YAAY,OAAO,QAAQ,QAAQ,gBAAgB,eAAe,gBAAgB,eAAe,QACvG,IAAA;AACF,QAAM,MAAM,OAAO,mCAAU,gBAAe,GAAG,WAAW,UAAU;AAC9D,QAAA,QAAQ,MAAM,IAAM,EAAA,MAAM,KAAK,EAAE,MAAM,CAAC,EAAE,SAAY,IAAI;AAEhE,MAAI,OAAO;AACT,WAAM,iDAAiB;AACvB,UAAM,aAAaC,KAAAA,WAAW,MAAM,IAAI,OAAO,KAAK,IAAI;AACxD,UAAM,UAAU,MAAM,IAAI,EAAE,MAAM,EAAE,KAAK,MAAM,IAAI,CAAC,EAAE,OAAO,UAAU;AACvD;AAET,WAAA;AAAA,EAAA;AAGT,SAAM;AACN,QAAM,UAAU,MAAM,MAAM,OAAO,MAAM;AACnC,SAAA,+CAAgB,QAAQ;AAEvB,SAAA;AACT;AA4BsB,eAAA,cAAiB,UAAkC,eAAgC;AACjGC,QAAAA,MAAM,iBAAiB,SAAS,SAAS;AAEzC,QAAA,CAAC,MAAM,WAAW,IAAI,MAAMC,KAAAA,WAAWD,IAAG,kBAAkB;AAClE,MAAI,KAAY,OAAA;AAEhB,QAAM,KAAK,IAAI,GAAG,EAAE,YAAY,IAAI,aAAa;AACjD,QAAM,CAAC,MAAM,MAAM,IAAI,MAAMC,gBAAW,YAAY;AAC5CC,UAAAA,UAAS,MAAM,SAAS,EAAE;AAChC,UAAM,YAAY,OAAO;AAClBA,WAAAA;AAAAA,EAAA,CACR;AAED,MAAI,MAAM;AACF,UAAAD,KAAA,WAAW,YAAY,UAAU;AACjC,UAAA;AAAA,EAAA;AAGD,SAAA;AACT;;;;;;;;"}
1
+ {"version":3,"file":"database.cjs","sources":["../src/database/db.ts","../src/database/fns.ts"],"sourcesContent":["import { errorAssign } from '@cloudcome/utils-core/error';\nimport { objectEach, objectOmit } from '@cloudcome/utils-core/object';\nimport type {\n AnyObject,\n HasProperty,\n IsEmptyObject,\n IsOnlyProperty,\n LowercaseStartString,\n} from '@cloudcome/utils-core/types';\nimport type { UniClientDatabaseOutput, UniCloudDatabaseOutput } from './types';\n\nexport type DbWhere<T> = {\n [K in keyof T]?: unknown;\n};\nexport type DbSelect<T> = {\n [K in keyof T]?: K extends '_id' ? false : true;\n};\nexport type DbFieldsDefault<T> = {\n [K in keyof T]: true;\n};\ntype _DbFields<T, S extends DbSelect<T>> = IsEmptyObject<S> extends true // 判断是否为空对象\n ? // 默认全部字段\n DbFieldsDefault<T>\n : // 判断 _id 是否为唯一属性\n IsOnlyProperty<S, '_id'> extends true\n ? // 从默认字段里排除 _id\n Omit<DbFieldsDefault<T>, '_id'>\n : // 判断是否有 _id\n HasProperty<S, '_id'> extends true\n ? // 有的话保留 {_id, ...}\n S\n : // 没有的话补上 {_id, ...}\n S & { _id: true };\ntype _DbQuery<T, S extends Record<keyof T, boolean>> = {\n [K in keyof T as S[K] extends true ? K : never]: T[K];\n};\n// @ts-ignore\nexport type DbQuery<T, S extends DbSelect<T>> = _DbQuery<T, _DbFields<T, S>>;\nexport type DbCreate<T> = Partial<T>;\nexport type DbUpdate<T> = Partial<T>;\nexport type DbOrder<T> = Record<keyof T, 'asc' | 'desc'>;\n\ntype _WhereFrom = 'where' | 'whereId';\n\nconst db0 = uniCloud.database();\n/**\n * 数据库操作符命令\n */\nexport const dbCmd = db0.command;\n\n/**\n * 数据库聚合操作符命令\n */\nexport const dbAgg = db0.command.aggregate;\n\nclass Aggregate {\n #db: UniCloud.CollectionReference;\n\n constructor(db: UniCloud.CollectionReference) {\n this.#db = db;\n }\n\n start() {\n return this.#db.aggregate();\n }\n}\n\nexport type DbOptions = {\n /**\n * 数据表名称\n */\n table: string;\n\n /**\n * 事务对象,用于事务操作\n */\n // biome-ignore lint/suspicious/noExplicitAny: <explanation>\n transaction?: any;\n\n /**\n * 模拟数据库,用于单元测试\n */\n // biome-ignore lint/suspicious/noExplicitAny: <explanation>\n _mockDatabase?: any;\n};\n\nexport class Db<T, S extends DbSelect<T> = Record<string, never>> {\n #host: UniCloud.CollectionReference;\n\n /**\n * 是否为事务环境\n * - 查询条件只能是 id\n * - 不能聚合操作\n */\n #isTransaction: boolean;\n\n /**\n * 构造函数,初始化数据库集合引用\n * @param collection 数据表名称\n * @param _mockDatabase 模拟数据库,用于单元测试\n */\n constructor(options: DbOptions) {\n this.#host = options._mockDatabase || options.transaction || db0.collection(options.table);\n this.#isTransaction = !!options.transaction;\n }\n\n /**\n * 创建聚合操作实例\n * @returns 聚合操作实例\n */\n aggregate() {\n if (!this.#isTransaction) throw new Error('db.aggregate() 不支持事务模式');\n\n return this.#host.aggregate();\n }\n\n #hasWhere: _WhereFrom | undefined = undefined;\n #hasWhereId: _WhereFrom | undefined = undefined;\n\n #where(where: DbWhere<T>, from: _WhereFrom) {\n if (this.#hasWhere) throw new Error(`已调用过一次 db.${_toWhereMethod(this.#hasWhere)} 了`);\n\n const whereKeys = Object.keys(where);\n const isWhereId = whereKeys.length === 1 && whereKeys[0] === '_id';\n\n if (isWhereId && this.#hasLimit) {\n throw new Error(`db.${_toWhereIdMethod(from)} 方法不能与 db.limit() 方法同时调用`);\n }\n\n this.#hasWhere = from;\n\n if (isWhereId) {\n this.#hasWhereId = from;\n // @ts-ignore\n this.#host = this.#host.doc(where._id);\n } else {\n // @ts-ignore\n this.#host = this.#host.where(where);\n }\n\n return this;\n }\n\n /**\n * 设置查询条件\n * @param where 查询条件对象\n * @returns 当前Db实例,支持链式调用\n */\n where(where: DbWhere<T>) {\n return this.#where(where, 'where');\n }\n\n /**\n * 根据ID设置查询条件\n * @param id 记录ID\n * @returns 当前Db实例,支持链式调用\n */\n whereId(id: string) {\n // @ts-ignore\n return this.#where({ _id: id }, 'whereId');\n }\n\n #hasSelect = 0;\n\n /**\n * 指定要返回的字段\n * @param fields 要返回的字段对象,true表示返回,false表示不返回\n * @returns 当前Db实例,支持链式调用\n */\n select<U extends DbSelect<T>>(fields: U): Db<T, U> {\n if (this.#hasSelect) throw new Error('db.select() 方法只能调用一次');\n\n this.#hasSelect++;\n // @ts-ignore\n this.#host = this.#host.field(fields);\n // @ts-ignore\n return this;\n }\n\n #hasOrder = 0;\n\n /**\n * 设置排序规则\n * @param order 排序规则对象,key为字段名,value为\"asc\"或\"desc\"\n * @returns 当前Db实例,支持链式调用\n */\n order(order: DbOrder<T>) {\n this.#hasOrder++;\n objectEach(order, (val, key) => {\n // @ts-ignore\n this.#host = this.#host.orderBy(key, val);\n });\n\n return this;\n }\n\n #hasSkip = 0;\n\n /**\n * 跳过指定数量的记录\n * @param skip 要跳过的记录数\n * @returns 当前Db实例,支持链式调用\n */\n skip(skip: number) {\n if (this.#hasSkip) throw new Error('db.skip() 方法只能调用一次');\n\n this.#hasSkip++;\n // @ts-ignore\n this.#host = this.#host.skip(skip);\n return this;\n }\n\n #hasLimit = 0;\n\n /**\n * 限制返回的记录数量\n * @param limit 最大返回记录数\n * @returns 当前Db实例,支持链式调用\n */\n limit(limit: number) {\n if (this.#hasLimit) throw new Error('db.limit() 方法只能调用一次');\n\n if (this.#hasWhereId) {\n throw new Error(`db.limit() 方法不能与 ${_toWhereIdMethod(this.#hasWhereId)} 方法同时调用`);\n }\n\n this.#hasLimit++;\n // @ts-ignore\n this.#host = this.#host.limit(limit);\n return this;\n }\n\n /**\n * 创建新记录\n * @param data 要创建的数据\n * @returns 创建结果\n */\n async create(data: DbCreate<T>) {\n if (this.#hasWhere) throw new Error('db.create() 方法不支持 where 条件');\n if (this.#hasSelect) throw new Error('db.create() 方法不支持 select 条件');\n if (this.#hasOrder) throw new Error('db.create() 方法不支持 order 条件');\n if (this.#hasSkip) throw new Error('db.create() 方法不支持 skip 条件');\n if (this.#hasLimit) throw new Error('db.create() 方法不支持 limit 条件');\n\n const res = await this.#host.add(data);\n const { id } = parseDatabaseOutput<{ id: string }>(res);\n return id;\n }\n\n /**\n * 执行查询操作\n * @returns 查询结果\n */\n async query() {\n const res = await this.#host.get();\n const { data } = parseDatabaseOutput<{ data: DbQuery<T, S>[] }>(res);\n return data;\n }\n\n /**\n * 只查询一条,自动添加 limit(1) 条件\n * @param ignoreMiss 是否忽略没有匹配到记录\n * @returns 查询结果\n */\n async queryOne(): Promise<DbQuery<T, S>>;\n async queryOne(ignoreMiss: false): Promise<DbQuery<T, S>>;\n async queryOne(ignoreMiss: true): Promise<DbQuery<T, S> | undefined>;\n async queryOne(ignoreMiss = false): Promise<DbQuery<T, S> | undefined> {\n if (this.#hasLimit) throw new Error('db.queryOne() 方法不支持 limit 条件');\n if (!this.#hasWhereId) this.limit(1);\n\n const data = await this.query();\n const res = data.at(0);\n\n if (!ignoreMiss && !res) throw new Error('未找到匹配记录');\n return res;\n }\n\n /**\n * 获取匹配记录的数量\n * @returns 记录总数\n */\n async count() {\n if (this.#hasSelect) throw new Error('db.count() 方法不支持 select 条件');\n if (this.#hasOrder) throw new Error('db.count() 方法不支持 order 条件');\n if (this.#hasSkip) throw new Error('db.count() 方法不支持 skip 条件');\n if (this.#hasLimit) throw new Error('db.count() 方法不支持 limit 条件');\n\n const res = await this.#host.count();\n const { total } = parseDatabaseOutput<{ total: number }>(res);\n return total;\n }\n\n /**\n * 更新记录\n * @param data 要更新的数据\n * @returns 更新结果\n */\n async update(data: AnyObject) {\n if (!this.#hasWhere) throw new Error('设置 where 条件后才能执行 db.update() 方法');\n if (this.#hasSelect) throw new Error('db.update() 方法不支持 select 条件');\n if (this.#hasOrder) throw new Error('db.update() 方法不支持 order 条件');\n if (this.#hasSkip) throw new Error('db.update() 方法不支持 skip 条件');\n if (this.#hasLimit) throw new Error('db.update() 方法不支持 limit 条件');\n\n if (this.#isTransaction && !this.#hasWhereId) throw new Error('事务模式下 db.update() 的 where 条件必须是 _id');\n\n const res = await this.#host.update(data);\n const { updated } = parseDatabaseOutput<{ updated: number }>(res);\n return updated;\n }\n\n /**\n * 删除记录\n * @returns 删除结果\n */\n async remove() {\n if (!this.#hasWhere) throw new Error('设置 where 条件后才能执行 db.remove() 方法');\n if (this.#hasSelect) throw new Error('db.remove() 方法不支持 select 条件');\n if (this.#hasOrder) throw new Error('db.remove() 方法不支持 order 条件');\n if (this.#hasSkip) throw new Error('db.remove() 方法不支持 skip 条件');\n if (this.#hasLimit) throw new Error('db.remove() 方法不支持 limit 条件');\n\n if (this.#isTransaction && !this.#hasWhereId) throw new Error('事务模式下 db.remove() 的 where 条件必须是 _id');\n\n const res = await this.#host.remove();\n const { deleted } = parseDatabaseOutput<{ deleted: number }>(res);\n return deleted;\n }\n}\n\n/**\n * 数据库操作对象\n */\nexport const db = {\n /**\n * 获取指定名称的数据库集合实例\n * @param table 数据表名称\n * @returns Db类实例,用于执行数据库操作\n */\n table<T, S extends DbSelect<T> = Record<never, never>>(table: string) {\n return new Db<T, S>({ table });\n },\n};\n\n/**\n * 解析数据库执行结果\n * @param res 客户端、云端响应结果\n * @returns 处理后的结果\n */\nexport function parseDatabaseOutput<T>(res: UniClientDatabaseOutput<T> | UniCloudDatabaseOutput<T>) {\n const keys = Object.keys(res as AnyObject);\n // 客户端 { result: {errCode: 0, errMsg: 'ok'} & 数据 }\n const isClient = keys.length === 1 && keys[0] === 'result';\n\n if (isClient) {\n const { result } = res as UniClientDatabaseOutput<T>;\n if (!result.errCode) return objectOmit(result, ['errCode', 'errMsg', 'code', 'message']);\n throw errorAssign(new Error(result.errMsg), result);\n }\n\n // 云端 数据\n return res as T;\n}\n\nfunction _toWhereMethod(whereFrom: _WhereFrom) {\n return whereFrom === 'where' ? 'where({...})' : 'whereId(id)';\n}\n\nfunction _toWhereIdMethod(whereFrom: _WhereFrom) {\n return whereFrom === 'where' ? 'where({ _id })' : 'whereId(id)';\n}\n","import { tryFlatten } from '@cloudcome/utils-core/try';\nimport { isFunction } from '@cloudcome/utils-core/type';\nimport type { AnyObject, MaybeCallable } from '@cloudcome/utils-core/types';\nimport { Db, type DbCreate, type DbQuery, type DbSelect, type DbUpdate, type DbWhere, db } from './db';\n\n/**\n * 数据库 upsert 操作的配置选项\n *\n * @template W - 查询条件类型\n * @template S - 查询返回字段类型\n * @template C - 创建数据类型\n * @template U - 更新数据类型\n * @template R - 查询结果类型\n */\nexport type DbUpsertOptions<\n T,\n W extends DbWhere<T>,\n S extends DbSelect<T>,\n C extends DbCreate<T>,\n U extends DbUpdate<T>,\n> = {\n /** 集合名称 */\n collection: string;\n\n /** 查询条件 */\n where: W;\n\n /** 查询返回字段 */\n select?: S;\n\n /** 创建数据 */\n create: C;\n\n /**\n * 更新数据,可以是对象或根据查询结果生成更新对象的函数\n * @param row 查询到的文档数据,仅在传入函数时可用\n */\n update: U | ((row: DbQuery<T, S>) => U);\n\n /** 创建前回调函数 */\n onBeforeCreate?: () => unknown;\n\n /**\n * 创建后回调函数\n * @param id 创建的文档ID\n */\n onAfterCreate?: (id: string) => unknown;\n\n /**\n * 更新前回调函数\n * @param row 查询到的文档\n */\n onBeforeUpdate?: (row: DbQuery<T, S>) => unknown;\n\n /** 更新后回调函数 */\n onAfterUpdate?: () => unknown;\n\n /** 用于测试的模拟数据库实例 */\n // biome-ignore lint/suspicious/noExplicitAny: <explanation>\n _mockDb?: (collection: string) => any;\n};\n\nexport async function dbUpsert<\n T,\n W extends DbWhere<T>,\n S extends DbSelect<T>,\n C extends DbCreate<T>,\n U extends DbUpdate<T>,\n>(options: DbUpsertOptions<T, W, S, C, U>) {\n const {\n collection,\n where,\n select = {},\n create,\n update,\n onBeforeCreate,\n onAfterCreate,\n onBeforeUpdate,\n onAfterUpdate,\n _mockDb,\n } = options;\n\n // @ts-ignore\n if ('_id' in select) throw new Error('select 条件不能包含 _id 字段');\n\n const _db = () => (_mockDb?.(collection) || db.table(collection)) as Db<T, S>;\n const found = (await _db()\n .where(where)\n .select(select || {})\n .limit(1)\n .queryOne(true)) as DbQuery<T, S> | undefined;\n\n if (found) {\n await onBeforeUpdate?.(found);\n const updateData = isFunction(update) ? update(found) : update;\n // @ts-ignore\n const updated = await _db().whereId(found._id).update(updateData);\n onAfterUpdate?.();\n\n return updated;\n }\n\n await onBeforeCreate?.();\n const createdId = await _db().create(create);\n await onAfterCreate?.(createdId);\n\n return createdId;\n}\n\ntype _TransactionDb = {\n startTransaction: () => Promise<_Transaction>;\n};\n\ntype _Transaction = {\n commit: () => Promise<unknown>;\n rollback: () => Promise<unknown>;\n};\n\n/**\n * 在数据库事务中执行操作\n *\n * @template T - 事务操作返回值类型\n * @param transact - 事务执行函数,接收事务数据库实例作为参数\n * @param _mockDatabase - 用于测试的模拟数据库实例\n * @returns 事务操作的返回结果\n *\n * @example\n * ```typescript\n * const result = await dbTransaction(async (ta) => {\n * const user = await ta.collection('users').create({ name: 'John' });\n * const order = await ta.collection('orders').create({ userId: user.id, amount: 100 });\n * return { user, order };\n * });\n * ```\n */\n// biome-ignore lint/complexity/noBannedTypes: <explanation>\nexport async function dbTransaction<T, S extends DbSelect<T> = {}>(\n transact: (ta: Db<T, S>) => Promise<unknown>,\n _mockDatabase?: _TransactionDb,\n) {\n const db = (_mockDatabase || uniCloud.database()) as _TransactionDb;\n\n const [err1, transaction] = await tryFlatten(db.startTransaction());\n if (err1) throw err1;\n\n const ta = new Db<T, S>({ table: '', transaction });\n const [err2, result] = await tryFlatten(async () => {\n const result = await transact(ta);\n await transaction.commit();\n return result;\n });\n\n if (err2) {\n await tryFlatten(transaction.rollback());\n throw err2;\n }\n\n return result;\n}\n"],"names":["objectEach","objectOmit","errorAssign","isFunction","db","tryFlatten","result"],"mappings":";;;;;;AA4CA,MAAM,MAAM,SAAS,SAAS;AAIvB,MAAM,QAAQ,IAAI;AAKZ,MAAA,QAAQ,IAAI,QAAQ;AAiC1B,MAAM,GAAqD;AAAA,EAChE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,YAAY,SAAoB;AACzB,SAAA,QAAQ,QAAQ,iBAAiB,QAAQ,eAAe,IAAI,WAAW,QAAQ,KAAK;AACpF,SAAA,iBAAiB,CAAC,CAAC,QAAQ;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOlC,YAAY;AACV,QAAI,CAAC,KAAK,eAAsB,OAAA,IAAI,MAAM,wBAAwB;AAE3D,WAAA,KAAK,MAAM,UAAU;AAAA,EAAA;AAAA,EAG9B,YAAoC;AAAA,EACpC,cAAsC;AAAA,EAEtC,OAAO,OAAmB,MAAkB;AACtC,QAAA,KAAK,UAAW,OAAM,IAAI,MAAM,aAAa,eAAe,KAAK,SAAS,CAAC,IAAI;AAE7E,UAAA,YAAY,OAAO,KAAK,KAAK;AACnC,UAAM,YAAY,UAAU,WAAW,KAAK,UAAU,CAAC,MAAM;AAEzD,QAAA,aAAa,KAAK,WAAW;AAC/B,YAAM,IAAI,MAAM,MAAM,iBAAiB,IAAI,CAAC,0BAA0B;AAAA,IAAA;AAGxE,SAAK,YAAY;AAEjB,QAAI,WAAW;AACb,WAAK,cAAc;AAEnB,WAAK,QAAQ,KAAK,MAAM,IAAI,MAAM,GAAG;AAAA,IAAA,OAChC;AAEL,WAAK,QAAQ,KAAK,MAAM,MAAM,KAAK;AAAA,IAAA;AAG9B,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQT,MAAM,OAAmB;AAChB,WAAA,KAAK,OAAO,OAAO,OAAO;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQnC,QAAQ,IAAY;AAElB,WAAO,KAAK,OAAO,EAAE,KAAK,GAAA,GAAM,SAAS;AAAA,EAAA;AAAA,EAG3C,aAAa;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOb,OAA8B,QAAqB;AACjD,QAAI,KAAK,WAAkB,OAAA,IAAI,MAAM,sBAAsB;AAEtD,SAAA;AAEL,SAAK,QAAQ,KAAK,MAAM,MAAM,MAAM;AAE7B,WAAA;AAAA,EAAA;AAAA,EAGT,YAAY;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOZ,MAAM,OAAmB;AAClB,SAAA;AACMA,WAAAA,WAAA,OAAO,CAAC,KAAK,QAAQ;AAE9B,WAAK,QAAQ,KAAK,MAAM,QAAQ,KAAK,GAAG;AAAA,IAAA,CACzC;AAEM,WAAA;AAAA,EAAA;AAAA,EAGT,WAAW;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOX,KAAK,MAAc;AACjB,QAAI,KAAK,SAAgB,OAAA,IAAI,MAAM,oBAAoB;AAElD,SAAA;AAEL,SAAK,QAAQ,KAAK,MAAM,KAAK,IAAI;AAC1B,WAAA;AAAA,EAAA;AAAA,EAGT,YAAY;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOZ,MAAM,OAAe;AACnB,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,qBAAqB;AAEzD,QAAI,KAAK,aAAa;AACpB,YAAM,IAAI,MAAM,oBAAoB,iBAAiB,KAAK,WAAW,CAAC,SAAS;AAAA,IAAA;AAG5E,SAAA;AAEL,SAAK,QAAQ,KAAK,MAAM,MAAM,KAAK;AAC5B,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQT,MAAM,OAAO,MAAmB;AAC9B,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,4BAA4B;AAChE,QAAI,KAAK,WAAkB,OAAA,IAAI,MAAM,6BAA6B;AAClE,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,4BAA4B;AAChE,QAAI,KAAK,SAAgB,OAAA,IAAI,MAAM,2BAA2B;AAC9D,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,4BAA4B;AAEhE,UAAM,MAAM,MAAM,KAAK,MAAM,IAAI,IAAI;AACrC,UAAM,EAAE,GAAA,IAAO,oBAAoC,GAAG;AAC/C,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOT,MAAM,QAAQ;AACZ,UAAM,MAAM,MAAM,KAAK,MAAM,IAAI;AACjC,UAAM,EAAE,KAAA,IAAS,oBAA+C,GAAG;AAC5D,WAAA;AAAA,EAAA;AAAA,EAWT,MAAM,SAAS,aAAa,OAA2C;AACrE,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,8BAA8B;AAClE,QAAI,CAAC,KAAK,YAAa,MAAK,MAAM,CAAC;AAE7B,UAAA,OAAO,MAAM,KAAK,MAAM;AACxB,UAAA,MAAM,KAAK,GAAG,CAAC;AAErB,QAAI,CAAC,cAAc,CAAC,IAAW,OAAA,IAAI,MAAM,SAAS;AAC3C,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOT,MAAM,QAAQ;AACZ,QAAI,KAAK,WAAkB,OAAA,IAAI,MAAM,4BAA4B;AACjE,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,2BAA2B;AAC/D,QAAI,KAAK,SAAgB,OAAA,IAAI,MAAM,0BAA0B;AAC7D,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,2BAA2B;AAE/D,UAAM,MAAM,MAAM,KAAK,MAAM,MAAM;AACnC,UAAM,EAAE,MAAA,IAAU,oBAAuC,GAAG;AACrD,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQT,MAAM,OAAO,MAAiB;AAC5B,QAAI,CAAC,KAAK,UAAiB,OAAA,IAAI,MAAM,iCAAiC;AACtE,QAAI,KAAK,WAAkB,OAAA,IAAI,MAAM,6BAA6B;AAClE,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,4BAA4B;AAChE,QAAI,KAAK,SAAgB,OAAA,IAAI,MAAM,2BAA2B;AAC9D,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,4BAA4B;AAE5D,QAAA,KAAK,kBAAkB,CAAC,KAAK,YAAmB,OAAA,IAAI,MAAM,qCAAqC;AAEnG,UAAM,MAAM,MAAM,KAAK,MAAM,OAAO,IAAI;AACxC,UAAM,EAAE,QAAA,IAAY,oBAAyC,GAAG;AACzD,WAAA;AAAA,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOT,MAAM,SAAS;AACb,QAAI,CAAC,KAAK,UAAiB,OAAA,IAAI,MAAM,iCAAiC;AACtE,QAAI,KAAK,WAAkB,OAAA,IAAI,MAAM,6BAA6B;AAClE,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,4BAA4B;AAChE,QAAI,KAAK,SAAgB,OAAA,IAAI,MAAM,2BAA2B;AAC9D,QAAI,KAAK,UAAiB,OAAA,IAAI,MAAM,4BAA4B;AAE5D,QAAA,KAAK,kBAAkB,CAAC,KAAK,YAAmB,OAAA,IAAI,MAAM,qCAAqC;AAEnG,UAAM,MAAM,MAAM,KAAK,MAAM,OAAO;AACpC,UAAM,EAAE,QAAA,IAAY,oBAAyC,GAAG;AACzD,WAAA;AAAA,EAAA;AAEX;AAKO,MAAM,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAMhB,MAAuD,OAAe;AACpE,WAAO,IAAI,GAAS,EAAE,OAAO;AAAA,EAAA;AAEjC;AAOO,SAAS,oBAAuB,KAA6D;AAC5F,QAAA,OAAO,OAAO,KAAK,GAAgB;AAEzC,QAAM,WAAW,KAAK,WAAW,KAAK,KAAK,CAAC,MAAM;AAElD,MAAI,UAAU;AACN,UAAA,EAAE,WAAW;AACf,QAAA,CAAC,OAAO,QAAgB,QAAAC,kBAAW,QAAQ,CAAC,WAAW,UAAU,QAAQ,SAAS,CAAC;AACvF,UAAMC,MAAAA,YAAY,IAAI,MAAM,OAAO,MAAM,GAAG,MAAM;AAAA,EAAA;AAI7C,SAAA;AACT;AAEA,SAAS,eAAe,WAAuB;AACtC,SAAA,cAAc,UAAU,iBAAiB;AAClD;AAEA,SAAS,iBAAiB,WAAuB;AACxC,SAAA,cAAc,UAAU,mBAAmB;AACpD;ACrTA,eAAsB,SAMpB,SAAyC;AACnC,QAAA;AAAA,IACJ;AAAA,IACA;AAAA,IACA,SAAS,CAAC;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EAAA,IACE;AAGJ,MAAI,SAAS,OAAc,OAAA,IAAI,MAAM,sBAAsB;AAE3D,QAAM,MAAM,MAAO,UAAU,UAAU,KAAK,GAAG,MAAM,UAAU;AAC/D,QAAM,QAAS,MAAM,IAClB,EAAA,MAAM,KAAK,EACX,OAAO,UAAU,CAAA,CAAE,EACnB,MAAM,CAAC,EACP,SAAS,IAAI;AAEhB,MAAI,OAAO;AACT,UAAM,iBAAiB,KAAK;AAC5B,UAAM,aAAaC,KAAAA,WAAW,MAAM,IAAI,OAAO,KAAK,IAAI;AAElD,UAAA,UAAU,MAAM,MAAM,QAAQ,MAAM,GAAG,EAAE,OAAO,UAAU;AAChD,oBAAA;AAET,WAAA;AAAA,EAAA;AAGT,QAAM,iBAAiB;AACvB,QAAM,YAAY,MAAM,MAAM,OAAO,MAAM;AAC3C,QAAM,gBAAgB,SAAS;AAExB,SAAA;AACT;AA6BsB,eAAA,cACpB,UACA,eACA;AACMC,QAAAA,MAAM,iBAAiB,SAAS,SAAS;AAEzC,QAAA,CAAC,MAAM,WAAW,IAAI,MAAMC,KAAAA,WAAWD,IAAG,kBAAkB;AAClE,MAAI,KAAY,OAAA;AAEhB,QAAM,KAAK,IAAI,GAAS,EAAE,OAAO,IAAI,aAAa;AAClD,QAAM,CAAC,MAAM,MAAM,IAAI,MAAMC,gBAAW,YAAY;AAC5CC,UAAAA,UAAS,MAAM,SAAS,EAAE;AAChC,UAAM,YAAY,OAAO;AAClBA,WAAAA;AAAAA,EAAA,CACR;AAED,MAAI,MAAM;AACF,UAAAD,KAAA,WAAW,YAAY,UAAU;AACjC,UAAA;AAAA,EAAA;AAGD,SAAA;AACT;;;;;;;;"}