mirror of https://github.com/snail007/goproxy
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
865 B
Go
49 lines
865 B
Go
3 years ago
|
package services
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"log"
|
||
|
"runtime/debug"
|
||
|
)
|
||
|
|
||
|
type Service interface {
|
||
|
Start(args interface{}) (err error)
|
||
|
Clean()
|
||
|
}
|
||
|
type ServiceItem struct {
|
||
|
S Service
|
||
|
Args interface{}
|
||
|
Name string
|
||
|
}
|
||
|
|
||
|
var servicesMap = map[string]*ServiceItem{}
|
||
|
|
||
|
func Regist(name string, s Service, args interface{}) {
|
||
|
servicesMap[name] = &ServiceItem{
|
||
|
S: s,
|
||
|
Args: args,
|
||
|
Name: name,
|
||
|
}
|
||
|
}
|
||
|
func Run(name string) (service *ServiceItem, err error) {
|
||
|
service, ok := servicesMap[name]
|
||
|
if ok {
|
||
|
go func() {
|
||
|
defer func() {
|
||
|
err := recover()
|
||
|
if err != nil {
|
||
|
log.Fatalf("%s servcie crashed, ERR: %s\ntrace:%s", name, err, string(debug.Stack()))
|
||
|
}
|
||
|
}()
|
||
|
err := service.S.Start(service.Args)
|
||
|
if err != nil {
|
||
|
log.Fatalf("%s servcie fail, ERR: %s", name, err)
|
||
|
}
|
||
|
}()
|
||
|
}
|
||
|
if !ok {
|
||
|
err = fmt.Errorf("service %s not found", name)
|
||
|
}
|
||
|
return
|
||
|
}
|