87 lines
1.6 KiB
Go
87 lines
1.6 KiB
Go
package crontab
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
|
|
"474420502.top/eson/structure/priority_list"
|
|
randomdata "github.com/Pallinder/go-randomdata"
|
|
)
|
|
|
|
type randLR struct {
|
|
left, right int
|
|
}
|
|
|
|
// NodeCount 用于priority_list
|
|
type NodeCount struct {
|
|
plist.Node
|
|
randLR
|
|
}
|
|
|
|
// Compare NodeCount比较函数
|
|
func (rlr *NodeCount) Compare(v plist.INode) bool {
|
|
return rlr.GetValue().(int) > v.GetValue().(int)
|
|
}
|
|
|
|
func parseRandLR(lrvalue string) randLR {
|
|
vlen := len(lrvalue)
|
|
lastchar := lrvalue[vlen-1]
|
|
|
|
lr := strings.Split(lrvalue, "-")
|
|
switch len(lr) {
|
|
case 1:
|
|
lr := randLR{parseTimeValue(lr[0], lastchar), parseTimeValue(lr[0], lastchar)}
|
|
return lr
|
|
case 2:
|
|
|
|
lr := randLR{parseTimeValue(lr[0], lastchar), parseTimeValue(lr[1], lastchar)}
|
|
return lr
|
|
default:
|
|
panic("lr is error")
|
|
}
|
|
}
|
|
|
|
// intervalPriorityListISecond 获取优先链表比较的值
|
|
func intervalPriorityListISecond(planlist *plist.PriorityList, count int) int {
|
|
|
|
if planlist.Size() > 0 {
|
|
node := new(NodeCount)
|
|
node.SetValue(count)
|
|
|
|
iwantNode := planlist.GetCompare(node)
|
|
if iwantNode != nil {
|
|
wantNode := iwantNode.(*NodeCount)
|
|
lr := wantNode.randLR
|
|
return randomdata.Number(lr.left, lr.right+1)
|
|
}
|
|
}
|
|
|
|
return -1
|
|
}
|
|
|
|
func getInt(v string) int {
|
|
vint, err := strconv.Atoi(v)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return vint
|
|
}
|
|
|
|
func parseTimeValue(v string, lastchar byte) int {
|
|
|
|
vlen := len(v)
|
|
|
|
switch lastchar {
|
|
case 's':
|
|
return getInt(v[:vlen-1])
|
|
case 'm':
|
|
return getInt(v[:vlen-1]) * 60
|
|
case 'h':
|
|
return getInt(v[:vlen-1]) * 3600
|
|
case 'd':
|
|
return getInt(v[:vlen-1]) * 3600 * 24
|
|
default:
|
|
return getInt(v)
|
|
}
|
|
}
|