You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
701 B
54 lines
701 B
package dao |
|
|
|
import ( |
|
"flag" |
|
|
|
"path/filepath" |
|
"sync" |
|
"time" |
|
|
|
"go-common/app/job/main/figure/conf" |
|
"go-common/library/log" |
|
|
|
. "github.com/smartystreets/goconvey/convey" |
|
) |
|
|
|
var ( |
|
once sync.Once |
|
d *Dao |
|
) |
|
|
|
func initConf() { |
|
if err := conf.Init(); err != nil { |
|
panic(err) |
|
} |
|
log.Init(conf.Conf.Log) |
|
defer log.Close() |
|
} |
|
|
|
func startService() { |
|
initConf() |
|
if d == nil { |
|
d = New(conf.Conf) |
|
} |
|
time.Sleep(time.Second * 2) |
|
} |
|
|
|
func CleanDB() { |
|
} |
|
|
|
func init() { |
|
dir, _ := filepath.Abs("../cmd/figure-job-test.toml") |
|
flag.Set("conf", dir) |
|
conf.Init() |
|
if d == nil { |
|
d = New(conf.Conf) |
|
} |
|
} |
|
|
|
func WithDao(f func(d *Dao)) func() { |
|
return func() { |
|
Reset(func() { CleanDB() }) |
|
f(d) |
|
} |
|
}
|
|
|