lc_veterans/app/device/service/onDemandGroupService.go

package service

import (
	"fmt"
	"iot_manager_service/app/device/dao"
	"iot_manager_service/app/device/model"
	"iot_manager_service/app/system/service"
	"iot_manager_service/util"
	"strconv"
	"time"
)

// 中间件管理服务
var OnDemandGroupService = new(onDemandGroupService)

type onDemandGroupService struct{}

func (s *onDemandGroupService) Get(id int) (*dao.OnDemandGroup, *util.Errors) {
	// 创建查询实例
	device := &dao.OnDemandGroup{
		ID: id,
	}
	err := device.GetDevice()
	if err != nil {
		return nil, util.FailResponse(err.Error(), nil)
	}

	return device, nil
}

func (s *onDemandGroupService) CreateOrUpdate(userId int64, tenantId int, req *dao.OnDemandGroup) *util.Errors {
	device := req
	device.TenantId = tenantId
	device.UpdateUser = userId
	device.UpdateTime = time.Now()

	if device.ID == 0 {
		device.CreateTime = time.Now()
		device.CreateUser = userId

		if device.IsExistedBySN() {
			fmt.Printf("Create IsExistedBySN \n")
			return util.ParamsInvalidResponse(model.RepeatedPrompts, nil)
		}
		fmt.Printf("device = %+v \n", device)
		if err := device.Create(); err != nil {
			fmt.Printf("Create err = %s \n", err.Error())
			return util.FailResponse(err.Error(), nil)
		}
		service.OperationHisService.Save(userId, tenantId, util.OperationCreate, util.ModuleTypeDevice,
			util.DeviceTypeAlarmServer, util.GetDeviceObject(device.ID, device.PoleGroupName), util.OperationSuccess)
		return util.SuccessResponse(util.Succeeded, nil)
	}

	if err := device.Update(); err != nil {
		fmt.Printf("Update err = %s \n", err.Error())
		return util.FailResponse(err.Error(), nil)
	}

	service.OperationHisService.Save(userId, tenantId, util.OperationUpdate, util.ModuleTypeDevice,
		util.DeviceTypeAlarmServer, util.GetDeviceObject(device.ID, device.PoleGroupName), util.OperationSuccess)
	return util.SuccessResponse(util.Succeeded, nil)
}

func (s *onDemandGroupService) List(searchValue string, current, size int) ([]dao.OnDemandGroup, *util.Errors) {
	device := dao.OnDemandGroup{}

	if searchValue != "" {
		device.PoleGroupName = searchValue
	}

	offset := (current - 1) * size
	limit := size
	devices, err := device.GetDevices(offset, limit)
	if err != nil {
		return nil, util.FailResponse(err.Error(), nil)
	}
	return devices, nil
}

func (s *onDemandGroupService) Remove(userId int64, tenantId int, id int) *util.Errors {
	// 创建查询实例
	device := &dao.OnDemandGroup{
		ID:         id,
		IsDeleted:  1,
		UpdateUser: userId,
		UpdateTime: time.Now(),
	}
	err := device.Delete()
	if err != nil {
		return util.FailResponse(err.Error(), nil)
	}
	service.OperationHisService.Save(userId, tenantId, util.OperationRemove, util.ModuleTypeDevice,
		util.DeviceTypeAlarmServer, util.GetDeviceObject(device.ID, device.PoleGroupName), util.OperationSuccess)
	return nil
}

func (s *onDemandGroupService) GetList(tenantId int) ([]*dao.OnDemandGroup, *util.Errors) {
	var devices []*dao.OnDemandGroup
	err := util.Redis.Get(getOnDemandGroupListRedisKey(tenantId)).Scan(&devices)
	if err == nil {
		return devices, nil
	}

	device := &dao.OnDemandGroup{
		TenantId:  tenantId,
		IsDeleted: 0,
	}
	devices, err = device.GetAllDevices()
	if err != nil {
		return nil, util.FailResponse(err.Error(), nil)
	}
	_ = util.Redis.Set(getOnDemandGroupListRedisKey(tenantId), devices, 0).Err()

	return devices, nil
}

func getOnDemandGroupListRedisKey(tenantId int) string {
	return fmt.Sprintf(model.OnDemandGroupList, tenantId)
}

func (s *onDemandGroupService) GroupNumberList(id int) ([]model.RspOnDemandGroupNumber, *util.Errors) {
	// 创建查询实例
	device := &dao.OnDemandGroup{
		ID: id,
	}
	err := device.GetDevice()
	if err != nil {
		return nil, util.FailResponse(err.Error(), nil)
	}
	var list []model.RspOnDemandGroupNumber
	for i := 1; i <= device.LampPoleCount; i++ {
		list = append(list, model.RspOnDemandGroupNumber{Name: strconv.Itoa(i), Value: i})
	}

	return list, nil
}