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.
96 lines
2.5 KiB
96 lines
2.5 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
relation "go-common/app/service/main/relation/api" |
|
"time" |
|
|
|
recallv1 "go-common/app/service/bbq/recsys-recall/api/grpc/v1" |
|
"go-common/app/service/bbq/recsys/conf" |
|
searchv1 "go-common/app/service/bbq/search/api/grpc/v1" |
|
user "go-common/app/service/bbq/user/api" |
|
"go-common/library/cache/memcache" |
|
"go-common/library/cache/redis" |
|
xsql "go-common/library/database/sql" |
|
"go-common/library/net/rpc/warden" |
|
|
|
"github.com/Dai0522/workpool" |
|
) |
|
|
|
// Dao dao |
|
type Dao struct { |
|
c *conf.Config |
|
mc *memcache.Pool |
|
redis *redis.Pool |
|
bfRedis *redis.Pool |
|
db *xsql.DB |
|
wp *workpool.Pool |
|
SearchClient searchv1.SearchClient |
|
RecallClient recallv1.RecsysRecallClient |
|
UserClient user.UserClient |
|
RelationClient relation.RelationClient |
|
} |
|
|
|
// New init mysql db |
|
func New(c *conf.Config) (dao *Dao) { |
|
wpConf := &workpool.PoolConfig{ |
|
MaxWorkers: c.WorkerPool.MaxWorkers, |
|
MaxIdleWorkers: c.WorkerPool.MaxIdleWorkers, |
|
MinIdleWorkers: c.WorkerPool.MinIdleWorkers, |
|
KeepAlive: time.Duration(c.WorkerPool.KeepAlive), |
|
} |
|
wp, err := workpool.NewWorkerPool(1024, wpConf) |
|
if err != nil { |
|
panic(err) |
|
} |
|
wp.Start() |
|
dao = &Dao{ |
|
c: c, |
|
redis: redis.NewPool(c.Redis), |
|
bfRedis: redis.NewPool(c.BFRedis), |
|
db: xsql.NewMySQL(c.MySQL), |
|
wp: wp, |
|
RecallClient: newRecallClient(c.GRPCClient["recall"]), |
|
UserClient: newUserClient(c.GRPCClient["user"]), |
|
RelationClient: newRelationClient(c.GRPCClient["relation"]), |
|
} |
|
return |
|
} |
|
|
|
func newRecallClient(cfg *conf.GRPCConfig) recallv1.RecsysRecallClient { |
|
cc, err := warden.NewClient(cfg.WardenConf).Dial(context.Background(), cfg.Addr) |
|
if err != nil { |
|
panic(err) |
|
} |
|
return recallv1.NewRecsysRecallClient(cc) |
|
} |
|
|
|
func newUserClient(cfg *conf.GRPCConfig) user.UserClient { |
|
cc, err := warden.NewClient(cfg.WardenConf).Dial(context.Background(), cfg.Addr) |
|
if err != nil { |
|
panic(err) |
|
} |
|
return user.NewUserClient(cc) |
|
} |
|
|
|
func newRelationClient(cfg *conf.GRPCConfig) relation.RelationClient { |
|
cc, err := warden.NewClient(cfg.WardenConf).Dial(context.Background(), cfg.Addr) |
|
if err != nil { |
|
panic(err) |
|
} |
|
return relation.NewRelationClient(cc) |
|
} |
|
|
|
// Close close the resource. |
|
func (d *Dao) Close() { |
|
d.mc.Close() |
|
d.redis.Close() |
|
d.bfRedis.Close() |
|
d.db.Close() |
|
} |
|
|
|
// Ping dao ping |
|
func (d *Dao) Ping(c context.Context) error { |
|
// TODO: if you need use mc,redis, please add |
|
return d.db.Ping(c) |
|
}
|
|
|