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
|
"use strict";
const path = require("path");
const Database = require("better-sqlite3");
const db = new Database(path.resolve(__dirname, "../data/count.db"));
db.exec(`CREATE TABLE IF NOT EXISTS tb_count (
id INTEGER PRIMARY KEY AUTOINCREMENT
NOT NULL
UNIQUE,
name VARCHAR (32) NOT NULL
UNIQUE,
num BIGINT NOT NULL
DEFAULT (0)
);`);
function getNum(name) {
return new Promise((resolve, reject) => {
const stmt = db.prepare(
"SELECT `name`, `num` from tb_count WHERE `name` = ?"
);
const row = stmt.get(name);
resolve(row || { name, num: 0 });
});
}
function getAll(name) {
return new Promise((resolve, reject) => {
const stmt = db.prepare("SELECT * from tb_count");
const rows = stmt.all();
resolve(rows);
});
}
function setNum(name, num) {
return new Promise((resolve, reject) => {
db.exec(
`INSERT INTO tb_count(\`name\`, \`num\`)
VALUES($name, $num)
ON CONFLICT(name) DO
UPDATE SET \`num\` = $num;`,
{ $name: name, $num: num }
);
resolve();
});
}
function setNumMulti(counters) {
return new Promise((resolve, reject) => {
const stmt = db.prepare(`INSERT INTO tb_count(\`name\`, \`num\`)
VALUES($name, $num)
ON CONFLICT(name) DO
UPDATE SET \`num\` = $num;`);
const setMany = db.transaction((counters) => {
for (const counter of counters) stmt.run(counter);
});
setMany(counters);
resolve();
});
}
module.exports = {
getNum,
getAll,
setNum,
setNumMulti,
};
|