优化sdk,支持并发启动/关闭操作

This commit is contained in:
arraykeys@gmail.com
2018-08-24 11:40:36 +08:00
parent c9f8f6c6fb
commit abdcae31c9

View File

@ -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 {