package util
|
|
import (
|
"basic.com/valib/logger.git"
|
"bytes"
|
"encoding/json"
|
"errors"
|
"fmt"
|
"math"
|
"os/exec"
|
"strconv"
|
"time"
|
)
|
|
// 脚本封装
|
func RunScript(str string) string {
|
cmd := exec.Command("sh", "-c", str)
|
var out bytes.Buffer
|
cmd.Stdout = &out
|
err := cmd.Run()
|
if err != nil {
|
return "运行失败"
|
}
|
return out.String()
|
}
|
|
func Sourcelist(buf []byte) (sources []map[string]interface{}, err error) {
|
var info interface{}
|
json.Unmarshal(buf, &info)
|
out, ok := info.(map[string]interface{})
|
if !ok {
|
return nil, errors.New("http response interface can not change map[string]interface{}")
|
}
|
|
middle, ok := out["hits"].(map[string]interface{})
|
if !ok {
|
return nil, errors.New("first hits change error!")
|
}
|
for _, in := range middle["hits"].([]interface{}) {
|
tmpbuf, ok := in.(map[string]interface{})
|
if !ok {
|
fmt.Println("change to source error!")
|
continue
|
}
|
source, ok := tmpbuf["_source"].(map[string]interface{})
|
if !ok {
|
fmt.Println("change _source error!")
|
continue
|
}
|
sources = append(sources, source)
|
}
|
return sources, nil
|
}
|
|
func SourceTotal(buf []byte) (total int, err error) {
|
var info interface{}
|
json.Unmarshal(buf, &info)
|
out, ok := info.(map[string]interface{})
|
if !ok {
|
return -1, errors.New("http response interface can not change map[string]interface{}")
|
}
|
|
middle, ok := out["hits"].(map[string]interface{})
|
if !ok {
|
return -1, errors.New("first total change error!")
|
}
|
|
tmp, b := middle["total"].(map[string]interface{})
|
if b != true {
|
v := middle["total"].(float64)
|
t := int(v)
|
return t, nil
|
}
|
value := tmp["value"].(float64)
|
total = int(value)
|
return total, nil
|
}
|
|
func SourceCreated(buf []byte) (result bool, err error) {
|
var info interface{}
|
json.Unmarshal(buf, &info)
|
out, ok := info.(map[string]interface{})
|
if !ok {
|
return false, errors.New("http response interface can not change map[string]interface{}")
|
}
|
|
middle, ok := out["result"].(string)
|
if !ok {
|
return false, errors.New("first total change error!")
|
}
|
if middle == "created" || middle == "updated" {
|
result = true
|
}
|
return result, nil
|
}
|
|
func SourceDeleted(buf []byte) (total int, err error) {
|
var info interface{}
|
json.Unmarshal(buf, &info)
|
out, ok := info.(map[string]interface{})
|
if !ok {
|
return -1, errors.New("http response interface can not change map[string]interface{}")
|
}
|
|
middle, ok := out["deleted"].(float64)
|
if !ok {
|
return -1, errors.New("first total change error!")
|
}
|
total = int(middle)
|
return total, nil
|
}
|
|
func SourceUpdated(buf []byte) (total int, err error) {
|
var info interface{}
|
json.Unmarshal(buf, &info)
|
out, ok := info.(map[string]interface{})
|
if !ok {
|
return -1, errors.New("http response interface can not change map[string]interface{}")
|
}
|
//fmt.Println(out)
|
middle, ok := out["updated"].(float64)
|
if !ok {
|
return -1, errors.New("first total change error!")
|
}
|
total = int(middle)
|
return total, nil
|
}
|
|
func SourceAggregationList(buf []byte) (sources []map[string]interface{}, err error) {
|
var info interface{}
|
json.Unmarshal(buf, &info)
|
out, ok := info.(map[string]interface{})
|
if !ok {
|
return nil, errors.New("http response interface can not change map[string]interface{}")
|
}
|
|
middle, ok := out["aggregations"].(map[string]interface{})
|
if !ok {
|
return nil, errors.New("first hits change error!")
|
}
|
|
documentAggregations := middle["group_by_documentnumber"].(map[string]interface{})
|
buckets := documentAggregations["buckets"].([]interface{})
|
if len(buckets) == 0 {
|
return nil, nil
|
}
|
for _, in := range buckets {
|
tmpbuf, ok := in.(map[string]interface{})
|
if !ok {
|
return nil, errors.New("")
|
}
|
sources = append(sources, tmpbuf)
|
}
|
return sources, nil
|
}
|
|
func StartTimer(f func()) {
|
for {
|
f()
|
now := time.Now()
|
// 计算下一个零点
|
next := now.Add(time.Hour * 24)
|
next = time.Date(next.Year(), next.Month(), next.Day(), 0, 0, 0, 0, next.Location())
|
t := time.NewTimer(next.Sub(now))
|
<-t.C
|
t.Stop()
|
}
|
}
|
|
// 日期换算工具
|
func GetTimeArr(earliestDate string) int {
|
now := time.Now().Format("2006-01-02")
|
fmt.Println(earliestDate, now)
|
a, _ := time.Parse("2006-01-02", earliestDate)
|
b, _ := time.Parse("2006-01-02", now)
|
d := a.Sub(b)
|
date := d.Hours() / 24
|
return int(math.Abs(date))
|
}
|
|
// 日期遍历工具
|
// GetBetweenDates 根据开始日期和结束日期计算出时间段内所有日期
|
// 参数为日期格式，如：2020-01-01
|
func GetBetweenDates(sdate, edate string) []string {
|
d := []string{}
|
timeFormatTpl := "2006-01-02"
|
if len(timeFormatTpl) != len(sdate) {
|
timeFormatTpl = timeFormatTpl[0:len(sdate)]
|
}
|
date, err := time.Parse(timeFormatTpl, sdate)
|
if err != nil {
|
// 时间解析，异常
|
return d
|
}
|
date2, err := time.Parse(timeFormatTpl, edate)
|
if err != nil {
|
// 时间解析，异常
|
return d
|
}
|
if date2.Before(date) {
|
// 如果结束时间小于开始时间，异常
|
return d
|
}
|
// 输出日期格式固定
|
timeFormatTpl = "2006-01-02"
|
date2Str := date2.Format(timeFormatTpl)
|
d = append(d, date.Format(timeFormatTpl))
|
for {
|
date = date.AddDate(0, 0, 1)
|
dateStr := date.Format(timeFormatTpl)
|
d = append(d, dateStr)
|
if dateStr == date2Str {
|
break
|
}
|
}
|
return d
|
}
|
|
func IpIntToString(ipInt int) string {
|
ipSegs := make([]string, 4)
|
var len int = len(ipSegs)
|
buffer := bytes.NewBufferString("")
|
for i := 0; i < len; i++ {
|
tempInt := ipInt & 0xFF
|
ipSegs[len-i-1] = strconv.Itoa(tempInt)
|
|
ipInt = ipInt >> 8
|
}
|
for i := 0; i < len; i++ {
|
buffer.WriteString(ipSegs[i])
|
if i < len-1 {
|
buffer.WriteString(".")
|
}
|
}
|
return buffer.String()
|
}
|
|
// 计算时间阈值
|
func CheckTimeDifference(timestampStr1 string, timestampStr2 string, intervalInMinutes int) bool {
|
layout := "2006-01-02 15:04:05"
|
timestampStr1 = timestampStr1[:19]
|
timestampStr2 = timestampStr2[:19]
|
// 将字符串解析为时间
|
time1, err := time.Parse(layout, timestampStr1)
|
if err != nil {
|
fmt.Println("时间解析失败:", err)
|
return false
|
}
|
time2, err := time.Parse(layout, timestampStr2)
|
if err != nil {
|
fmt.Println("时间解析失败:", err)
|
return false
|
}
|
|
// 计算时间差
|
diff := time2.Sub(time1)
|
|
// 检查时间差是否小于等于指定的间隔
|
if diff.Minutes() <= float64(intervalInMinutes) {
|
return true
|
} else {
|
return false
|
}
|
}
|
|
func CalculateDays(startDate string, endDate string) int {
|
layout := "2006-01-02" // 日期格式
|
startTime, err := time.Parse(layout, startDate)
|
if err != nil {
|
// 错误处理
|
logger.Error("Error parsing start date:", err)
|
return 0
|
}
|
|
endTime, err := time.Parse(layout, endDate)
|
if err != nil {
|
// 错误处理
|
logger.Error("Error parsing end date:", err)
|
return 0
|
}
|
|
duration := endTime.Sub(startTime)
|
days := int(duration.Hours() / 24) // 将时间差转换为天数
|
return days
|
}
|
