kpt-pasture/module/backend/feeding.go

package backend

import (
	"context"
	"encoding/json"
	"fmt"
	"kpt-pasture/service/httpclient"

	feedingPb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
	"gitee.com/xuyiping_admin/pkg/xerr"
)

var FeedingHeaders = []*httpclient.Header{
	{
		Key:   "Content-Type",
		Value: "application/json",
	}, {
		Key:   "Accept",
		Value: "application/json",
	},
}

func (s *StoreEntry) GetFeedingHomepage(ctx context.Context, req *feedingPb.FeedingHomepageRequest) (*feedingPb.FeedingHomepageResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	pasture := userModel.AppPasture
	if pasture.FeedPastureId == 0 {
		return nil, xerr.Customf("饲喂数据未配置")
	}

	url := fmt.Sprintf("%d/feeding/tmrdata", pasture.FeedPastureId)
	res := &feedingPb.FeedingHomepageResponse{
		Code: 0,
		Msg:  "",
		Data: make([]*feedingPb.FeedingHomepageData, 0),
	}
	result, err := s.HttpClient.DoGet(url, FeedingHeaders)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	if err = json.Unmarshal(result, res); err != nil {
		return nil, xerr.WithStack(err)
	}
	return res, nil
}

func (s *StoreEntry) GetFeedingManagement(ctx context.Context, req *feedingPb.FeedingManagementRequest) (*feedingPb.FeedingManagementResponse, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, xerr.WithStack(err)
	}
	pasture := userModel.AppPasture
	if pasture.FeedPastureId == 0 {
		return nil, xerr.Customf("饲喂数据未配置")
	}

	url := fmt.Sprintf("%d/feeding/management?typea=%s&startdate=%s&enddate=%s", pasture.FeedPastureId, req.Typea, req.Startdate, req.Enddate)
	res := &feedingPb.FeedingManagementResponse{
		Code: 0,
		Msg:  "",
		Data: []*feedingPb.FeedingManagementData{},
	}
	result, err := s.HttpClient.DoGet(url, FeedingHeaders)
	if err != nil {
		return nil, xerr.WithStack(err)
	}

	if err = json.Unmarshal(result, res); err != nil {
		return nil, xerr.WithStack(err)
	}
	return res, nil
}