-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdb_helper.js
91 lines (83 loc) · 2.78 KB
/
db_helper.js
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
const sqlite3 = require('sqlite3').verbose();
(function () {
module.exports.openConn = (databaseName) => {
var db = new sqlite3.Database(`${databaseName}.db`, sqlite3.OPEN_CREATE | sqlite3.OPEN_READWRITE, (err) => {
if (err) {
console.error(err.message);
} else {
console.log(`Connected to the ${databaseName} database.`);
}
});
return db;
};
module.exports.createTable = (db, tableName, columns) => {
var sql = `CREATE TABLE IF NOT EXISTS ${tableName} (${columns})`;
db.run(sql, [], function(err) {
if (err) {
return console.log(err.message);
}
// get the last insert id
console.log(`Created table`);
});
}
module.exports.insertInto = (db, tableName, columns, values) => {
var sql = `INSERT INTO ${tableName}(${columns}) VALUES(${values});`;
db.run(sql, [], function(err) {
if (err) {
return console.log(err.message);
}
// get the last insert id
console.log(`A row has been inserted with rowid ${this.lastID}`);
});
}
module.exports.selectAll = (db, tableName, columns, callback) => {
var sql = `SELECT ${columns} FROM ${tableName}`;
db.all(sql, [], (err, rows) => {
if (err) {
console.error(err);
} else {
var data = [];
rows.forEach(row => {
data.push(row);
});
var m = {
'message': 'selectResult',
'table': tableName,
'data': data
}
callback(m);
}
});
}
module.exports.select = (db, tableName, columns, condition, callback) => {
var ret, sql = `SELECT ${columns} FROM ${tableName} WHERE ${condition}`;
db.get(sql, [], (err, row) => {
if (err) {
console.error(err);
} else {
var data = [];
data.push(row);
var m = {
'message': 'selectResult',
'table': tableName,
'data': data
}
callback(m);
}
});
return ret;
}
module.exports.update = (db, tableName, updates, condition) => {
var sql = `UPDATE ${tableName} SET ${updates} WHERE ${condition}`;
db.run(sql);
}
module.exports.closeConn = (db) => {
db.close((err) => {
if (err) {
console.error(err.message);
} else {
console.log('Close the database connection.');
}
});
};
}());