countdb.js 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. let sqlite3 = require('sqlite3').verbose();
  2. let { open } = require('sqlite');
  3. let db = null;
  4. async function openDB() {
  5. if (dataSet == {}) {
  6. console.log(dataSet);
  7. }
  8. else {
  9. console.log(dataSet[0]);
  10. };
  11. };
  12. exports.increaseCount = async function increaseCount(keys) {
  13. if (keys === {}) {
  14. return;
  15. };
  16. if (!db) {
  17. db = await open({filename: './data/codecounter.db', driver: sqlite3.Database});
  18. }
  19. let product = keys.product;
  20. let version = keys.version;
  21. let countdate = keys.countdate;
  22. let codekey = keys.codekey;
  23. let dataSet = await db.all(`SELECT * FROM counters WHERE product = '${product}' AND version = '${version}' AND countdate = '${countdate}' AND codekey = '${codekey}'`);
  24. let icount;
  25. if (dataSet.length > 0) {
  26. icount = dataSet[0].counter + 1;
  27. await db.run(`UPDATE counters SET counter = '${icount}' WHERE product = '${product}' AND version = '${version}' AND countdate = '${countdate}' AND codekey = '${codekey}'`);
  28. }
  29. else {
  30. icount = 1;
  31. await db.run(`INSERT INTO counters VALUES('${product}', '${version}', '${countdate}', '${codekey}', ${icount})`);
  32. }
  33. //db.close();
  34. }
  35. exports.queryData = async function queryData() {
  36. if (!db) {
  37. db = await open({filename: './data/codecounter.db', driver: sqlite3.Database});
  38. }
  39. let dataSet = await db.all(`SELECT * FROM counters`);
  40. return dataSet;
  41. }
  42. exports.closeDB = async function closeDB() {
  43. if (db) {
  44. await db.closeDB();
  45. }
  46. }