Spaces:
Runtime error
Runtime error
File size: 1,605 Bytes
07f7653 |
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 |
'use strict'
const path = require('path')
const Database = require('better-sqlite3')
const db = new Database(path.resolve(__dirname, '../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
}
|