xps-server-go/service/api_trans_unit_service.go

package service

import (
    "errors"
    "xps/cache"
    "xps/datamodels"
    "xps/datasource"
    "xps/repositories"
    "xps/viewmodels"
)

type ApiTransUnitService interface {
    GetList(m map[string]interface{}) ([]viewmodels.ApiTransUnitInfo, error)
    GetPage(m map[string]interface{}) (*viewmodels.PageResult, error)
    GetById(apiId, atuId int64) (*viewmodels.ApiTransUnitInfo, error)
    Create(d *datamodels.ApiTransUnit) error
    Update(d *datamodels.ApiTransUnit) error
    Delete(m map[string]interface{}) error
}

type apiTransUnitService struct {
    apiRepo *repositories.ApiRepo
    transRepo *repositories.ApiTransUnitRepo
}

func NewApiTransUnitService() ApiTransUnitService {
    return &apiTransUnitService{
        apiRepo: repositories.NewApiRepo(datasource.InstanceMaster()),
        transRepo: repositories.NewApiTransUnitRepo(datasource.InstanceMaster()),
    }
}

func (s *apiTransUnitService) GetList(m map[string]interface{}) ([]viewmodels.ApiTransUnitInfo, error) {
    return s.transRepo.GetList(m)
}

func (s *apiTransUnitService) GetPage(m map[string]interface{}) (*viewmodels.PageResult, error) {
    return s.transRepo.GetPage(m)
}

func (s *apiTransUnitService) GetById(apiId, atuId int64) (*viewmodels.ApiTransUnitInfo, error) {
    return s.transRepo.GetById(apiId, atuId)
}

func (s *apiTransUnitService) Create(atu *datamodels.ApiTransUnit) error {
    atu.AtuId = NewID()
    err := s.transRepo.Create(atu)
    if err != nil {
        return err
    }
    
    modId := atu.ModelId
    apiId := atu.ApiId
    return s.CacheApiData(modId, apiId)
}

func (s *apiTransUnitService) Update(atu *datamodels.ApiTransUnit) error {
    err := s.transRepo.Update(atu)
    if err != nil {
        return err
    }
    
    modId := atu.ModelId
    apiId := atu.ApiId
    return s.CacheApiData(modId, apiId)
}

func (s *apiTransUnitService) Delete(m map[string]interface{}) error {
    return s.transRepo.Delete(m)
}

func (s *apiTransUnitService) CacheApiData(modId, apiId int64) error {
    api, err := s.apiRepo.GetById(modId, apiId)
    if err != nil {
        return nil
    }
    
    trans, err := s.transRepo.GetListById(apiId)
    if err != nil {
        return nil
    }
    
    cacheData := &viewmodels.ApiDataInfo{
        ModId: api.ModId,
        ModTitle: api.ModTitle,
        ApiId: api.ApiId,
        ApiCode: api.ApiCode,
        ApiTitle: api.ApiTitle,
        ApiType: api.ApiType,
        ApiParams: api.ApiParams,
        ApiDesc: api.ApiDesc,
        Status: api.Status,
        TransUnits: trans,
    }
    
    // 缓存数据
    err = cache.SetApiData(api.ApiCode, cacheData)
    if err != nil {
        return errors.New("Redis缓存Api数据失败")
    }
    
    return nil
}