|
1 |
| -import * as fs from 'fs'; |
2 |
| -import * as util from 'util'; |
3 |
| -import { v4 as uuidv4 } from 'uuid'; |
4 |
| -var db = {}; |
5 |
| - |
6 |
| -const UPDATE_INTERVAL = 1000; |
7 |
| - |
8 |
| -async function saveDB() { |
9 |
| - return util.promisify(fs.writeFile)('data.json', JSON.stringify(db)); |
10 |
| -} |
11 |
| - |
12 |
| -export function loadDB() { |
13 |
| - if (fs.existsSync('data.json')) { |
14 |
| - db = JSON.parse(fs.readFileSync('data.json', 'utf8')); |
15 |
| - console.log("data.json loaded"); |
16 |
| - } else { |
17 |
| - console.log("data.json not loaded. starting with default database"); |
18 |
| - } |
19 |
| -} |
20 |
| - |
21 |
| -export function startDBWorker() { |
22 |
| - setInterval(() => { |
23 |
| - saveDB().catch(e => { |
24 |
| - console.error(e); |
25 |
| - }); |
26 |
| - }, UPDATE_INTERVAL); |
27 |
| -} |
28 |
| - |
29 |
| -export function logTransaction(type, id, amount, currentPrice) { |
30 |
| - (async () => { |
31 |
| - if (!fs.existsSync('logs')) { |
32 |
| - fs.mkdirSync('logs'); |
33 |
| - } |
34 |
| - const data = { |
35 |
| - time: Date.now(), |
36 |
| - currentPrice: currentPrice, |
37 |
| - userId: id, |
38 |
| - type: type, |
39 |
| - amount: amount |
40 |
| - }; |
41 |
| - return util.promisify(fs.writeFile)(`logs/${Date.now()}-${uuidv4()}.json`, JSON.stringify(data)); |
42 |
| - })().catch(e => {console.error(e)}); |
43 |
| -} |
44 |
| - |
45 |
| -export function existsUser(id) { |
46 |
| - return id in db; |
47 |
| -} |
48 |
| - |
49 |
| -export function getUser(id) { |
50 |
| - return db[id]; |
51 |
| -} |
52 |
| - |
53 |
| -export function setUser(id, user) { |
54 |
| - db[id] = user; |
55 |
| -} |
56 |
| - |
| 1 | +import * as fs from "fs"; |
| 2 | +import * as util from "util"; |
| 3 | +import { v4 as uuidv4 } from "uuid"; |
| 4 | +var db = {}; |
| 5 | + |
| 6 | +const UPDATE_INTERVAL = 1000; |
| 7 | + |
| 8 | +async function saveDB() { |
| 9 | + return util.promisify(fs.writeFile)("data.json", JSON.stringify(db)); |
| 10 | +} |
| 11 | + |
| 12 | +export function loadDB() { |
| 13 | + if (fs.existsSync("data.json")) { |
| 14 | + db = JSON.parse(fs.readFileSync("data.json", "utf8")); |
| 15 | + console.log("data.json loaded"); |
| 16 | + } else { |
| 17 | + console.log("data.json not loaded. starting with default database"); |
| 18 | + } |
| 19 | +} |
| 20 | + |
| 21 | +export function startDBWorker() { |
| 22 | + setInterval(() => { |
| 23 | + saveDB().catch((e) => { |
| 24 | + console.error(e); |
| 25 | + }); |
| 26 | + }, UPDATE_INTERVAL); |
| 27 | +} |
| 28 | + |
| 29 | +export function logTransaction(type, id, amount, currentPrice) { |
| 30 | + (async () => { |
| 31 | + if (!fs.existsSync("logs")) { |
| 32 | + fs.mkdirSync("logs"); |
| 33 | + } |
| 34 | + const data = { |
| 35 | + time: Date.now(), |
| 36 | + currentPrice: currentPrice, |
| 37 | + userId: id, |
| 38 | + type: type, |
| 39 | + amount: amount, |
| 40 | + }; |
| 41 | + return util.promisify(fs.writeFile)( |
| 42 | + `logs/${Date.now()}-${uuidv4()}.json`, |
| 43 | + JSON.stringify(data) |
| 44 | + ); |
| 45 | + })().catch((e) => { |
| 46 | + console.error(e); |
| 47 | + }); |
| 48 | +} |
| 49 | + |
| 50 | +export function existsUser(id) { |
| 51 | + return id in db; |
| 52 | +} |
| 53 | + |
| 54 | +export function getUser(id) { |
| 55 | + return db[id]; |
| 56 | +} |
| 57 | + |
| 58 | +export function setUser(id, user) { |
| 59 | + db[id] = user; |
| 60 | +} |
0 commit comments