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.
55 lines
1.2 KiB
55 lines
1.2 KiB
package leshuazf
|
|
|
|
import (
|
|
"github.com/dtapps/go-library/utils/dorm"
|
|
"github.com/dtapps/go-library/utils/golog"
|
|
"github.com/dtapps/go-library/utils/gorequest"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ConfigClient struct {
|
|
AgentId string // 服务商编号，由乐刷分配的接入方唯一标识，明文传输。
|
|
Environment string // 环境
|
|
KeyAgent string
|
|
MongoDb *dorm.MongoClient // 日志数据库
|
|
PgsqlDb *gorm.DB // 日志数据库
|
|
DatabaseName string // 库名
|
|
}
|
|
|
|
// Client 乐刷
|
|
type Client struct {
|
|
client *gorequest.App // 请求客户端
|
|
log *golog.ApiClient // 日志服务
|
|
config *ConfigClient // 配置
|
|
}
|
|
|
|
func NewClient(config *ConfigClient) (*Client, error) {
|
|
|
|
var err error
|
|
c := &Client{config: config}
|
|
|
|
c.client = gorequest.NewHttp()
|
|
|
|
if c.config.PgsqlDb != nil {
|
|
c.log, err = golog.NewApiClient(
|
|
golog.WithGormClient(c.config.PgsqlDb),
|
|
golog.WithTableName(logTable),
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if c.config.MongoDb != nil {
|
|
c.log, err = golog.NewApiClient(
|
|
golog.WithMongoClient(c.config.MongoDb),
|
|
golog.WithDatabaseName(c.config.DatabaseName),
|
|
golog.WithCollectionName(logTable),
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return c, nil
|
|
}
|