package wikeyun import ( "context" "go.dtapp.net/dorm" "go.dtapp.net/goip" "go.dtapp.net/golog" "go.dtapp.net/gorequest" ) // client *dorm.GormClient type gormClientFun func() *dorm.GormClient // client *dorm.MongoClient // databaseName string type mongoClientFun func() (*dorm.MongoClient, string) // ClientConfig 实例配置 type ClientConfig struct { StoreId int // 店铺ID AppKey int // key AppSecret string // secret GormClientFun gormClientFun // 日志配置 MongoClientFun mongoClientFun // 日志配置 Debug bool // 日志开关 } // Client 实例 type Client struct { requestClient *gorequest.App // 请求服务 config struct { clientIp string // 当前Ip storeId int // 店铺ID appKey int // key appSecret string // secret } log struct { gorm bool // 日志开关 gormClient *dorm.GormClient // 日志数据库 logGormClient *golog.ApiClient // 日志服务 mongo bool // 日志开关 mongoClient *dorm.MongoClient // 日志数据库 logMongoClient *golog.ApiClient // 日志服务 } } // NewClient 创建实例化 func NewClient(config *ClientConfig) (*Client, error) { var err error c := &Client{} c.config.storeId = config.StoreId c.config.appKey = config.AppKey c.config.appSecret = config.AppSecret c.requestClient = gorequest.NewHttp() gormClient := config.GormClientFun() if gormClient != nil && gormClient.Db != nil { c.log.logGormClient, err = golog.NewApiGormClient(&golog.ApiGormClientConfig{ GormClientFun: func() (*dorm.GormClient, string) { return gormClient, logTable }, Debug: config.Debug, }) if err != nil { return nil, err } c.log.gorm = true c.log.gormClient = gormClient } mongoClient, databaseName := config.MongoClientFun() if mongoClient != nil && mongoClient.Db != nil { c.log.logMongoClient, err = golog.NewApiMongoClient(&golog.ApiMongoClientConfig{ MongoClientFun: func() (*dorm.MongoClient, string, string) { return mongoClient, databaseName, logTable }, Debug: config.Debug, }) if err != nil { return nil, err } c.log.mongo = true c.log.mongoClient = mongoClient } xip := goip.GetOutsideIp(context.Background()) if xip != "" && xip != "0.0.0.0" { c.config.clientIp = xip } return c, nil }