forked from hellosean1025/http-mysql-server
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathindex.js
More file actions
103 lines (89 loc) · 2.33 KB
/
index.js
File metadata and controls
103 lines (89 loc) · 2.33 KB
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
92
93
94
95
96
97
98
99
100
101
102
103
const koa = require("koa");
const koaBody = require("koa-body");
const mysql = require("mysql");
const argv = require("yargs").argv;
module.exports = function(options) {
function query(pool, sql) {
return new Promise((resolve, reject) => {
pool.getConnection(function(err, connection) {
if(err){
err.message += '数据库连接失败'
return reject(err)
}
connection.query(sql, function(error, results, fields) {
connection.release();
if (error) reject(error);
resolve({
results,
fields
});
});
});
});
}
/**
* config Object
* host: "localhost",
* user: "root",
* password: "12345678",
* database: "test"
*
* 详细 config 配置查看 https://github.com/mysqljs/mysql#connection-options
*/
config = {
connectionLimit: 10,
host: options.host || 'localhost',
port: options.port || 3306,
user: options.user || 'root',
database: options.database || 'test',
password: (options.password + '') || 'root'
}
const port = options.http_port || "5555";
const pool = mysql.createPool(config);
const app = new koa();
app.use(
koaBody({
multipart: true
})
);
app.use((ctx, next)=>{
return next()
})
app.use(async (ctx, next) => {
if (ctx.path.indexOf("/api/query") === 0) {
let params = ctx.request.body;
let sql = params.sql;
try {
let data = await query(pool, sql);
return (ctx.body = data.results);
} catch (err) {
return (ctx.body = {
errcode: 400,
message: err.message
});
}
}
return await next();
});
app.use(async ctx => {
let result
try{
result = await query(pool, "show tables");
}catch(err){
return ctx.body = err.message
}
let tables = result.results.map(item => item.Tables_in_test);
let tableHtml = [];
tables.forEach((item, index) =>
tableHtml.push(
"<tr><td> " + (index + 1) + " </td><td>" + item + "</td></tr>"
)
);
ctx.body = require('./view')(tableHtml, port);
});
app.listen(port, function() {
process.stdout.write(
"服务启动成功, 请访问 “http://127.0.0.1:" + port + '"\n'
);
});
};