添加物理信息

platformapi/q_bida/q_bida_access.go

@@ -0,0 +1,120 @@
+package q_bida
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"git.rosy.net.cn/baseapi"
+	"git.rosy.net.cn/baseapi/platformapi"
+	"git.rosy.net.cn/baseapi/utils"
+	"net/http"
+	"sync"
+	"time"
+)
+
+type Api struct {
+	token          string // 登录凭证
+	expirationTime int64  // 过期时间
+	account        string // 账号
+	password       string // 密码
+	locker         sync.RWMutex
+	client         *http.Client
+	config         *platformapi.APIConfig
+}
+
+// SetToken 设置token
+func (a *Api) SetToken(token string) {
+	a.locker.Lock()
+	defer a.locker.Unlock()
+	a.token = token
+}
+
+// SetExpirationTime 设置过期时间
+func (a *Api) SetExpirationTime(expiration int64) {
+	a.locker.Lock()
+	defer a.locker.Unlock()
+	a.expirationTime = expiration
+}
+
+// NewQBiDa 初始化token
+func NewQBiDa(account, password string, config ...*platformapi.APIConfig) *Api {
+	if account == "" || password == "" {
+		panic("账号密码不能为空")
+	}
+	// 获取token，设置token和过期时间
+	curConfig := platformapi.DefAPIConfig
+	if len(config) > 0 {
+		curConfig = *config[0]
+	}
+	api := &Api{
+		token:          "",
+		expirationTime: 0,
+		account:        account,
+		password:       password,
+		locker:         sync.RWMutex{},
+		client:         &http.Client{Timeout: curConfig.ClientTimeout},
+		config:         &curConfig,
+	}
+	if err := api.CheckTokenExpiration(); err != nil {
+		panic(err)
+	}
+
+	return api
+}
+
+// CheckTokenExpiration 检查token是否存在，或者是否过期
+func (a *Api) CheckTokenExpiration() error {
+	if a.account == "" || a.password == "" {
+		return errors.New("账号或者密码为空")
+	}
+
+	// 获取token NewQBiDa
+	if a.token == "" || a.expirationTime < time.Now().Unix() {
+		return a.GetToken()
+	}
+
+	return nil
+}
+
+// AccessInfo 发送请求
+func (a *Api) AccessInfo(baseUrl, url, requestMethods string, param map[string]interface{}) (retVal map[string]interface{}, err error) {
+	err = platformapi.AccessPlatformAPIWithRetry(a.client,
+		func() *http.Request {
+			var request *http.Request
+			if requestMethods == http.MethodPost {
+				fullURL := utils.GenerateGetURL(baseUrl, url, nil)
+				// 获取json结构体的参数体
+				requestBody := new(bytes.Buffer)
+				json.NewEncoder(requestBody).Encode(param)
+				// 发送请求
+				request, _ = http.NewRequest(http.MethodPost, fullURL, requestBody)
+				request.Header.Set("Content-Type", "application/json")
+			} else {
+				request, _ = http.NewRequest(http.MethodGet, utils.GenerateGetURL(baseUrl, url, param), nil)
+			}
+			if url != LogInUrl {
+				request.Header.Set("token", a.token)
+			}
+			return request
+		},
+		a.config,
+		func(response *http.Response, bodyStr string, jsonResult1 map[string]interface{}) (errLevel string, err error) {
+			if jsonResult1 == nil {
+				return platformapi.ErrLevelRecoverableErr, fmt.Errorf("mapData is nil")
+			}
+			retVal = jsonResult1
+			if jsonResult1["code"] == nil {
+				return platformapi.ErrLevelGeneralFail, fmt.Errorf("返回结果格式不正常")
+			}
+			code := int(utils.MustInterface2Int64(jsonResult1["code"]))
+			if code == ResponseCodeSuccess {
+				retVal, _ = jsonResult1["data"].(map[string]interface{})
+				return platformapi.ErrLevelSuccess, nil
+			}
+			newErr := utils.NewErrorIntCode(jsonResult1["msg"].(string), code)
+			baseapi.SugarLogger.Debugf("QBiDa AccessUserPage failed, jsonResult1:%s", utils.Format4Output(jsonResult1, true))
+			return platformapi.ErrLevelCodeIsNotOK, newErr
+		})
+	return retVal, err
+}