103 lines
1.8 KiB
Go
103 lines
1.8 KiB
Go
package uzone
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"log"
|
|
|
|
"github.com/loveuer/uzone/pkg/api"
|
|
"github.com/loveuer/uzone/pkg/cache"
|
|
"github.com/loveuer/uzone/pkg/db"
|
|
"github.com/loveuer/uzone/pkg/es"
|
|
"github.com/loveuer/uzone/pkg/interfaces"
|
|
)
|
|
|
|
type module func(u *uzone)
|
|
|
|
func InitDB(uri string, models ...any) module {
|
|
db, err := db.New(uri)
|
|
if err != nil {
|
|
log.Panic(err.Error())
|
|
}
|
|
|
|
if err = db.AutoMigrate(models...); err != nil {
|
|
log.Panic(err.Error())
|
|
}
|
|
|
|
return func(u *uzone) {
|
|
u.db = db
|
|
}
|
|
}
|
|
|
|
func InitCache(uri string) module {
|
|
cache, err := cache.New(uri)
|
|
if err != nil {
|
|
log.Panic(err.Error())
|
|
}
|
|
|
|
return func(u *uzone) {
|
|
u.cache = cache
|
|
}
|
|
}
|
|
|
|
func InitES(uri string) module {
|
|
client, err := es.New(context.TODO(), uri)
|
|
if err != nil {
|
|
log.Panic(err.Error())
|
|
}
|
|
|
|
return func(u *uzone) {
|
|
u.es = client
|
|
}
|
|
}
|
|
|
|
type ApiConfig struct {
|
|
Address string
|
|
TLSConfig *tls.Config
|
|
}
|
|
|
|
func InitApi(api *api.App, cfgs ...ApiConfig) module {
|
|
cfg := ApiConfig{}
|
|
if len(cfgs) > 0 {
|
|
cfg = cfgs[0]
|
|
}
|
|
|
|
if cfg.Address == "" {
|
|
cfg.Address = "localhost:8080"
|
|
}
|
|
|
|
return func(u *uzone) {
|
|
api.Uzone = u
|
|
u.api = &uzoneApi{
|
|
engine: api,
|
|
config: cfg,
|
|
}
|
|
}
|
|
}
|
|
|
|
func InitTaskChan(ch <-chan func(uzone interfaces.Uzone) error) module {
|
|
return func(u *uzone) {
|
|
if u.taskCh == nil {
|
|
u.taskCh = make([]<-chan func(u interfaces.Uzone) error, 0)
|
|
}
|
|
|
|
u.taskCh = append(u.taskCh, ch)
|
|
}
|
|
}
|
|
|
|
// sync functions
|
|
// 添加 同步执行函数
|
|
func InitFn(fns ...func(interfaces.Uzone)) module {
|
|
return func(u *uzone) {
|
|
u.initFns._sync = append(u.initFns._sync, fns...)
|
|
}
|
|
}
|
|
|
|
// async functions
|
|
// 添加 异步执行函数
|
|
func InitAsyncFn(fns ...func(interfaces.Uzone)) module {
|
|
return func(u *uzone) {
|
|
u.initFns._async = append(u.initFns._async, fns...)
|
|
}
|
|
}
|