go_dreamfactory/comm/pool.go
2023-03-10 15:50:10 +08:00

25 lines
400 B
Go

package comm
import (
"go_dreamfactory/pb"
"sync"
)
var taskParamPool = &sync.Pool{
New: func() interface{} {
return &pb.AgentMessage{}
},
}
func GettaskParam(t TaskType, p ...int32) *TaskParam {
task := taskParamPool.Get().(*TaskParam)
task.TT = t
task.Params = append(task.Params, p...)
return task
}
func PuttaskParam(r *TaskParam) {
r.Params = r.Params[:0]
taskParamPool.Put(r)
}