|
|
|
@ -6,36 +6,43 @@ import (
|
|
|
|
|
"gorm.io/gorm"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// App 美团联盟
|
|
|
|
|
type App struct {
|
|
|
|
|
secret string // 秘钥
|
|
|
|
|
appKey string // 渠道标记
|
|
|
|
|
pgsql *gorm.DB // pgsql数据库
|
|
|
|
|
type ConfigClient struct {
|
|
|
|
|
Secret string // 秘钥
|
|
|
|
|
AppKey string // 渠道标记
|
|
|
|
|
PgsqlDb *gorm.DB // pgsql数据库
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Client 美团联盟
|
|
|
|
|
type Client struct {
|
|
|
|
|
client *gorequest.App // 请求客户端
|
|
|
|
|
log *golog.Api // 日志服务
|
|
|
|
|
logTableName string // 日志表名
|
|
|
|
|
logStatus bool // 日志状态
|
|
|
|
|
config *ConfigClient
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func NewApp(secret string, appKey string, pgsql *gorm.DB) *App {
|
|
|
|
|
app := &App{secret: secret, appKey: appKey}
|
|
|
|
|
app.client = gorequest.NewHttp()
|
|
|
|
|
if pgsql != nil {
|
|
|
|
|
app.pgsql = pgsql
|
|
|
|
|
app.logStatus = true
|
|
|
|
|
app.logTableName = "meituan"
|
|
|
|
|
app.log = golog.NewApi(&golog.ApiConfig{
|
|
|
|
|
Db: pgsql,
|
|
|
|
|
TableName: app.logTableName,
|
|
|
|
|
func NewClient(config *ConfigClient) *Client {
|
|
|
|
|
|
|
|
|
|
c := &Client{}
|
|
|
|
|
c.config = config
|
|
|
|
|
|
|
|
|
|
c.client = gorequest.NewHttp()
|
|
|
|
|
if c.config.PgsqlDb != nil {
|
|
|
|
|
c.logStatus = true
|
|
|
|
|
c.logTableName = "meituan"
|
|
|
|
|
c.log = golog.NewApi(&golog.ApiConfig{
|
|
|
|
|
Db: c.config.PgsqlDb,
|
|
|
|
|
TableName: c.logTableName,
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
return app
|
|
|
|
|
|
|
|
|
|
return c
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (app *App) request(url string, params map[string]interface{}, method string) (resp gorequest.Response, err error) {
|
|
|
|
|
func (c *Client) request(url string, params map[string]interface{}, method string) (resp gorequest.Response, err error) {
|
|
|
|
|
|
|
|
|
|
// 创建请求
|
|
|
|
|
client := app.client
|
|
|
|
|
client := c.client
|
|
|
|
|
|
|
|
|
|
// 设置请求地址
|
|
|
|
|
client.SetUri(url)
|
|
|
|
@ -56,13 +63,13 @@ func (app *App) request(url string, params map[string]interface{}, method string
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// 日志
|
|
|
|
|
if app.logStatus == true {
|
|
|
|
|
go app.postgresqlLog(request)
|
|
|
|
|
if c.logStatus == true {
|
|
|
|
|
go c.postgresqlLog(request)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return request, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (app *App) GetAppKey() string {
|
|
|
|
|
return app.appKey
|
|
|
|
|
func (c *Client) GetAppKey() string {
|
|
|
|
|
return c.config.AppKey
|
|
|
|
|
}
|
|
|
|
|