kpt-tmr-group/module/backend/pasture_sync_service.go

package backend

import (
	"context"
	"kpt-tmr-group/model"
	"kpt-tmr-group/pkg/xerr"
	operationPb "kpt-tmr-group/proto/go/backend/operation"
)

const (
	FeedCategory = "feed"
	CowCategory  = "cow"
)

func (s *StoreEntry) CategoryData(ctx context.Context, req *operationPb.ClassRequest) error {

	pastureDateil, err := s.GetGroupPastureListById(ctx, int64(req.PastureId))
	if err != nil {
		return xerr.WithStack(err)
	}
	switch req.KeyWord {
	case FeedCategory:
		newFeedData := model.NewPastureForageCategory(req, pastureDateil)
		if err := s.DB.Model(new(model.ForageCategory)).Create(newFeedData).Error; err != nil {
			return xerr.WithStack(err)
		}
	case CowCategory:
		newCattleData := model.NewPastureCattleCategory(req, pastureDateil)
		if err := s.DB.Model(new(model.CattleCategory)).Create(newCattleData).Error; err != nil {
			return xerr.WithStack(err)
		}
	}
	return nil
}