- update log

master
李光春 2 years ago
parent 3ee8ec2378
commit c385963fab

@ -73,8 +73,6 @@ func NewApiClient(attrs ...*OperationAttr) (*ApiClient, error) {
return nil, errors.New("表名不能为空") return nil, errors.New("表名不能为空")
} }
c.mongoCollectionClient = c.mongoCollectionClient.Collection(c.config.tableName)
default: default:
return nil, errors.New("驱动为空") return nil, errors.New("驱动为空")
} }
@ -121,12 +119,12 @@ func (c *ApiClient) MongoRecord(mongoLog ApiMongoLog) error {
mongoLog.LogId = primitive.NewObjectID() mongoLog.LogId = primitive.NewObjectID()
_, err := c.mongoCollectionClient.InsertOne(mongoLog) _, err := c.mongoCollectionClient.Collection(c.config.tableName).InsertOne(mongoLog)
log.Printf("api.mongoRecord%s\n", err) log.Printf("api.mongoRecord%s\n", err)
return err return err
} }
// MongoQuery 查询 // MongoQuery 查询
func (c *ApiClient) MongoQuery() *dorm.MongoClient { func (c *ApiClient) MongoQuery() *dorm.MongoClient {
return c.mongoCollectionClient return c.mongoCollectionClient.Collection(c.config.tableName)
} }

@ -23,7 +23,7 @@ import (
// GinClient 框架 // GinClient 框架
type GinClient struct { type GinClient struct {
gormClient *gorm.DB // 驱动 gormClient *gorm.DB // 驱动
mongoCollectionClient *dorm.MongoClient // 驱动(温馨提示:需要已选择库和表) mongoCollectionClient *dorm.MongoClient // 驱动(温馨提示:需要已选择库)
ipService *goip.Client // ip服务 ipService *goip.Client // ip服务
config struct { config struct {
logType string // 日志类型 logType string // 日志类型
@ -77,7 +77,9 @@ func NewGinClient(attrs ...*OperationAttr) (*GinClient, error) {
return nil, errors.New("驱动不能为空") return nil, errors.New("驱动不能为空")
} }
c.mongoCollectionClient = c.mongoCollectionClient.Collection(c.config.tableName) if c.config.tableName == "" {
return nil, errors.New("表名不能为空")
}
default: default:
return nil, errors.New("驱动为空") return nil, errors.New("驱动为空")
@ -254,12 +256,12 @@ func (c *GinClient) mongoRecord(mongoLog GinMongoLog) error {
mongoLog.LogId = primitive.NewObjectID() mongoLog.LogId = primitive.NewObjectID()
_, err := c.mongoCollectionClient.InsertOne(mongoLog) _, err := c.mongoCollectionClient.Collection(c.config.tableName).InsertOne(mongoLog)
log.Printf("gin.mongoRecord%s\n", err) log.Printf("gin.mongoRecord%s\n", err)
return err return err
} }
// MongoQuery 查询 // MongoQuery 查询
func (c *GinClient) MongoQuery() *dorm.MongoClient { func (c *GinClient) MongoQuery() *dorm.MongoClient {
return c.mongoCollectionClient return c.mongoCollectionClient.Collection(c.config.tableName)
} }

@ -25,7 +25,7 @@ func WithGormClient(client *gorm.DB) *OperationAttr {
return &OperationAttr{gormClient: client, logType: logTypeGorm} return &OperationAttr{gormClient: client, logType: logTypeGorm}
} }
// WithMongoCollectionClient 数据库驱动(温馨提示:需要已选择库和表) // WithMongoCollectionClient 数据库驱动(温馨提示:需要已选择库)
func WithMongoCollectionClient(client *dorm.MongoClient) *OperationAttr { func WithMongoCollectionClient(client *dorm.MongoClient) *OperationAttr {
return &OperationAttr{mongoCollectionClient: client, logType: logTypeMongo} return &OperationAttr{mongoCollectionClient: client, logType: logTypeMongo}
} }

Loading…
Cancel
Save