优化sdk,支持并发启动/关闭操作
This commit is contained in:
@ -4,6 +4,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
logger "log"
|
logger "log"
|
||||||
"runtime/debug"
|
"runtime/debug"
|
||||||
|
"sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Service interface {
|
type Service interface {
|
||||||
@ -17,31 +18,31 @@ type ServiceItem struct {
|
|||||||
Log *logger.Logger
|
Log *logger.Logger
|
||||||
}
|
}
|
||||||
|
|
||||||
var servicesMap = map[string]*ServiceItem{}
|
var servicesMap = sync.Map{}
|
||||||
|
|
||||||
func Regist(name string, s Service, args interface{}, log *logger.Logger) {
|
func Regist(name string, s Service, args interface{}, log *logger.Logger) {
|
||||||
Stop(name)
|
Stop(name)
|
||||||
servicesMap[name] = &ServiceItem{
|
servicesMap.Store(name, &ServiceItem{
|
||||||
S: s,
|
S: s,
|
||||||
Args: args,
|
Args: args,
|
||||||
Name: name,
|
Name: name,
|
||||||
Log: log,
|
Log: log,
|
||||||
}
|
})
|
||||||
}
|
}
|
||||||
func GetService(name string) *ServiceItem {
|
func GetService(name string) *ServiceItem {
|
||||||
if s, ok := servicesMap[name]; ok && s.S != nil {
|
if s, ok := servicesMap.Load(name); ok && s.(*ServiceItem).S != nil {
|
||||||
return s
|
return s.(*ServiceItem)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
|
||||||
}
|
}
|
||||||
func Stop(name string) {
|
func Stop(name string) {
|
||||||
if s, ok := servicesMap[name]; ok && s.S != nil {
|
if s, ok := servicesMap.Load(name); ok && s.(*ServiceItem).S != nil {
|
||||||
s.S.Clean()
|
s.(*ServiceItem).S.Clean()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
func Run(name string, args interface{}) (service *ServiceItem, err error) {
|
func Run(name string, args interface{}) (service *ServiceItem, err error) {
|
||||||
service, ok := servicesMap[name]
|
_service, ok := servicesMap.Load(name)
|
||||||
if ok {
|
if ok {
|
||||||
defer func() {
|
defer func() {
|
||||||
e := recover()
|
e := recover()
|
||||||
@ -49,6 +50,7 @@ func Run(name string, args interface{}) (service *ServiceItem, err error) {
|
|||||||
err = fmt.Errorf("%s servcie crashed, ERR: %s\ntrace:%s", name, e, string(debug.Stack()))
|
err = fmt.Errorf("%s servcie crashed, ERR: %s\ntrace:%s", name, e, string(debug.Stack()))
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
service = _service.(*ServiceItem)
|
||||||
if args != nil {
|
if args != nil {
|
||||||
err = service.S.Start(args, service.Log)
|
err = service.S.Start(args, service.Log)
|
||||||
} else {
|
} else {
|
||||||
|
|||||||
Reference in New Issue
Block a user