35 lines
581 B
Go
35 lines
581 B
Go
package watermark
|
|
|
|
import (
|
|
"flag"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"go-common/app/interface/main/creative/conf"
|
|
"go-common/app/interface/main/creative/service"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
)
|
|
|
|
var (
|
|
s *Service
|
|
p *service.Public
|
|
)
|
|
|
|
func init() {
|
|
dir, _ := filepath.Abs("../../cmd/creative.toml")
|
|
flag.Set("conf", dir)
|
|
conf.Init()
|
|
rpcdaos := service.NewRPCDaos(conf.Conf)
|
|
p = service.New(conf.Conf, rpcdaos)
|
|
s = New(conf.Conf, rpcdaos, p)
|
|
time.Sleep(time.Second)
|
|
}
|
|
|
|
func WithService(f func(s *Service)) func() {
|
|
return func() {
|
|
Reset(func() {})
|
|
f(s)
|
|
}
|
|
}
|