nodejs常用代码片段
2018年5月24日大约 2 分钟
自动创建目录(多级)
相比起使用递归创建,调用 sheljsl 模块简单得多
const shell = require('shelljs')
const fs = require('fs')
if (!fs.existsSync(dir)) {
shell.mkdir('-p', dir)
}
自动创建目录函数
/**
* 创建文件,自动创建目录
*/
function createFile(newPath, txt) {
var dir = require('path').dirname(newPath)
if (!require('fs').existsSync(dir)) {
require('shelljs').mkdir('-p', dir)
}
require('fs').writeFileSync(newPath, txt)
}
ejs 自定义函数的使用
版本:2.6.1
安装:npm install ejs
在模板中使用自定义函数处理数据的Demo
const ejs = require('ejs')
ejs.delimiter = '%'
// 自定义函数 返回首字母大写的字符串
const def_func={
capitalize(str) {
return str ? (str.substring(0, 1).toUpperCase() + str.substring(1)) : str
}
}
const config={
title:'me is test'
}
var result=ejs.render('<%= capitalize(title) %>', Object.assign(def_func,config)) //Test
//const path='./template/test.txt'
//ejs.renderFile(path, Object.assign(def_func, config), {},function(err,str){
// //str 渲染结果
//})
获取指定目录下所有指定格式文件
readAllFile(root,reg) 参数传递正则
获取所有js文件:
readAllFile('D:/..',/\.js$/)
获取所有json文件:
readAllFile('D:/..',/\.json$/)
/*
* 读取指定文件夹下的全部文件,可通过正则进行过滤,返回文件路径数组
* @param root 指定文件夹路径
* @param reg 对文件的过滤正则表达式,可选参数,示例: 获取指定目录下所有js文件:/\.js$/
*
*/
function readAllFile(root, reg) {
var resultArr = []
return (function fn(root, reg) {
var fs = require('fs')
if (fs.existsSync(root)) {
//文件或文件夹存在
var stat = fs.lstatSync(root) // 对于不存在的文件或文件夹,此函数会报错
if (stat.isDirectory()) {
// 文件夹
var files = fs.readdirSync(root)
files.forEach(function(file) {
var t = fn(root + '/' + file, reg)
resultArr = resultArr.concat(t)
})
} else {
if (reg !== undefined) {
if (typeof reg.test == 'function' && reg.test(root)) {
resultArr.push(root)
}
} else {
resultArr.push(root)
}
}
}
return resultArr
})()
}
nodejs 获取传入参数的封装
传入需要获取的 keys及前缀(可选,默认--)
调用:node index.js --target test
接收:const config=loadConifg(['target'],'--')
//config.target----->test
/**
* 根据参数数组加载参数
* @param {Array} params 键值
* @param {String} prefix 前缀 默认为 --
*/
function loadConfig(params, prefix) {
if (typeof params !== 'object' || params.length === 0) return {}
var config = {}
prefix = prefix || '--'
var args = process.argv
for (let i = 0; i < args.length; i++) {
var key = args[i].indexOf(prefix) === 0 ? args[i].replace(prefix, '') : ''
if (key && params.indexOf(key) > -1 && i < args.length - 1) {
config[key] = args[i + 1]
}
}
return config
}
递归根据最后一级ID找父节点属性
方法定义
/**
* 查找父节点
* @param {String} key 对象匹配键值
* @param {String} value 对象匹配值
* @param {Array} items 带children节点的数组
* @param {Array} resultArr 父对象结果集
*/
function fn(key, value, items, resultArr) {
var checkResult = false
for (let index = 0; index < items.length; index++) {
const e = items[index]
checkResult =
e[key] === value || (e.children && fn(key, value, e.children, resultArr))
if (checkResult) {
resultArr.push(e)
break
}
}
return checkResult
}
调用
var arr = []
fn(
'value',
2,
[
{
text: 't1',
value: 1,
children: [
{
text: 't2',
value: 2
}
]
}
],
arr
)
console.log(arr)