64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
|
package weeklyhonor
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"go-common/app/job/main/creative/conf"
|
||
|
|
arcrpc "go-common/app/service/main/archive/api/gorpc"
|
||
|
|
upgrpc "go-common/app/service/main/up/api/v1"
|
||
|
|
"go-common/library/database/hbase.v2"
|
||
|
|
"go-common/library/database/sql"
|
||
|
|
binfoc "go-common/library/log/infoc"
|
||
|
|
bm "go-common/library/net/http/blademaster"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Dao is creative dao.
|
||
|
|
type Dao struct {
|
||
|
|
// config
|
||
|
|
c *conf.Config
|
||
|
|
// db
|
||
|
|
db *sql.DB
|
||
|
|
// httpClient
|
||
|
|
httpClient *bm.Client
|
||
|
|
// hbase
|
||
|
|
hbase *hbase.Client
|
||
|
|
hbaseTimeOut time.Duration
|
||
|
|
// rpc
|
||
|
|
arc *arcrpc.Service2
|
||
|
|
infoc *binfoc.Infoc
|
||
|
|
// grpc
|
||
|
|
upClient upgrpc.UpClient
|
||
|
|
}
|
||
|
|
|
||
|
|
// New init api url
|
||
|
|
func New(c *conf.Config) (d *Dao) {
|
||
|
|
d = &Dao{
|
||
|
|
c: c,
|
||
|
|
db: sql.NewMySQL(c.DB.Creative),
|
||
|
|
httpClient: bm.NewClient(c.HTTPClient.Normal),
|
||
|
|
// hbase
|
||
|
|
hbase: hbase.NewClient(c.HBaseOld.Config),
|
||
|
|
hbaseTimeOut: time.Duration(time.Millisecond * 200),
|
||
|
|
arc: arcrpc.New2(c.ArchiveRPC),
|
||
|
|
infoc: binfoc.New(c.WeeklyHonorInfoc),
|
||
|
|
}
|
||
|
|
var err error
|
||
|
|
d.upClient, err = upgrpc.NewClient(c.UpGRPCClient)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// Ping creativeDb
|
||
|
|
func (d *Dao) Ping(c context.Context) (err error) {
|
||
|
|
return d.pingMySQL(c)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close creativeDb
|
||
|
|
func (d *Dao) Close() (err error) {
|
||
|
|
_ = d.infoc.Close()
|
||
|
|
return d.db.Close()
|
||
|
|
}
|