71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
package services
|
|
|
|
import (
|
|
"fmt"
|
|
logger "log"
|
|
"runtime/debug"
|
|
"sync"
|
|
)
|
|
|
|
type Service interface {
|
|
Start(args interface{}, log *logger.Logger) (err error)
|
|
Clean()
|
|
}
|
|
type ServiceItem struct {
|
|
S Service
|
|
Args interface{}
|
|
Name string
|
|
Log *logger.Logger
|
|
}
|
|
|
|
var servicesMap = sync.Map{}
|
|
|
|
func Regist(name string, s Service, args interface{}, log *logger.Logger) {
|
|
Stop(name)
|
|
servicesMap.Store(name, &ServiceItem{
|
|
S: s,
|
|
Args: args,
|
|
Name: name,
|
|
Log: log,
|
|
})
|
|
}
|
|
func GetService(name string) *ServiceItem {
|
|
if s, ok := servicesMap.Load(name); ok && s.(*ServiceItem).S != nil {
|
|
return s.(*ServiceItem)
|
|
}
|
|
return nil
|
|
|
|
}
|
|
func Stop(name string) {
|
|
if s, ok := servicesMap.Load(name); ok && s.(*ServiceItem).S != nil {
|
|
s.(*ServiceItem).S.Clean()
|
|
*s.(*ServiceItem) = ServiceItem{}
|
|
s = nil
|
|
servicesMap.Store(name, nil)
|
|
servicesMap.Delete(name)
|
|
}
|
|
}
|
|
func Run(name string, args interface{}) (service *ServiceItem, err error) {
|
|
_service, ok := servicesMap.Load(name)
|
|
if ok {
|
|
defer func() {
|
|
e := recover()
|
|
if e != nil {
|
|
err = fmt.Errorf("%s servcie crashed, ERR: %s\ntrace:%s", name, e, string(debug.Stack()))
|
|
}
|
|
}()
|
|
service = _service.(*ServiceItem)
|
|
if args != nil {
|
|
err = service.S.Start(args, service.Log)
|
|
} else {
|
|
err = service.S.Start(service.Args, service.Log)
|
|
}
|
|
if err != nil {
|
|
err = fmt.Errorf("%s servcie fail, ERR: %s", name, err)
|
|
}
|
|
} else {
|
|
err = fmt.Errorf("service %s not found", name)
|
|
}
|
|
return
|
|
}
|