mirror of
https://github.com/duanhf2012/origin.git
synced 2026-02-07 01:04:41 +08:00
Compare commits
10 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8e0ed62fca | ||
|
|
7116b509e9 | ||
|
|
73d384361d | ||
|
|
ce56b19fe8 | ||
|
|
1367d776e6 | ||
|
|
987d35ff15 | ||
|
|
d225bb4bd2 | ||
|
|
ea37fb5081 | ||
|
|
0a92f48d0b | ||
|
|
f5e86fee02 |
121
README.md
121
README.md
@@ -661,51 +661,7 @@ Module1 Release.
|
||||
第四章:事件使用
|
||||
----------------
|
||||
|
||||
事件是origin中一个重要的组成部分,可以在同一个node中的service与service或者与module之间进行事件通知。系统内置的几个服务,如:TcpService/HttpService等都是通过事件功能实现。他也是一个典型的观察者设计模型。在event中有两个类型的interface,一个是event.IEventProcessor它提供注册与卸载功能,另一个是event.IEventHandler提供消息广播等功能。
|
||||
|
||||
在目录simple_event/TestService4.go中
|
||||
|
||||
```
|
||||
package simple_event
|
||||
|
||||
import (
|
||||
"github.com/duanhf2012/origin/v2/event"
|
||||
"github.com/duanhf2012/origin/v2/node"
|
||||
"github.com/duanhf2012/origin/v2/service"
|
||||
"time"
|
||||
)
|
||||
|
||||
const (
|
||||
//自定义事件类型,必需从event.Sys_Event_User_Define开始
|
||||
//event.Sys_Event_User_Define以内给系统预留
|
||||
EVENT1 event.EventType =event.Sys_Event_User_Define+1
|
||||
)
|
||||
|
||||
func init(){
|
||||
node.Setup(&TestService4{})
|
||||
}
|
||||
|
||||
type TestService4 struct {
|
||||
service.Service
|
||||
}
|
||||
|
||||
func (slf *TestService4) OnInit() error {
|
||||
//10秒后触发广播事件
|
||||
slf.AfterFunc(time.Second*10,slf.TriggerEvent)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (slf *TestService4) TriggerEvent(){
|
||||
//广播事件,传入event.Event对象,类型为EVENT1,Data可以自定义任何数据
|
||||
//这样,所有监听者都可以收到该事件
|
||||
slf.GetEventHandler().NotifyEvent(&event.Event{
|
||||
Type: EVENT1,
|
||||
Data: "event data.",
|
||||
})
|
||||
}
|
||||
|
||||
|
||||
```
|
||||
事件是origin中一个重要的组成部分,可以在服务与各module之间进行事件通知。它也是一个典型的观察者设计模型。在event中有两个类型的interface,一个是event.IEventProcessor它提供注册与卸载功能,另一个是event.IEventHandler提供消息广播等功能。
|
||||
|
||||
在目录simple_event/TestService5.go中
|
||||
|
||||
@@ -713,53 +669,68 @@ func (slf *TestService4) TriggerEvent(){
|
||||
package simple_event
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/duanhf2012/origin/v2/event"
|
||||
"github.com/duanhf2012/origin/v2/node"
|
||||
"github.com/duanhf2012/origin/v2/service"
|
||||
"fmt"
|
||||
"github.com/duanhf2012/origin/v2/event"
|
||||
"github.com/duanhf2012/origin/v2/node"
|
||||
"github.com/duanhf2012/origin/v2/service"
|
||||
"github.com/duanhf2012/origin/v2/util/timer"
|
||||
"time"
|
||||
)
|
||||
|
||||
func init(){
|
||||
node.Setup(&TestService5{})
|
||||
func init() {
|
||||
node.Setup(&TestService5{})
|
||||
}
|
||||
|
||||
const (
|
||||
//自定义事件类型,必需从event.Sys_Event_User_Define开始
|
||||
//event.Sys_Event_User_Define以内给系统预留
|
||||
EVENT1 event.EventType = event.Sys_Event_User_Define + 1
|
||||
)
|
||||
|
||||
type TestService5 struct {
|
||||
service.Service
|
||||
service.Service
|
||||
}
|
||||
|
||||
type TestModule struct {
|
||||
service.Module
|
||||
service.Module
|
||||
}
|
||||
|
||||
func (slf *TestModule) OnInit() error{
|
||||
//在当前node中查找TestService4
|
||||
pService := node.GetService("TestService4")
|
||||
func (slf *TestModule) OnInit() error {
|
||||
//在TestModule中注册监听EVENT1事件
|
||||
slf.GetEventProcessor().RegEventReceiverFunc(EVENT1, slf.GetEventHandler(), slf.OnModuleEvent)
|
||||
|
||||
//在TestModule中,往TestService4中注册EVENT1类型事件监听
|
||||
pService.(*TestService4).GetEventProcessor().RegEventReciverFunc(EVENT1,slf.GetEventHandler(),slf.OnModuleEvent)
|
||||
return nil
|
||||
return nil
|
||||
}
|
||||
|
||||
func (slf *TestModule) OnModuleEvent(ev event.IEvent){
|
||||
event := ev.(*event.Event)
|
||||
fmt.Printf("OnModuleEvent type :%d data:%+v\n",event.GetEventType(),event.Data)
|
||||
// OnModuleEvent 模块监听事件回调
|
||||
func (slf *TestModule) OnModuleEvent(ev event.IEvent) {
|
||||
event := ev.(*event.Event)
|
||||
fmt.Printf("OnModuleEvent type :%d data:%+v\n", event.GetEventType(), event.Data)
|
||||
}
|
||||
|
||||
|
||||
//服务初始化函数,在安装服务时,服务将自动调用OnInit函数
|
||||
// OnInit 服务初始化函数,在安装服务时,服务将自动调用OnInit函数
|
||||
func (slf *TestService5) OnInit() error {
|
||||
//通过服务名获取服务对象
|
||||
pService := node.GetService("TestService4")
|
||||
//在服务中注册监听EVENT1类型事件
|
||||
slf.RegEventReceiverFunc(EVENT1, slf.GetEventHandler(), slf.OnServiceEvent)
|
||||
slf.AddModule(&TestModule{})
|
||||
|
||||
////在TestModule中,往TestService4中注册EVENT1类型事件监听
|
||||
pService.(*TestService4).GetEventProcessor().RegEventReciverFunc(EVENT1,slf.GetEventHandler(),slf.OnServiceEvent)
|
||||
slf.AddModule(&TestModule{})
|
||||
return nil
|
||||
slf.AfterFunc(time.Second*10, slf.TriggerEvent)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (slf *TestService5) OnServiceEvent(ev event.IEvent){
|
||||
event := ev.(*event.Event)
|
||||
fmt.Printf("OnServiceEvent type :%d data:%+v\n",event.Type,event.Data)
|
||||
// OnServiceEvent 服务监听事件回调
|
||||
func (slf *TestService5) OnServiceEvent(ev event.IEvent) {
|
||||
event := ev.(*event.Event)
|
||||
fmt.Printf("OnServiceEvent type :%d data:%+v\n", event.Type, event.Data)
|
||||
}
|
||||
|
||||
func (slf *TestService5) TriggerEvent(t *timer.Timer) {
|
||||
//广播事件,传入event.Event对象,类型为EVENT1,Data可以自定义任何数据
|
||||
//这样,所有监听者都可以收到该事件
|
||||
slf.GetEventHandler().NotifyEvent(&event.Event{
|
||||
Type: EVENT1,
|
||||
Data: "event data.",
|
||||
})
|
||||
}
|
||||
|
||||
|
||||
@@ -768,8 +739,8 @@ func (slf *TestService5) OnServiceEvent(ev event.IEvent){
|
||||
程序运行10秒后,调用slf.TriggerEvent函数广播事件,于是在TestService5中会收到
|
||||
|
||||
```
|
||||
OnServiceEvent type :1001 data:event data.
|
||||
OnModuleEvent type :1001 data:event data.
|
||||
OnServiceEvent type :2 data:event data.
|
||||
OnModuleEvent type :2 data:event data.
|
||||
```
|
||||
|
||||
在上面的TestModule中监听的事情,当这个Module被Release时监听会自动卸载。
|
||||
|
||||
@@ -469,6 +469,11 @@ func (cls *Cluster) GetNodeInfo(nodeId string) (NodeInfo,bool) {
|
||||
func (dc *Cluster) CanDiscoveryService(fromMasterNodeId string,serviceName string) bool{
|
||||
canDiscovery := true
|
||||
|
||||
splitServiceName := strings.Split(serviceName,":")
|
||||
if len(splitServiceName) == 2 {
|
||||
serviceName = splitServiceName[0]
|
||||
}
|
||||
|
||||
for i:=0;i<len(dc.GetLocalNodeInfo().DiscoveryService);i++{
|
||||
masterNodeId := dc.GetLocalNodeInfo().DiscoveryService[i].MasterNodeId
|
||||
//无效的配置,则跳过
|
||||
|
||||
@@ -160,6 +160,12 @@ func (dc *OriginDiscoveryMaster) OnNatsDisconnect(){
|
||||
}
|
||||
|
||||
func (ds *OriginDiscoveryMaster) OnNodeConnected(nodeId string) {
|
||||
var notifyDiscover rpc.SubscribeDiscoverNotify
|
||||
notifyDiscover.IsFull = true
|
||||
notifyDiscover.NodeInfo = ds.nodeInfo
|
||||
notifyDiscover.MasterNodeId = cluster.GetLocalNodeInfo().NodeId
|
||||
|
||||
ds.GoNode(nodeId, SubServiceDiscover, ¬ifyDiscover)
|
||||
}
|
||||
|
||||
func (ds *OriginDiscoveryMaster) OnNodeDisconnect(nodeId string) {
|
||||
|
||||
@@ -10,16 +10,20 @@ import (
|
||||
"sync"
|
||||
)
|
||||
|
||||
const defaultSkip = 7
|
||||
type IOriginHandler interface {
|
||||
slog.Handler
|
||||
Lock()
|
||||
UnLock()
|
||||
SetSkip(skip int)
|
||||
GetSkip() int
|
||||
}
|
||||
|
||||
type BaseHandler struct {
|
||||
addSource bool
|
||||
w io.Writer
|
||||
locker sync.Mutex
|
||||
skip int
|
||||
}
|
||||
|
||||
type OriginTextHandler struct {
|
||||
@@ -32,6 +36,14 @@ type OriginJsonHandler struct {
|
||||
*slog.JSONHandler
|
||||
}
|
||||
|
||||
func (bh *BaseHandler) SetSkip(skip int){
|
||||
bh.skip = skip
|
||||
}
|
||||
|
||||
func (bh *BaseHandler) GetSkip() int{
|
||||
return bh.skip
|
||||
}
|
||||
|
||||
func getStrLevel(level slog.Level) string{
|
||||
switch level {
|
||||
case LevelTrace:
|
||||
@@ -78,6 +90,7 @@ func NewOriginTextHandler(level slog.Level,w io.Writer,addSource bool,replaceAtt
|
||||
ReplaceAttr: replaceAttr,
|
||||
})
|
||||
|
||||
textHandler.skip = defaultSkip
|
||||
return &textHandler
|
||||
}
|
||||
|
||||
@@ -124,6 +137,7 @@ func NewOriginJsonHandler(level slog.Level,w io.Writer,addSource bool,replaceAtt
|
||||
ReplaceAttr: replaceAttr,
|
||||
})
|
||||
|
||||
jsonHandler.skip = defaultSkip
|
||||
return &jsonHandler
|
||||
}
|
||||
|
||||
@@ -141,7 +155,7 @@ func (oh *OriginJsonHandler) Handle(context context.Context, record slog.Record)
|
||||
func (b *BaseHandler) Fill(context context.Context, record *slog.Record) {
|
||||
if b.addSource {
|
||||
var pcs [1]uintptr
|
||||
runtime.Callers(7, pcs[:])
|
||||
runtime.Callers(b.skip, pcs[:])
|
||||
record.PC = pcs[0]
|
||||
}
|
||||
}
|
||||
|
||||
@@ -239,6 +239,10 @@ func (iw *IoWriter) swichFile() error{
|
||||
return nil
|
||||
}
|
||||
|
||||
func GetDefaultHandler() IOriginHandler{
|
||||
return gLogger.(*Logger).Slogger.Handler().(IOriginHandler)
|
||||
}
|
||||
|
||||
func NewTextLogger(level slog.Level,pathName string,filePrefix string,addSource bool,logChannelCap int) (ILogger,error){
|
||||
var logger Logger
|
||||
logger.ioWriter.filePath = pathName
|
||||
|
||||
14
node/node.go
14
node/node.go
@@ -29,6 +29,7 @@ var preSetupTemplateService []func()service.IService
|
||||
var profilerInterval time.Duration
|
||||
var bValid bool
|
||||
var configDir = "./config/"
|
||||
var NodeIsRun = false
|
||||
|
||||
const(
|
||||
SingleStop syscall.Signal = 10
|
||||
@@ -57,7 +58,7 @@ func init() {
|
||||
console.RegisterCommandString("loglevel", "debug", "<-loglevel debug|release|warning|error|fatal> Set loglevel.", setLevel)
|
||||
console.RegisterCommandString("logpath", "", "<-logpath path> Set log file path.", setLogPath)
|
||||
console.RegisterCommandInt("logsize", 0, "<-logsize size> Set log size(MB).", setLogSize)
|
||||
console.RegisterCommandInt("logchannelcap", 0, "<-logchannelcap num> Set log channel cap.", setLogChannelCapNum)
|
||||
console.RegisterCommandInt("logchannelcap", -1, "<-logchannelcap num> Set log channel cap.", setLogChannelCapNum)
|
||||
console.RegisterCommandString("pprof", "", "<-pprof ip:port> Open performance analysis.", setPprof)
|
||||
}
|
||||
|
||||
@@ -354,13 +355,14 @@ func startNode(args interface{}) error {
|
||||
cluster.GetCluster().Start()
|
||||
|
||||
//6.监听程序退出信号&性能报告
|
||||
bRun := true
|
||||
|
||||
var pProfilerTicker *time.Ticker = &time.Ticker{}
|
||||
if profilerInterval > 0 {
|
||||
pProfilerTicker = time.NewTicker(profilerInterval)
|
||||
}
|
||||
|
||||
for bRun {
|
||||
NodeIsRun = true
|
||||
for NodeIsRun {
|
||||
select {
|
||||
case s := <-sig:
|
||||
signal := s.(syscall.Signal)
|
||||
@@ -368,7 +370,7 @@ func startNode(args interface{}) error {
|
||||
log.Info("receipt retire signal.")
|
||||
notifyAllServiceRetire()
|
||||
}else {
|
||||
bRun = false
|
||||
NodeIsRun = false
|
||||
log.Info("receipt stop signal.")
|
||||
}
|
||||
case <-pProfilerTicker.C:
|
||||
@@ -504,6 +506,10 @@ func setLogChannelCapNum(args interface{}) error {
|
||||
return errors.New("param logsize is error")
|
||||
}
|
||||
|
||||
if logChannelCap == -1 {
|
||||
return nil
|
||||
}
|
||||
|
||||
log.LogChannelCap = logChannelCap
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -151,28 +151,28 @@ func (s *Service) Start() {
|
||||
s.startStatus = true
|
||||
atomic.StoreInt32(&s.isRelease,0)
|
||||
var waitRun sync.WaitGroup
|
||||
log.Info(s.GetName()+" service is running",)
|
||||
s.self.(IService).OnStart()
|
||||
|
||||
for i:=int32(0);i< s.goroutineNum;i++{
|
||||
s.wg.Add(1)
|
||||
waitRun.Add(1)
|
||||
go func(){
|
||||
log.Info(s.GetName()+" service is running",)
|
||||
waitRun.Done()
|
||||
s.Run()
|
||||
s.run()
|
||||
}()
|
||||
}
|
||||
|
||||
waitRun.Wait()
|
||||
}
|
||||
|
||||
func (s *Service) Run() {
|
||||
func (s *Service) run() {
|
||||
defer s.wg.Done()
|
||||
var bStop = false
|
||||
|
||||
concurrent := s.IConcurrent.(*concurrent.Concurrent)
|
||||
concurrentCBChannel := concurrent.GetCallBackChannel()
|
||||
|
||||
s.self.(IService).OnStart()
|
||||
for{
|
||||
var analyzer *profiler.Analyzer
|
||||
select {
|
||||
|
||||
@@ -47,6 +47,10 @@ func (mm *MongoModule) Start() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (mm *MongoModule) Stop() error {
|
||||
return mm.client.Disconnect(context.Background())
|
||||
}
|
||||
|
||||
func (mm *MongoModule) TakeSession() Session {
|
||||
return Session{Client: mm.client, maxOperatorTimeOut: mm.maxOperatorTimeOut}
|
||||
}
|
||||
|
||||
@@ -1,129 +0,0 @@
|
||||
package math
|
||||
|
||||
import (
|
||||
"github.com/duanhf2012/origin/v2/log"
|
||||
)
|
||||
|
||||
type NumberType interface {
|
||||
int | int8 | int16 | int32 | int64 | float32 | float64 | uint | uint8 | uint16 | uint32 | uint64
|
||||
}
|
||||
|
||||
type SignedNumberType interface {
|
||||
int | int8 | int16 | int32 | int64 | float32 | float64
|
||||
}
|
||||
|
||||
type FloatType interface {
|
||||
float32 | float64
|
||||
}
|
||||
|
||||
func Max[NumType NumberType](number1 NumType, number2 NumType) NumType {
|
||||
if number1 > number2 {
|
||||
return number1
|
||||
}
|
||||
|
||||
return number2
|
||||
}
|
||||
|
||||
func Min[NumType NumberType](number1 NumType, number2 NumType) NumType {
|
||||
if number1 < number2 {
|
||||
return number1
|
||||
}
|
||||
|
||||
return number2
|
||||
}
|
||||
|
||||
func Abs[NumType SignedNumberType](Num NumType) NumType {
|
||||
if Num < 0 {
|
||||
return -1 * Num
|
||||
}
|
||||
|
||||
return Num
|
||||
}
|
||||
|
||||
func AddSafe[NumType NumberType](number1 NumType, number2 NumType) (NumType, bool) {
|
||||
ret := number1 + number2
|
||||
if number2 > 0 && ret < number1 {
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, false
|
||||
} else if number2 < 0 && ret > number1 {
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, false
|
||||
}
|
||||
|
||||
return ret, true
|
||||
}
|
||||
|
||||
func SubSafe[NumType NumberType](number1 NumType, number2 NumType) (NumType, bool) {
|
||||
ret := number1 - number2
|
||||
if number2 > 0 && ret > number1 {
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, false
|
||||
} else if number2 < 0 && ret < number1 {
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, false
|
||||
}
|
||||
|
||||
return ret, true
|
||||
}
|
||||
|
||||
func MulSafe[NumType NumberType](number1 NumType, number2 NumType) (NumType, bool) {
|
||||
ret := number1 * number2
|
||||
if number1 == 0 || number2 == 0 {
|
||||
return ret, true
|
||||
}
|
||||
|
||||
if ret/number2 == number1 {
|
||||
return ret, true
|
||||
}
|
||||
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, true
|
||||
}
|
||||
|
||||
func Add[NumType NumberType](number1 NumType, number2 NumType) NumType {
|
||||
ret, _ := AddSafe(number1, number2)
|
||||
return ret
|
||||
}
|
||||
|
||||
func Sub[NumType NumberType](number1 NumType, number2 NumType) NumType {
|
||||
ret, _ := SubSafe(number1, number2)
|
||||
return ret
|
||||
}
|
||||
|
||||
func Mul[NumType NumberType](number1 NumType, number2 NumType) NumType {
|
||||
ret, _ := MulSafe(number1, number2)
|
||||
return ret
|
||||
}
|
||||
|
||||
// 安全的求比例
|
||||
func PercentRateSafe[NumType NumberType, OutNumType NumberType](maxValue int64, rate NumType, numbers ...NumType) (OutNumType, bool) {
|
||||
// 比例不能为负数
|
||||
if rate < 0 {
|
||||
log.Stack("rate must not positive")
|
||||
return 0, false
|
||||
}
|
||||
|
||||
if rate == 0 {
|
||||
// 比例为0
|
||||
return 0, true
|
||||
}
|
||||
|
||||
ret := int64(rate)
|
||||
for _, number := range numbers {
|
||||
number64 := int64(number)
|
||||
result, success := MulSafe(number64, ret)
|
||||
if !success {
|
||||
// 基数*比例越界了,int64都越界了,没办法了
|
||||
return 0, false
|
||||
}
|
||||
|
||||
ret = result
|
||||
}
|
||||
|
||||
ret = ret / 10000
|
||||
if ret > maxValue {
|
||||
return 0, false
|
||||
}
|
||||
|
||||
return OutNumType(ret), true
|
||||
}
|
||||
@@ -1,91 +0,0 @@
|
||||
package rand
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"time"
|
||||
)
|
||||
|
||||
func init() {
|
||||
rand.Seed(time.Now().UnixNano())
|
||||
}
|
||||
|
||||
func RandGroup(p ...uint32) int {
|
||||
if p == nil {
|
||||
panic("args not found")
|
||||
}
|
||||
|
||||
r := make([]uint32, len(p))
|
||||
for i := 0; i < len(p); i++ {
|
||||
if i == 0 {
|
||||
r[0] = p[0]
|
||||
} else {
|
||||
r[i] = r[i-1] + p[i]
|
||||
}
|
||||
}
|
||||
|
||||
rl := r[len(r)-1]
|
||||
if rl == 0 {
|
||||
return 0
|
||||
}
|
||||
|
||||
rn := uint32(rand.Int63n(int64(rl)))
|
||||
for i := 0; i < len(r); i++ {
|
||||
if rn < r[i] {
|
||||
return i
|
||||
}
|
||||
}
|
||||
|
||||
panic("bug")
|
||||
}
|
||||
|
||||
func RandInterval(b1, b2 int32) int32 {
|
||||
if b1 == b2 {
|
||||
return b1
|
||||
}
|
||||
|
||||
min, max := int64(b1), int64(b2)
|
||||
if min > max {
|
||||
min, max = max, min
|
||||
}
|
||||
return int32(rand.Int63n(max-min+1) + min)
|
||||
}
|
||||
|
||||
func RandIntervalN(b1, b2 int32, n uint32) []int32 {
|
||||
if b1 == b2 {
|
||||
return []int32{b1}
|
||||
}
|
||||
|
||||
min, max := int64(b1), int64(b2)
|
||||
if min > max {
|
||||
min, max = max, min
|
||||
}
|
||||
l := max - min + 1
|
||||
if int64(n) > l {
|
||||
n = uint32(l)
|
||||
}
|
||||
|
||||
r := make([]int32, n)
|
||||
m := make(map[int32]int32)
|
||||
for i := uint32(0); i < n; i++ {
|
||||
v := int32(rand.Int63n(l) + min)
|
||||
|
||||
if mv, ok := m[v]; ok {
|
||||
r[i] = mv
|
||||
} else {
|
||||
r[i] = v
|
||||
}
|
||||
|
||||
lv := int32(l - 1 + min)
|
||||
if v != lv {
|
||||
if mv, ok := m[lv]; ok {
|
||||
m[v] = mv
|
||||
} else {
|
||||
m[v] = lv
|
||||
}
|
||||
}
|
||||
|
||||
l--
|
||||
}
|
||||
|
||||
return r
|
||||
}
|
||||
86
util/smath/smath.go
Normal file
86
util/smath/smath.go
Normal file
@@ -0,0 +1,86 @@
|
||||
package smath
|
||||
|
||||
import (
|
||||
"github.com/duanhf2012/origin/v2/log"
|
||||
"github.com/duanhf2012/origin/v2/util/typedef"
|
||||
)
|
||||
|
||||
func Max[NumType typedef.Number](number1 NumType, number2 NumType) NumType {
|
||||
if number1 > number2 {
|
||||
return number1
|
||||
}
|
||||
|
||||
return number2
|
||||
}
|
||||
|
||||
func Min[NumType typedef.Number](number1 NumType, number2 NumType) NumType {
|
||||
if number1 < number2 {
|
||||
return number1
|
||||
}
|
||||
|
||||
return number2
|
||||
}
|
||||
|
||||
func Abs[NumType typedef.Signed|typedef.Float](Num NumType) NumType {
|
||||
if Num < 0 {
|
||||
return -1 * Num
|
||||
}
|
||||
|
||||
return Num
|
||||
}
|
||||
|
||||
func AddSafe[NumType typedef.Number](number1 NumType, number2 NumType) (NumType, bool) {
|
||||
ret := number1 + number2
|
||||
if number2 > 0 && ret < number1 {
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, false
|
||||
} else if number2 < 0 && ret > number1 {
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, false
|
||||
}
|
||||
|
||||
return ret, true
|
||||
}
|
||||
|
||||
func SubSafe[NumType typedef.Number](number1 NumType, number2 NumType) (NumType, bool) {
|
||||
ret := number1 - number2
|
||||
if number2 > 0 && ret > number1 {
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, false
|
||||
} else if number2 < 0 && ret < number1 {
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, false
|
||||
}
|
||||
|
||||
return ret, true
|
||||
}
|
||||
|
||||
func MulSafe[NumType typedef.Number](number1 NumType, number2 NumType) (NumType, bool) {
|
||||
ret := number1 * number2
|
||||
if number1 == 0 || number2 == 0 {
|
||||
return ret, true
|
||||
}
|
||||
|
||||
if ret/number2 == number1 {
|
||||
return ret, true
|
||||
}
|
||||
|
||||
log.Stack("Calculation overflow", log.Any("number1", number1), log.Any("number2", number2))
|
||||
return ret, true
|
||||
}
|
||||
|
||||
func Add[NumType typedef.Number](number1 NumType, number2 NumType) NumType {
|
||||
ret, _ := AddSafe(number1, number2)
|
||||
return ret
|
||||
}
|
||||
|
||||
func Sub[NumType typedef.Number](number1 NumType, number2 NumType) NumType {
|
||||
ret, _ := SubSafe(number1, number2)
|
||||
return ret
|
||||
}
|
||||
|
||||
func Mul[NumType typedef.Number](number1 NumType, number2 NumType) NumType {
|
||||
ret, _ := MulSafe(number1, number2)
|
||||
return ret
|
||||
}
|
||||
|
||||
107
util/srand/slice.go
Normal file
107
util/srand/slice.go
Normal file
@@ -0,0 +1,107 @@
|
||||
package srand
|
||||
|
||||
import (
|
||||
"github.com/duanhf2012/origin/v2/util/typedef"
|
||||
"math/rand"
|
||||
"slices"
|
||||
)
|
||||
|
||||
func Sum[E ~[]T, T typedef.Number](arr E) T {
|
||||
var sum T
|
||||
for i := range arr {
|
||||
sum += arr[i]
|
||||
}
|
||||
return sum
|
||||
}
|
||||
|
||||
func SumFunc[E ~[]V, V any, T typedef.Number](arr E, getValue func(i int) T) T {
|
||||
var sum T
|
||||
for i := range arr {
|
||||
sum += getValue(i)
|
||||
}
|
||||
return sum
|
||||
}
|
||||
|
||||
func Shuffle[E ~[]T, T any](arr E) {
|
||||
rand.Shuffle(len(arr), func(i, j int) {
|
||||
arr[i], arr[j] = arr[j], arr[i]
|
||||
})
|
||||
}
|
||||
|
||||
func RandOne[E ~[]T, T any](arr E) T {
|
||||
return arr[rand.Intn(len(arr))]
|
||||
}
|
||||
|
||||
func RandN[E ~[]T, T any](arr E, num int) []T {
|
||||
index := make([]int, 0, len(arr))
|
||||
for i := range arr {
|
||||
index = append(index, i)
|
||||
}
|
||||
Shuffle(index)
|
||||
if len(index) > num {
|
||||
index = index
|
||||
}
|
||||
ret := make([]T, 0, len(index))
|
||||
for i := range index {
|
||||
ret = append(ret, arr[index[i]])
|
||||
}
|
||||
return ret
|
||||
}
|
||||
|
||||
func RandWeight[E ~[]T, T typedef.Integer](weights E) int {
|
||||
totalWeight := Sum(weights)
|
||||
if totalWeight <= 0 {
|
||||
return -1
|
||||
}
|
||||
|
||||
t := T(rand.Intn(int(totalWeight)))
|
||||
for i := range weights {
|
||||
if t < weights[i] {
|
||||
return i
|
||||
}
|
||||
t -= weights[i]
|
||||
}
|
||||
return -1
|
||||
}
|
||||
|
||||
func RandWeightFunc[E ~[]U, U any, T typedef.Integer](arr E, getWeight func(i int) T) int {
|
||||
weights := make([]T, 0, len(arr))
|
||||
for i := range arr {
|
||||
weights = append(weights, getWeight(i))
|
||||
}
|
||||
return RandWeight(weights)
|
||||
}
|
||||
|
||||
func Get[E ~[]T, T any, U typedef.Integer](arr E, index U) (ret T, ok bool) {
|
||||
if index < 0 || int(index) >= len(arr) {
|
||||
return
|
||||
}
|
||||
ret = arr[index]
|
||||
ok = true
|
||||
return
|
||||
}
|
||||
|
||||
func GetPointer[E ~[]T, T any, U typedef.Integer](arr E, index U) *T {
|
||||
if index < 0 || int(index) >= len(arr) {
|
||||
return nil
|
||||
}
|
||||
return &arr[index]
|
||||
}
|
||||
|
||||
func GetFunc[E ~[]T, T any](arr E, f func(T) bool) (ret T, ok bool) {
|
||||
index := slices.IndexFunc(arr, f)
|
||||
if index < 0 {
|
||||
return
|
||||
}
|
||||
|
||||
return arr[index], true
|
||||
}
|
||||
|
||||
func GetPointerFunc[E ~[]T, T any](arr E, f func(T) bool) *T {
|
||||
index := slices.IndexFunc(arr, f)
|
||||
if index < 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
return &arr[index]
|
||||
}
|
||||
25
util/typedef/type.go
Normal file
25
util/typedef/type.go
Normal file
@@ -0,0 +1,25 @@
|
||||
package typedef
|
||||
|
||||
type Signed interface {
|
||||
~int | ~int8 | ~int16 | ~int32 | ~int64
|
||||
}
|
||||
|
||||
type Unsigned interface {
|
||||
~uint | ~uint8 | ~uint16 | ~uint32 | ~uint64
|
||||
}
|
||||
|
||||
type Float interface {
|
||||
~float32 | ~float64
|
||||
}
|
||||
|
||||
type Integer interface {
|
||||
Signed|Unsigned
|
||||
}
|
||||
|
||||
type Number interface {
|
||||
Signed|Unsigned|Float
|
||||
}
|
||||
|
||||
type Ordered interface {
|
||||
Number|Float|~string
|
||||
}
|
||||
Reference in New Issue
Block a user