27 lines
368 B
Go
27 lines
368 B
Go
|
package dao
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"go-common/app/interface/main/shorturl/conf"
|
||
|
|
|
||
|
|
. "github.com/smartystreets/goconvey/convey"
|
||
|
|
)
|
||
|
|
|
||
|
|
var d *Dao
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
dir, _ := filepath.Abs("../cmd/shorturl-test.toml")
|
||
|
|
flag.Set("conf", dir)
|
||
|
|
conf.Init()
|
||
|
|
d = New(conf.Conf)
|
||
|
|
}
|
||
|
|
|
||
|
|
func WithDao(f func(d *Dao)) func() {
|
||
|
|
return func() {
|
||
|
|
Reset(func() {})
|
||
|
|
f(d)
|
||
|
|
}
|
||
|
|
}
|