357 lines
9.9 KiB
Go
357 lines
9.9 KiB
Go
package util
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"apigo.cloud/git/apigo/plugin"
|
|
"github.com/ssgo/u"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func init() {
|
|
plugin.Register(plugin.Plugin{
|
|
Id: "util",
|
|
Name: "基础工具",
|
|
JsCode: `
|
|
util.formatDate = function (date, fmt, timezone) {
|
|
let ret
|
|
if (!date) return ''
|
|
if (!fmt) fmt = 'YYYY-mm-dd HH:MM:SS'
|
|
const opt = {
|
|
'YYYY': date.getFullYear().toString(),
|
|
'YY': date.getFullYear().toString().substring(2),
|
|
'm+': (date.getMonth() + 1).toString(),
|
|
'd+': date.getDate().toString(),
|
|
'H+': date.getHours().toString(),
|
|
'h+': (date.getHours() % 12).toString(),
|
|
'APM': (date.getHours() > 12 ? 'PM' : 'AM'),
|
|
'M+': date.getMinutes().toString(),
|
|
'S+': date.getSeconds().toString(),
|
|
'YW': 'W' + Math.ceil(((date.getTime() - new Date(date.getFullYear(), 0, 0))) / (24 * 60 * 60 * 1000) / 7),
|
|
}
|
|
for (let k in opt) {
|
|
ret = new RegExp('(' + k + ')').exec(fmt)
|
|
if (ret) {
|
|
fmt = fmt.replace(ret[1], (ret[1].length == 1) ? (opt[k]) : (opt[k].padStart(ret[1].length, '0')))
|
|
}
|
|
}
|
|
return fmt
|
|
}
|
|
|
|
util.parseDate = function (str) {
|
|
if (typeof str === 'number') str = str + ''
|
|
if (/^\d+[\\.]?\d*$/.test(str)) {
|
|
if (str.length === 10) str += '000'
|
|
return new Date(util.int(str))
|
|
}
|
|
if (str.length === 19) {
|
|
return new Date(parseInt(str.substr(0, 4)), parseInt(str.substr(5, 2)) - 1, parseInt(str.substr(8, 2)), parseInt(str.substr(11, 2)), parseInt(str.substr(14, 2)), parseInt(str.substr(17, 2)))
|
|
} else if (str.length === 10) {
|
|
return new Date(parseInt(str.substr(0, 4)), parseInt(str.substr(5, 2)) - 1, parseInt(str.substr(8, 2)))
|
|
} else if (str.length === 7) {
|
|
return new Date(parseInt(str.substr(0, 4)), parseInt(str.substr(5, 2)) - 1, 1)
|
|
} else if (str.length === 8) {
|
|
let date = new Date()
|
|
return new Date(date.getFullYear(), date.getMonth(), date.getDate(), parseInt(str.substr(0, 2)), parseInt(str.substr(3, 2)), parseInt(str.substr(6, 2)))
|
|
} else if (str.length === 5) {
|
|
let date = new Date()
|
|
return new Date(date.getFullYear(), date.getMonth(), date.getDate(), parseInt(str.substr(0, 2)), parseInt(str.substr(3, 2)), 0)
|
|
}
|
|
return new Date(str)
|
|
}
|
|
|
|
util.int = function (v) {
|
|
if (!v) return 0
|
|
if (typeof v === 'number' || v instanceof Number) return v
|
|
if (typeof v === 'object') v = v.toString ? v.toString() : ''
|
|
if (typeof v !== 'string') return 0
|
|
try {
|
|
return Math.round(parseFloat(v.replace(/,/g, '').trim())) || 0
|
|
} catch (e) {
|
|
return 0
|
|
}
|
|
}
|
|
|
|
util.float = function (v) {
|
|
if (!v) return 0.0
|
|
if (typeof v === 'number' || v instanceof Number) return v
|
|
if (typeof v === 'object') v = v.toString ? v.toString() : ''
|
|
if (typeof v !== 'string') return 0.0
|
|
try {
|
|
return parseFloat(v.replace(/,/g, '').trim()) || 0.0
|
|
} catch (e) {
|
|
return 0.0
|
|
}
|
|
}
|
|
|
|
util.str = function (v) {
|
|
if (!v) return ''
|
|
if (typeof v === 'string') return v
|
|
if (v.toString) return v.toString()
|
|
return ''
|
|
}
|
|
|
|
util.keysBy = function (obj, ...fieldAndValues) {
|
|
let keys = []
|
|
for (let k in obj) {
|
|
let match = true
|
|
if (fieldAndValues.length === 1) {
|
|
// 查找一位数组
|
|
if (obj[k] != fieldAndValues[0]) {
|
|
match = false
|
|
}
|
|
} else {
|
|
// 查找二维数组
|
|
for (let i = 0; i < fieldAndValues.length; i += 2) {
|
|
if (obj[k][fieldAndValues[i]] != fieldAndValues[i + 1]) {
|
|
match = false
|
|
break
|
|
}
|
|
}
|
|
}
|
|
if (match) {
|
|
keys.push(k)
|
|
}
|
|
}
|
|
return keys
|
|
}
|
|
|
|
util.listBy = function (obj, ...fieldAndValues) {
|
|
let list = obj instanceof Array || obj instanceof NodeList ? [] : {}
|
|
let keys = util.keysBy(obj, ...fieldAndValues)
|
|
for (let k of keys) {
|
|
if (obj instanceof Array || obj instanceof NodeList) {
|
|
list.push(obj[k])
|
|
} else {
|
|
list[k] = obj[k]
|
|
}
|
|
}
|
|
return list
|
|
}
|
|
|
|
util.hasBy = function (obj, ...fieldAndValues) {
|
|
let keys = util.keysBy(obj, ...fieldAndValues)
|
|
return keys.length > 0
|
|
}
|
|
|
|
util.getBy = function (obj, ...fieldAndValues) {
|
|
let keys = util.keysBy(obj, ...fieldAndValues)
|
|
if (keys.length > 0) return obj[keys[0]]
|
|
return null
|
|
}
|
|
|
|
util.setBy = function (obj, value, ...fieldAndValues) {
|
|
let keys = util.keysBy(obj, ...fieldAndValues)
|
|
if (keys.length > 0) obj[keys[0]] = value
|
|
}
|
|
|
|
util.indexBy = function (obj, ...fieldAndValues) {
|
|
let keys = util.keysBy(obj, ...fieldAndValues)
|
|
if (keys.length > 0) {
|
|
return obj instanceof Array || obj instanceof NodeList ? util.int(keys[0]) : keys[0]
|
|
}
|
|
return -1
|
|
}
|
|
|
|
util.removeBy = function (obj, ...fieldAndValues) {
|
|
let keys = util.keysBy(obj, ...fieldAndValues)
|
|
let n = 0
|
|
for (let i = keys.length - 1; i >= 0; i--) {
|
|
let k = keys[i]
|
|
if (obj instanceof Array || obj instanceof NodeList) {
|
|
obj.splice(k, 1)
|
|
} else {
|
|
delete obj[k]
|
|
}
|
|
n++
|
|
}
|
|
return n
|
|
}
|
|
|
|
util.removeArrayItem = function (list, item) {
|
|
let pos = list.indexOf(item)
|
|
if (pos !== -1) list.splice(pos, 1)
|
|
}
|
|
|
|
util.last = function (arr) {
|
|
if (arr && arr.length) {
|
|
return arr[arr.length - 1]
|
|
}
|
|
return null
|
|
}
|
|
|
|
util.len = function (obj) {
|
|
if (obj instanceof Array || obj instanceof NodeList) {
|
|
return obj.length
|
|
} else {
|
|
let n = 0
|
|
for (let k in obj) n++
|
|
return n
|
|
}
|
|
}
|
|
|
|
util.mergeBy = function (olds, news, ...fields) {
|
|
if (!olds) return news
|
|
for (let newItem of news) {
|
|
let fieldAndValues = []
|
|
for (let field of fields) {
|
|
fieldAndValues.push(field, newItem[field])
|
|
}
|
|
let oldIndex = util.indexBy(olds, ...fieldAndValues)
|
|
if (oldIndex === -1) {
|
|
olds.push(newItem)
|
|
} else {
|
|
olds[oldIndex] = newItem
|
|
}
|
|
}
|
|
return olds
|
|
}
|
|
|
|
util.sortBy = function (obj, field, isReverse = false, sortType) {
|
|
let list = obj instanceof Array || obj instanceof NodeList ? [] : {}
|
|
let sortedKeys = {}
|
|
let sortArr = []
|
|
for (let k in obj) {
|
|
let v = ''
|
|
if (field instanceof Array) {
|
|
for (let f of field) v += obj[k][f]
|
|
} else {
|
|
v = obj[k][field]
|
|
}
|
|
if (!sortedKeys[v]) {
|
|
sortedKeys[v] = true
|
|
sortArr.push(v)
|
|
}
|
|
}
|
|
sortArr.sort((a, b) => {
|
|
if(sortType === 'int'){
|
|
a = util.int(a)
|
|
b = util.int(b)
|
|
} else if(sortType === 'float'){
|
|
a = util.float(a)
|
|
b = util.float(b)
|
|
}
|
|
if (a == b) return 0
|
|
if (typeof a === 'number' && typeof b === 'number') {
|
|
return isReverse ? b - a : a - b
|
|
} else {
|
|
return (isReverse ? a < b : a > b) ? 1 : -1
|
|
}
|
|
})
|
|
for (let sortKey of sortArr) {
|
|
for (let k in obj) {
|
|
let v = ''
|
|
if (field instanceof Array) {
|
|
for (let f of field) v += obj[k][f]
|
|
} else {
|
|
v = obj[k][field]
|
|
}
|
|
|
|
if (obj instanceof Array || obj instanceof NodeList) {
|
|
if (v == sortKey) list.push(obj[k])
|
|
} else {
|
|
if (v == sortKey) list[k] = obj[k]
|
|
}
|
|
}
|
|
}
|
|
return list
|
|
}
|
|
|
|
util.in = function (v1, v2) {
|
|
if (!(v1 instanceof Array)) v1 = util.split(v1, /,\s*/)
|
|
return v1.indexOf(String(v2)) !== -1
|
|
}
|
|
|
|
util.uniquePush = function (arr, ...values) {
|
|
for (let v of values) {
|
|
if (arr.indexOf(v) === -1) arr.push(v)
|
|
}
|
|
}
|
|
|
|
util.clearEmpty = function (arr) {
|
|
let a = []
|
|
for (let v of arr) if (v) a.push(v)
|
|
return a
|
|
}
|
|
|
|
util.split = function (v1, v2) {
|
|
return util.clearEmpty(util.str(v1).split(v2))
|
|
}
|
|
|
|
util.join = function (arr, separator) {
|
|
return util.clearEmpty(arr).join(separator)
|
|
}
|
|
|
|
util.copy = function (obj, isDeepCopy) {
|
|
let newObj
|
|
if (obj instanceof Array || obj instanceof NodeList) {
|
|
newObj = []
|
|
for (let o of obj) {
|
|
if (isDeepCopy && typeof o === 'object' && o) o = util.copy(o)
|
|
newObj.push(o)
|
|
}
|
|
} else {
|
|
newObj = {}
|
|
for (let k in obj) {
|
|
let v = obj[k]
|
|
if (isDeepCopy && typeof v === 'object' && v) v = util.copy(v)
|
|
newObj[k] = v
|
|
}
|
|
}
|
|
|
|
return newObj
|
|
}
|
|
|
|
`,
|
|
Objects: map[string]interface{}{
|
|
// makeToken 生成指定长度的随机二进制数组
|
|
// makeToken size token长度
|
|
// makeToken return Hex编码的字符串
|
|
"makeToken": func(size int) string {
|
|
return hex.EncodeToString(u.MakeToken(size))
|
|
},
|
|
|
|
// makeTokenBytes 生成指定长度的随机二进制数组
|
|
// makeTokenBytes size token长度
|
|
// makeTokenBytes return 二进制数据
|
|
"makeTokenBytes": u.MakeToken,
|
|
|
|
// makeId 生成指定长度的随机ID
|
|
// makeId size ID长度(6~20)
|
|
// makeId return 二进制数据
|
|
"makeId": func(size int) string {
|
|
if size > 20 {
|
|
return u.UniqueId()
|
|
} else if size > 14 {
|
|
return u.UniqueId()[0:size]
|
|
} else if size > 12 {
|
|
return u.ShortUniqueId()[0:size]
|
|
} else if size > 10 {
|
|
return u.Id12()[0:size]
|
|
} else if size > 8 {
|
|
return u.Id10()[0:size]
|
|
} else if size >= 6 {
|
|
return u.Id8()[0:size]
|
|
} else {
|
|
return u.Id6()
|
|
}
|
|
},
|
|
|
|
// encodeYaml 将对象转换为YAML格式
|
|
// encodeYaml content 对象
|
|
// encodeYaml return YAML字符串
|
|
"encodeYaml": func(content interface{}) (string, error) {
|
|
buf, err := yaml.Marshal(content)
|
|
return string(buf), err
|
|
},
|
|
// decodeYaml 将YAML转化为对象
|
|
// decodeYaml content YAML字符串
|
|
// decodeYaml return 对象
|
|
"decodeYaml": func(content string) (interface{}, error) {
|
|
var data interface{}
|
|
err := yaml.Unmarshal([]byte(content), &data)
|
|
return data, err
|
|
},
|
|
},
|
|
})
|
|
}
|