kpt-tmr-group/model/group_pasture.go

package model

import (
	"bytes"
	"encoding/json"
	"fmt"
	"go.uber.org/zap"
	"io/ioutil"
	"kpt-tmr-group/pkg/logger/zaplog"
	"kpt-tmr-group/pkg/tool"
	"kpt-tmr-group/pkg/xerr"
	operationPb "kpt-tmr-group/proto/go/backend/operation"
	"net/http"
	"time"
)

type GroupPasture struct {
	Id           int64                   `json:"id,omitempty"`
	Name         string                  `json:"name,omitempty"`
	Account      string                  `json:"account,omitempty"`
	Password     string                  `json:"password"`
	ManagerUser  string                  `json:"manager_user"`
	ManagerPhone string                  `json:"manager_phone"`
	Domain       string                  `json:"domain"`
	Extranet     string                  `json:"extranet"`
	Intranet     string                  `json:"intranet"`
	IsShow       operationPb.IsShow_Kind `json:"is_show,omitempty"`
	IsDelete     operationPb.IsShow_Kind `json:"is_delete,omitempty"`
	Address      string                  `json:"address"`
	CreatedAt    int64                   `json:"created_at,omitempty"`
	UpdatedAt    int64                   `json:"updated_at,omitempty"`
}

func (g *GroupPasture) TableName() string {
	return "group_pasture"
}

const (
	InitManagerPassword = "123456"
	UrlDataByName       = "authdata/GetDataByName"
	UrlReportForm       = "authdata/GetReportform"
)

var FeedCharUrl = map[string]string{
	"mr":   "authdata/chart/feedEffMR",
	"sl":   "authdata/chart/feedEffSL",
	"hl":   "authdata/chart/feedEffHL",
	"zh":   "authdata/chart/feedEffZH",
	"cbft": "authdata/chart/feedEffCBFT",
}

type PastureTokenRequest struct {
	UserName string `json:"username"`
	Password string `json:"password"`
}

type PastureTokenResponse struct {
	Code int32             `json:"code"`
	Msg  string            `json:"msg"`
	Data *PastureTokenData `json:"data"`
}

type PastureTokenData struct {
	Token string `json:"token"`
}

func NewGroupPasture(req *operationPb.AddPastureRequest) *GroupPasture {
	groupPasture := &GroupPasture{
		Name:         req.Name,
		Account:      req.Account,
		Password:     tool.Md5String(InitManagerPassword),
		ManagerUser:  req.ManagerUser,
		ManagerPhone: req.ManagerPhone,
		IsShow:       operationPb.IsShow_OK,
		IsDelete:     operationPb.IsShow_OK,
		Address:      req.Address,
	}
	return groupPasture
}

type PastureClient struct {
	Token      string        `json:"token"`
	Detail     *GroupPasture `json:"detail"`
	authClient *http.Client
}

func NewPastureClient(g *GroupPasture) *PastureClient {
	return &PastureClient{
		Detail: g,
		authClient: &http.Client{
			Timeout: time.Duration(5) * time.Second,
		},
	}
}

func (p *PastureClient) doRequest(req *http.Request) ([]byte, error) {
	resp, err := p.authClient.Do(req)
	if err != nil {
		zaplog.Error("PastureClient", zap.Any("authClient.Do", err))
		return nil, xerr.WithStack(err)
	}
	b, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		zaplog.Error("PastureClient", zap.Any("ioutil.ReadAll", err))
		return nil, xerr.WithStack(err)
	}
	if resp.StatusCode != http.StatusOK {
		if len(b) > 0 {
			return nil, xerr.Customf("err:%v,body:%s", err, string(b))
		} else {
			return nil, xerr.Customf("err:%v", err)
		}
	}
	return b, nil
}

func (p *PastureClient) DoGet(url string) ([]byte, error) {
	// 获取token
	if err := p.GetToken(); err != nil {
		zaplog.Error("PastureClient", zap.Any("GetToken Err", err), zap.Any("detail", p.Detail))
		return nil, xerr.WithStack(err)
	}

	req, err := http.NewRequest(http.MethodGet, url, nil)
	if err != nil {
		zaplog.Error("PastureClient", zap.Any("DoGet", err))
		return nil, err
	}
	req.Header.Add("Accept", "application/json")
	req.Header.Add("token", p.Token)
	req.Header.Add("Content-Type", "application/json")
	return p.doRequest(req)
}

func (p *PastureClient) DoPost(url string, body interface{}) ([]byte, error) {
	b, err := json.Marshal(body)
	if err != nil {
		zaplog.Error("PastureClient", zap.Any("DoPost-Marshal", err))
		return nil, xerr.WithStack(err)
	}
	// 获取token
	if err = p.GetToken(); err != nil {
		zaplog.Error("PastureClient", zap.Any("GetToken Err", err), zap.Any("detail", p.Detail))
		return nil, xerr.WithStack(err)
	}
	req, err := http.NewRequest(http.MethodPost, url, bytes.NewReader(b))
	if err != nil {
		zaplog.Error("PastureClient", zap.Any("NewRequest", err))
		return nil, xerr.WithStack(err)
	}
	req.Header.Add("Accept", "application/json")
	req.Header.Add("token", p.Token)
	req.Header.Add("Content-Type", "application/json")
	return p.doRequest(req)
}

func (p *PastureClient) GetToken() error {
	url := fmt.Sprintf("%s/%s", p.Detail.Domain, "auth")
	body := &PastureTokenRequest{
		UserName: p.Detail.Account,
		Password: p.Detail.Password,
	}
	b, err := json.Marshal(body)
	if err != nil {
		return xerr.WithStack(err)
	}
	req, err := http.NewRequest(http.MethodPost, url, bytes.NewReader(b))
	if err != nil {
		zaplog.Error("PastureClient", zap.Any("GetToken", err))
		return xerr.WithStack(err)
	}
	req.Header.Add("Accept", "application/json")
	req.Header.Add("Content-Type", "application/json")
	result, err := p.doRequest(req)
	if err != nil {
		zaplog.Error("PastureClient", zap.Any("GetToken-doRequest", err))
		return xerr.WithStack(err)
	}
	response := &PastureTokenResponse{}
	if err = json.Unmarshal(result, response); err != nil {
		zaplog.Error("PastureClient", zap.String("GetToken", string(result)))
		return xerr.WithStack(err)
	}
	if response.Code != http.StatusOK || response.Data == nil || response.Data.Token == "" {
		zaplog.Error("PastureClient", zap.Any("response", response))
		return xerr.Customf("获取牧场端token失败 Err:%s", err)
	}
	p.Token = response.Data.Token
	return nil
}

type GroupPastureSlice []*GroupPasture

func (g GroupPastureSlice) ToPB() []*operationPb.AddPastureRequest {
	res := make([]*operationPb.AddPastureRequest, len(g))
	for i, v := range g {
		res[i] = &operationPb.AddPastureRequest{
			Id:              int32(v.Id),
			Name:            v.Name,
			Account:         v.Account,
			ManagerUser:     v.ManagerUser,
			ManagerPhone:    v.ManagerPhone,
			Address:         v.Address,
			IsShow:          v.IsShow,
			CreatedAt:       int32(v.CreatedAt),
			CreatedAtFormat: time.Unix(v.CreatedAt, 0).Format(LayoutTime),
		}
	}
	return res
}

func (g *GroupPasture) ToPb() *operationPb.AddPastureRequest {
	return &operationPb.AddPastureRequest{
		Id:           int32(g.Id),
		Name:         g.Name,
		ManagerUser:  g.ManagerUser,
		ManagerPhone: g.ManagerPhone,
		Address:      g.Address,
		IsShow:       g.IsShow,
		CreatedAt:    int32(g.CreatedAt),
	}
}