diff --git a/client.go b/client.go index f2860d0..6660dc5 100644 --- a/client.go +++ b/client.go @@ -39,11 +39,11 @@ type Client struct { pid string // 推广位 } log struct { - gormClient *dorm.GormClient // 日志数据库 gorm bool // 日志开关 + gormClient *dorm.GormClient // 日志数据库 logGormClient *golog.ApiClient // 日志服务 - mongoClient *dorm.MongoClient // 日志数据库 mongo bool // 日志开关 + mongoClient *dorm.MongoClient // 日志数据库 logMongoClient *golog.ApiClient // 日志服务 } } @@ -63,7 +63,7 @@ func NewClient(config *ClientConfig) (*Client, error) { c.requestClient.Uri = apiUrl gormClient := config.GormClientFun() - if gormClient.Db != nil { + if gormClient != nil && gormClient.Db != nil { c.log.logGormClient, err = golog.NewApiGormClient(func() (*dorm.GormClient, string) { return gormClient, logTable }, config.Debug) @@ -71,11 +71,11 @@ func NewClient(config *ClientConfig) (*Client, error) { return nil, err } c.log.gorm = true + c.log.gormClient = gormClient } - c.log.gormClient = gormClient mongoClient, databaseName := config.MongoClientFun() - if mongoClient.Db != nil { + if mongoClient != nil && mongoClient.Db != nil { c.log.logMongoClient, err = golog.NewApiMongoClient(func() (*dorm.MongoClient, string, string) { return mongoClient, databaseName, logTable }, config.Debug) @@ -83,8 +83,8 @@ func NewClient(config *ClientConfig) (*Client, error) { return nil, err } c.log.mongo = true + c.log.mongoClient = mongoClient } - c.log.mongoClient = mongoClient return c, nil } diff --git a/const.go b/const.go index a936eba..22b642c 100644 --- a/const.go +++ b/const.go @@ -8,4 +8,4 @@ const ( logTable = "pinduoduo" ) -const Version = "1.0.10" +const Version = "1.0.11" diff --git a/request.go b/request.go index 7d74ec8..6ea213d 100644 --- a/request.go +++ b/request.go @@ -24,10 +24,10 @@ func (c *Client) request(ctx context.Context, params map[string]interface{}) (go } // 日志 - if c.log.gorm == true { + if c.log.gorm { go c.log.logGormClient.GormMiddlewareCustom(ctx, gostring.ToString(params["type"]), request, Version) } - if c.log.mongo == true { + if c.log.mongo { go c.log.logMongoClient.GormMiddlewareCustom(ctx, gostring.ToString(params["type"]), request, Version) }