add utils
This commit is contained in:
31
utils/dateTimeUtils.js
Normal file
31
utils/dateTimeUtils.js
Normal file
@@ -0,0 +1,31 @@
|
||||
/**
|
||||
* refer: https://www.npmjs.com/package/solarlunar
|
||||
*/
|
||||
const solarLunar = require('solarlunar');
|
||||
|
||||
|
||||
// 获取 今日 到 阳历指定日期 的间隔
|
||||
function getSpanDays_solar(solarYear, solarMonth, solarDay) {
|
||||
let nowDate_00_00 = new Date(new Date().toLocaleDateString()); /* 当日0点 */
|
||||
|
||||
let festivalDayDate = new Date(`${solarYear}-${solarMonth}-${solarDay} 00:00:00`);
|
||||
let festivalSpanDays = ((festivalDayDate - nowDate_00_00) / (24 * 3600 * 1000));
|
||||
// console.log(festivalSpanDays, nowDate_00_00, festivalDayDate);
|
||||
return festivalSpanDays;
|
||||
}
|
||||
|
||||
// 获取 今日 到 农历指定日期 的间隔
|
||||
function getSpanDays_lunar(lunarYear, lunarMonth, lunarDay) {
|
||||
let nowDate_00_00 = new Date(new Date().toLocaleDateString()); /* 当日0点 */
|
||||
|
||||
let festivalDay = solarLunar.lunar2solar(lunarYear, lunarMonth, lunarDay); // 农历
|
||||
let festivalDayDate = new Date(`${festivalDay.cYear}-${festivalDay.cMonth}-${festivalDay.cDay} 00:00:00`); // 2022-09-09T16:00:00.000Z
|
||||
let festivalSpanDays = ((festivalDayDate - nowDate_00_00) / (24 * 3600 * 1000));
|
||||
// console.log(festivalSpanDays, nowDate_00_00, festivalDayDate);
|
||||
return festivalSpanDays;
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
getSpanDays_solar,
|
||||
getSpanDays_lunar
|
||||
}
|
85
utils/dbPoolUtils.js
Normal file
85
utils/dbPoolUtils.js
Normal file
@@ -0,0 +1,85 @@
|
||||
const mysql = require('mysql');
|
||||
const fs = require('fs');
|
||||
const path = require('path');
|
||||
|
||||
let globalConfig = JSON.parse(fs.readFileSync(path.join(__dirname, '../config.json'), 'utf8'));
|
||||
let pool = null;
|
||||
|
||||
function createPool(database) {
|
||||
let config = {
|
||||
connectionLimit: 10, //连接数量,默认是10
|
||||
...globalConfig.mysql,
|
||||
database: database,
|
||||
};
|
||||
console.log(config);
|
||||
|
||||
//创建数据库连接池
|
||||
pool = mysql.createPool(config);
|
||||
|
||||
// //从连接池中获取连接时,将触发该事件
|
||||
// pool.on('acquire', function (conn) {
|
||||
// console.log('获取连接', conn.threadId);
|
||||
// });
|
||||
|
||||
// //在连接池中建立新连接时,将触发该事件
|
||||
// pool.on('connection', function (conn) {
|
||||
// console.log('建立新连接', conn.threadId);
|
||||
// });
|
||||
|
||||
// //等待可用连接时,将触发该事件
|
||||
// pool.on('enqueue', function () {
|
||||
// console.log('等待可用连接');
|
||||
// });
|
||||
|
||||
// //当连接释放回池中时,触发该事件
|
||||
// pool.on('release', function (conn) {
|
||||
// console.log('连接被释放回池中', conn.threadId);
|
||||
// });
|
||||
}
|
||||
|
||||
async function query(sql) {
|
||||
if (!pool) {
|
||||
console.error('Database connection pool is not initialized yet.');
|
||||
return;
|
||||
}
|
||||
return await new Promise(function (resolve, reject) {
|
||||
//pool.query()方法可以自动的帮我们在连接池中获取可用连接
|
||||
pool.query(sql, function (err, data) {
|
||||
if (err) reject(err);
|
||||
// console.log(data);
|
||||
resolve(data);
|
||||
});
|
||||
|
||||
// //当然我们也可以手动获取可用连接
|
||||
// pool.getConnection(function (err, conn) {
|
||||
// if (err) {
|
||||
// throw err;
|
||||
// }
|
||||
// conn.query('select * from `order`', function (err, data) {
|
||||
// if (err) {
|
||||
// throw err;
|
||||
// }
|
||||
// data.forEach(function (value) {
|
||||
// console.log(value.id, value.order_id, value.user_id);
|
||||
// });
|
||||
|
||||
// //连接用完之后,需要释放,重新放回连接池中。
|
||||
// //注意这里并没有销毁该连接,该连接仍然可用,但需要重新获取
|
||||
// conn.release();
|
||||
// });
|
||||
// });
|
||||
});
|
||||
}
|
||||
|
||||
function close() {
|
||||
if (!pool) return;
|
||||
pool.end(function (err) {
|
||||
console.log('关闭连接池', err);
|
||||
});
|
||||
pool = null;
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
query: query,
|
||||
close: close,
|
||||
}
|
31
utils/dbUtils.js
Normal file
31
utils/dbUtils.js
Normal file
@@ -0,0 +1,31 @@
|
||||
//引入MySQL
|
||||
const mysql = require('mysql');
|
||||
const fs = require('fs');
|
||||
const path = require('path');
|
||||
|
||||
let globalConfig = JSON.parse(fs.readFileSync(path.join(__dirname, '../config.json'), 'utf8'));
|
||||
let config = {
|
||||
...globalConfig.mysql,
|
||||
};
|
||||
console.log(config);
|
||||
|
||||
async function query(sql) {
|
||||
//通过MySQL中方法创建链接对象
|
||||
var connection = mysql.createConnection(config);
|
||||
//开始连接
|
||||
connection.connect();
|
||||
//执行SQL语句 (添加、删除、更新、查询)
|
||||
connection.query(sql, (err, result) => {
|
||||
if (err) {
|
||||
console.error('err', err);
|
||||
return;
|
||||
}
|
||||
console.log('result', result);
|
||||
})
|
||||
//最后需要关闭连接
|
||||
connection.end();
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
query: query,
|
||||
}
|
49
utils/fileUtils.js
Normal file
49
utils/fileUtils.js
Normal file
@@ -0,0 +1,49 @@
|
||||
const fs = require('fs');
|
||||
const path = require('path');
|
||||
|
||||
// 创建目录
|
||||
async function createFolder(folderToCreate) {
|
||||
let currentFolder = path.join(folderToCreate);
|
||||
let parentFolder = path.join(currentFolder, '../');
|
||||
// console.log({ currentFolder: currentFolder, parentFolder: parentFolder });
|
||||
if (!fs.existsSync(currentFolder)) {
|
||||
// 文件夹不存在,创建文件夹
|
||||
createFolder(parentFolder); // 保证父级文件夹存在
|
||||
fs.mkdirSync(currentFolder); // 创建当前级文件夹
|
||||
} else {
|
||||
// 否则就什么也不做
|
||||
}
|
||||
}
|
||||
|
||||
// 保存 JSON
|
||||
function saveJSON({ saveFolder, now, fileNameSuffix, object, compress = true, uncompress = true }) {
|
||||
if (LATEST_DATA_ONLY) return;
|
||||
|
||||
let year = now.substring(0, 4);
|
||||
let month = now.substring(5, 7);
|
||||
let day = now.substring(8, 10);
|
||||
let hour = now.substring(11, 13);
|
||||
let minute = now.substring(14, 16);
|
||||
// console.log(now);
|
||||
// console.log( "year, month, day, hour, minute: " + year + ", " + month + ", " + day + ", " + hour + ", " + minute);
|
||||
|
||||
// 创建当前文件夹
|
||||
let folder = `${saveFolder}/${fileNameSuffix}/${year}/${month}/${day}`;
|
||||
createFolder(folder);
|
||||
let fileName = `${folder}/${year}${month}${day}_${hour}${minute}`;
|
||||
|
||||
// 生成文件名
|
||||
// '2022-07-23T10:11:38.650Z' => '20220723_1011'
|
||||
// let fileName = now.replace(/T/, '_').replace(/:\d{2}.\d{3}Z/, '').replace(/[-:]/g, '');
|
||||
// console.log(`fileName is ${fileName}`);
|
||||
|
||||
if (compress)
|
||||
fs.writeFileSync(`${fileName}.min.json`, JSON.stringify(object));
|
||||
if (uncompress)
|
||||
fs.writeFileSync(`${fileName}.json`, JSON.stringify(object, "", "\t"));
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
createFolder,
|
||||
saveJSON,
|
||||
}
|
42
utils/requestUtils.js
Normal file
42
utils/requestUtils.js
Normal file
@@ -0,0 +1,42 @@
|
||||
const request = require('request');
|
||||
|
||||
async function get(url) {
|
||||
return await query({
|
||||
method: 'GET',
|
||||
url: url,
|
||||
json: true,
|
||||
});
|
||||
}
|
||||
|
||||
async function getApiResult(url, params = null) {
|
||||
return await query({
|
||||
method: 'GET',
|
||||
url: url,
|
||||
...params ?? {}
|
||||
});
|
||||
}
|
||||
|
||||
async function query(opts) {
|
||||
var return_data = await new Promise((resolve, reject) => {
|
||||
// console.log("opts", opts);
|
||||
request(opts, (error, response, result) => {
|
||||
// console.log("error, result", error, result);
|
||||
if (!error && (response.statusCode == 200)) {
|
||||
// 请求成功
|
||||
resolve(result);
|
||||
} else {
|
||||
// 请求失败
|
||||
console.error(`error is ${error}`);
|
||||
reject(error);
|
||||
}
|
||||
});
|
||||
});
|
||||
// console.log(`return_data is ${JSON.stringify(return_data)}`);
|
||||
return return_data;
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
get: get,
|
||||
getApiResult: getApiResult,
|
||||
query: query,
|
||||
}
|
7
utils/sleepUtils.js
Normal file
7
utils/sleepUtils.js
Normal file
@@ -0,0 +1,7 @@
|
||||
async function sleep(ms) {
|
||||
await new Promise(function (resolve, reject) {
|
||||
setTimeout(resolve, ms);
|
||||
});
|
||||
}
|
||||
|
||||
module.exports = sleep;
|
Reference in New Issue
Block a user