1
0
Code Issues Pull Requests Projects Releases Wiki Activity GitHub Gitee

add utils

This commit is contained in:
2022-09-30 00:51:08 +08:00
parent 06eeb65eac
commit 5b31b4bf98
10 changed files with 432 additions and 2 deletions

31
utils/dateTimeUtils.js Normal file
View 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
View 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
View 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
View 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
View 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
View File

@@ -0,0 +1,7 @@
async function sleep(ms) {
await new Promise(function (resolve, reject) {
setTimeout(resolve, ms);
});
}
module.exports = sleep;