device_collection.go 15.7 KB
package service

import (
	"errors"
	"fmt"
	"github.com/linmadan/egglib-go/core/application"
	"github.com/linmadan/egglib-go/utils/tool_funs"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/application/deviceCollection/command"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/application/deviceCollection/query"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/application/factory"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/infrastructure/domainService"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/infrastructure/redis"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/infrastructure/utils"
	"gitlab.fjmaimaimai.com/allied-creation/allied-creation-manufacture/pkg/log"
	"strconv"
	"sync"
	"time"
)

type DeviceCollectionService struct {
}

var (
	DeviceDataCache = NewDeviceDataInstance()
)

// 创建
func (deviceCollectionService *DeviceCollectionService) CreateDeviceCollection(createDeviceCollectionCommand *command.CreateDeviceCollectionCommand) (interface{}, error) {
	if err := createDeviceCollectionCommand.ValidateCommand(); err != nil {
		return nil, application.ThrowError(application.ARG_ERROR, err.Error())
	}
	transactionContext, err := factory.CreateTransactionContext(nil)
	if err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	if err := transactionContext.StartTransaction(); err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	defer func() {
		transactionContext.RollbackTransaction()
	}()
	newDeviceCollection := &domain.DeviceCollection{
		//DeviceCollectionId: createDeviceCollectionCommand.DeviceCollectionId,
		WorkShopName:   createDeviceCollectionCommand.WorkShopName,
		DeviceType:     createDeviceCollectionCommand.DeviceType,
		StartupStatus:  createDeviceCollectionCommand.StartupStatus,
		DeviceSn:       createDeviceCollectionCommand.DeviceSn,
		ComStatus:      createDeviceCollectionCommand.ComStatus,
		CollectionTime: createDeviceCollectionCommand.CollectionTime,
		Values:         createDeviceCollectionCommand.Values,
	}
	var deviceCollectionRepository domain.DeviceCollectionRepository
	if value, err := factory.CreateDeviceCollectionRepository(map[string]interface{}{
		"transactionContext": transactionContext,
	}); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		deviceCollectionRepository = value
	}
	if deviceCollection, err := deviceCollectionRepository.Save(newDeviceCollection); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		if err := transactionContext.CommitTransaction(); err != nil {
			return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
		}
		return map[string]interface{}{
			"deviceCollection": deviceCollection,
		}, nil
	}
}

// 创建
func (deviceCollectionService *DeviceCollectionService) DeviceCollection(createDeviceCollectionCommand *command.CreateDeviceCollectionCommand) (interface{}, error) {
	if err := createDeviceCollectionCommand.ValidateCommand(); err != nil {
		return nil, application.ThrowError(application.ARG_ERROR, err.Error())
	}

	newDeviceCollection := &domain.DeviceCollection{
		//DeviceCollectionId: createDeviceCollectionCommand.DeviceCollectionId,
		WorkShopName:     createDeviceCollectionCommand.WorkShopName,
		DeviceType:       createDeviceCollectionCommand.DeviceType,
		StartupStatus:    createDeviceCollectionCommand.StartupStatus,
		DeviceSn:         createDeviceCollectionCommand.DeviceSn,
		ComStatus:        createDeviceCollectionCommand.ComStatus,
		CollectionTime:   createDeviceCollectionCommand.CollectionTime,
		Values:           createDeviceCollectionCommand.Values,
		LatestUpdateTime: time.Now(),
	}
	var lastDeviceCollectionRecord = &domain.DeviceCollection{}
	var err error
	if createDeviceCollectionCommand.PreCheck {
		//前置验证,限制设备上报速率
		if lastDeviceCollectionRecord, err = DeviceDataCache.Add(newDeviceCollection.DeviceSn, newDeviceCollection.DeviceType, newDeviceCollection); err != nil {
			//log.Logger.Error(err.Error()+newDeviceCollection.DeviceSn)
			return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
		}
		if lastDeviceCollectionRecord == nil {
			//log.Logger.Error("未找到上一条设备数据")
			return nil, nil
		}
	}

	transactionContext, err := factory.CreateTransactionContext(nil)
	if err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	if err := transactionContext.StartTransaction(); err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	defer func() {
		transactionContext.RollbackTransaction()
	}()

	var deviceCollectionRepository domain.DeviceCollectionRepository
	if value, err := factory.CreateDeviceCollectionRepository(map[string]interface{}{
		"transactionContext": transactionContext,
	}); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		deviceCollectionRepository = value
	}

	//计算区间的产能

	if v, ok := newDeviceCollection.Values["Count"]; ok {
		newDeviceCollection.Values["Total"] = v // 记录原始值
		newDeviceCollection.Values["Count"] = 0
		curCount, errCurCount := strconv.Atoi(utils.AssertString(v))
		lastCount, errLastCount := strconv.Atoi(utils.AssertString(lastDeviceCollectionRecord.Values["Count"]))
		var count int
		if errLastCount == nil && errCurCount == nil && lastCount <= curCount {
			if lastCount <= curCount {
				count = curCount - lastCount
			} else {
				count = 0
				/*设备统计的数量超过一定范围会重置为0,特殊处理0操作*/
				if lastCount > 10000000 && curCount < 1000 {
					count = curCount
				}
			}
			newDeviceCollection.Values["Count"] = count
			newDeviceCollection.ProductCount = count
		}
	}
	if _, valErr := newDeviceCollection.Valid(); valErr != nil {
		newDeviceCollection.ResetProductCountToZero()
	}
	// TODO:测试假数据,后期注释掉
	//if createDeviceCollectionCommand.DeviceType == domain.DeviceTypeChuanChuanJi {
	//	newDeviceCollection.Values["Count"] = rand.Intn(300)
	//	newDeviceCollection.StartupStatus = 1
	//	newDeviceCollection.ComStatus = 1
	//	newDeviceCollection.CollectionTime = time.Date(2022,4,18,7,0,0,0,time.Local)
	//}

	// !!!!!!! 注意:保存到库以后 collect_time 变成了 UTC时间,后续处理的时候都要加上collectTime.Local()
	if newDeviceCollection, err = deviceCollectionRepository.Save(newDeviceCollection); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	}
	if err = domainService.SendWorkshopDeviceData(newDeviceCollection); err != nil {
		log.Logger.Error("车间设备数据加入redis失败:" + err.Error())
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	}
	if err := transactionContext.CommitTransaction(); err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	return nil, nil

}

// 返回
func (deviceCollectionService *DeviceCollectionService) GetDeviceCollection(getDeviceCollectionQuery *query.GetDeviceCollectionQuery) (interface{}, error) {
	if err := getDeviceCollectionQuery.ValidateQuery(); err != nil {
		return nil, application.ThrowError(application.ARG_ERROR, err.Error())
	}
	transactionContext, err := factory.CreateTransactionContext(nil)
	if err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	if err := transactionContext.StartTransaction(); err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	defer func() {
		transactionContext.RollbackTransaction()
	}()
	var deviceCollectionRepository domain.DeviceCollectionRepository
	if value, err := factory.CreateDeviceCollectionRepository(map[string]interface{}{
		"transactionContext": transactionContext,
	}); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		deviceCollectionRepository = value
	}
	deviceCollection, err := deviceCollectionRepository.FindOne(map[string]interface{}{"deviceCollectionId": getDeviceCollectionQuery.DeviceCollectionId})
	if err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	}
	if deviceCollection == nil {
		return nil, application.ThrowError(application.RES_NO_FIND_ERROR, fmt.Sprintf("%s", string(getDeviceCollectionQuery.DeviceCollectionId)))
	} else {
		if err := transactionContext.CommitTransaction(); err != nil {
			return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
		}
		return deviceCollection, nil
	}
}

// 返回列表
func (deviceCollectionService *DeviceCollectionService) ListDeviceCollection(listDeviceCollectionQuery *query.ListDeviceCollectionQuery) (interface{}, error) {
	if err := listDeviceCollectionQuery.ValidateQuery(); err != nil {
		return nil, application.ThrowError(application.ARG_ERROR, err.Error())
	}
	transactionContext, err := factory.CreateTransactionContext(nil)
	if err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	if err := transactionContext.StartTransaction(); err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	defer func() {
		transactionContext.RollbackTransaction()
	}()
	var deviceCollectionRepository domain.DeviceCollectionRepository
	if value, err := factory.CreateDeviceCollectionRepository(map[string]interface{}{
		"transactionContext": transactionContext,
	}); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		deviceCollectionRepository = value
	}
	if count, deviceCollections, err := deviceCollectionRepository.Find(tool_funs.SimpleStructToMap(listDeviceCollectionQuery)); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		if err := transactionContext.CommitTransaction(); err != nil {
			return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
		}
		return tool_funs.SimpleWrapGridMap(count, deviceCollections), nil
	}
}

// 移除
func (deviceCollectionService *DeviceCollectionService) RemoveDeviceCollection(removeDeviceCollectionCommand *command.RemoveDeviceCollectionCommand) (interface{}, error) {
	if err := removeDeviceCollectionCommand.ValidateCommand(); err != nil {
		return nil, application.ThrowError(application.ARG_ERROR, err.Error())
	}
	transactionContext, err := factory.CreateTransactionContext(nil)
	if err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	if err := transactionContext.StartTransaction(); err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	defer func() {
		transactionContext.RollbackTransaction()
	}()
	var deviceCollectionRepository domain.DeviceCollectionRepository
	if value, err := factory.CreateDeviceCollectionRepository(map[string]interface{}{
		"transactionContext": transactionContext,
	}); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		deviceCollectionRepository = value
	}
	deviceCollection, err := deviceCollectionRepository.FindOne(map[string]interface{}{"deviceCollectionId": removeDeviceCollectionCommand.DeviceCollectionId})
	if err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	}
	if deviceCollection == nil {
		return nil, application.ThrowError(application.RES_NO_FIND_ERROR, fmt.Sprintf("%s", string(removeDeviceCollectionCommand.DeviceCollectionId)))
	}
	if deviceCollection, err := deviceCollectionRepository.Remove(deviceCollection); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		if err := transactionContext.CommitTransaction(); err != nil {
			return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
		}
		return map[string]interface{}{
			"deviceCollection": deviceCollection,
		}, nil
	}
}

// 更新
func (deviceCollectionService *DeviceCollectionService) UpdateDeviceCollection(updateDeviceCollectionCommand *command.UpdateDeviceCollectionCommand) (interface{}, error) {
	if err := updateDeviceCollectionCommand.ValidateCommand(); err != nil {
		return nil, application.ThrowError(application.ARG_ERROR, err.Error())
	}
	transactionContext, err := factory.CreateTransactionContext(nil)
	if err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	if err := transactionContext.StartTransaction(); err != nil {
		return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
	}
	defer func() {
		transactionContext.RollbackTransaction()
	}()
	var deviceCollectionRepository domain.DeviceCollectionRepository
	if value, err := factory.CreateDeviceCollectionRepository(map[string]interface{}{
		"transactionContext": transactionContext,
	}); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		deviceCollectionRepository = value
	}
	deviceCollection, err := deviceCollectionRepository.FindOne(map[string]interface{}{"deviceCollectionId": updateDeviceCollectionCommand.DeviceCollectionId})
	if err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	}
	if deviceCollection == nil {
		return nil, application.ThrowError(application.RES_NO_FIND_ERROR, fmt.Sprintf("%s", string(updateDeviceCollectionCommand.DeviceCollectionId)))
	}
	if err := deviceCollection.Update(tool_funs.SimpleStructToMap(updateDeviceCollectionCommand)); err != nil {
		return nil, application.ThrowError(application.BUSINESS_ERROR, err.Error())
	}
	if deviceCollection, err := deviceCollectionRepository.Save(deviceCollection); err != nil {
		return nil, application.ThrowError(application.INTERNAL_SERVER_ERROR, err.Error())
	} else {
		if err := transactionContext.CommitTransaction(); err != nil {
			return nil, application.ThrowError(application.TRANSACTION_ERROR, err.Error())
		}
		return map[string]interface{}{
			"deviceCollection": deviceCollection,
		}, nil
	}
}

func NewDeviceCollectionService(options map[string]interface{}) *DeviceCollectionService {
	newDeviceCollectionService := &DeviceCollectionService{}
	return newDeviceCollectionService
}

const DefaultReceiveSpan = 60 // 60 sec

type DeviceDataInstance struct {
	//deviceDataCache  sync.Map
	deviceDataLastTime sync.Map
}

func NewDeviceDataInstance() *DeviceDataInstance {
	return &DeviceDataInstance{
		deviceDataLastTime: sync.Map{},
	}
}

func (d *DeviceDataInstance) Add(deviceSn, deviceType string, data interface{}) (*domain.DeviceCollection, error) {
	// 获取数据上一次的
	var v interface{}
	var ok bool
	var now = time.Now().Unix()
	var t = now
	if v, ok = d.deviceDataLastTime.Load(deviceSn); ok {
		t = v.(int64)
	} else {
		d.deviceDataLastTime.Store(deviceSn, t)
		redis.SaveDeviceRealTimeData(deviceSn, data, true)
		return nil, errors.New(fmt.Sprintf("ingnore this record"))
	}
	// 60秒接受一次数据,暂时不用太多数据
	if now < t+DefaultReceiveSpan {
		return nil, errors.New(fmt.Sprintf("receive too fast wait %v sec", t+DefaultReceiveSpan-now))
	}
	// 从redis获取最后的数据进行处理
	var result = &domain.DeviceCollection{}
	if err := redis.GetDeviceRealTimeData(deviceSn, result); err != nil {
		if err == domain.ErrorNotFound {
			redis.SaveDeviceRealTimeData(deviceSn, data, true)
			return nil, nil
		}
		return nil, err
	}
	//log.Logger.Debug("",map[string]interface{}{"t":t,"device":deviceSn})
	d.deviceDataLastTime.Store(deviceSn, now)
	return result, redis.SaveDeviceRealTimeData(deviceSn, data, false)
}