|
|
|
package wechatqy
|
|
|
|
|
|
|
|
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 {
|
|
|
|
AppId string
|
|
|
|
AgentId int
|
|
|
|
Secret string
|
|
|
|
RedirectUri string
|
|
|
|
Key string // key
|
|
|
|
MongoDb *dorm.MongoClient // 日志数据库
|
|
|
|
PgsqlDb *gorm.DB // 日志数据库
|
|
|
|
DatabaseName string // 库名
|
|
|
|
}
|
|
|
|
|
|
|
|
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
|
|
|
|
}
|