smart_streetlights/cloud/ipolesvr/ymlampcontroller.go

package main

import (
	"strconv"
	"time"

	"github.com/go-redis/redis/v7"
	"github.com/sirupsen/logrus"

	"lc/common/models"
	"lc/common/protocol"
	"lc/common/util"
)

type YMLampController struct {
	tenant        string             //租户
	gid           string             //基础数据，网关ID
	did           string             //基础数据，灯控设备编码
	tid           uint16             //基础数据，集控器物模型ID
	errCnt        uint               //连续错误计数
	lampOn        uint8              //开灯状态,1开灯，0关灯，0xFF未知
	state         uint8              //实时数据，0在线,1离线
	lastStateTime time.Time          //实时数据，最新状态时间
	mapData       map[uint16]float64 //实时数据，最新数据
	lastDataTime  time.Time          //实时数据，最新数据时间
	nextHourTime  time.Time          //实时数据
	mapAlarm      map[string]int64   //告警开始时间->数据库记录ID
}

func (o *YMLampController) Set(tenant, did string) {
	if o.did == "" {
		o.state = 0xFF
		o.lampOn = 0xFF
	}
	if o.mapAlarm == nil {
		o.mapAlarm = make(map[string]int64)
	}
	o.did = did
	o.tenant = tenant
}

func (o *YMLampController) HandleData(gid string, tid uint16, t time.Time, mapData *protocol.CHZB_LampData) {
	o.gid = gid
	o.tid = tid
	if o.nextHourTime.IsZero() {
		o.nextHourTime = protocol.ToBJTime(util.BeginningOfHour().Add(1 * time.Hour))
	}
	if o.mapData == nil {
		o.mapData = make(map[uint16]float64)
	}

	if len(mapData.Data) > 0 {
		if v, ok := mapData.Data[1]; ok {
			o.handleLampEvent(t, v)
		}
		for k, v := range mapData.Data {
			o.mapData[k] = v
		}
		//需要告警，则推入告警管理器
		bv := BizValue{ID: o.did, Time: t, Tid: tid, Data: o.mapData}
		GetBizAlarmMgr().PushData(&bv)

		o.lastDataTime = t
		cacheData(o.did, t, mapData.Data)
		o.errCnt = 0
	} else {
		o.errCnt++
	}

	o.handleStateChange(t)
	cacheState(o.did, protocol.ToBJTime(t).Format("2006-01-02 15:04:05"), o.state)

	o.checkSaveData()
}

func (o *YMLampController) UpdateState() {
	if o.lastStateTime.IsZero() {
		t0, s0, err := getState(o.did)
		if err == nil {
			o.state = s0
			o.lastStateTime = t0
		} else {
			err := redisCltRawData.HGet(DeviceAlarmId, o.did).Err()
			if err == redis.Nil {
				o.state = protocol.FAILED
				o.lastStateTime = util.MlNow()
				GetEventMgr().PushEvent(&EventObject{ID: o.did, EventType: models.ET_OFFLINE, Time: o.lastStateTime})
			}
		}
	}

	if o.state == protocol.FAILED ||
		(!o.lastDataTime.IsZero() && util.MlNow().Sub(o.lastDataTime).Minutes() < OfflineInterval) {
		return
	}
	//如果之前一直是在线状态的，则置为离线；若之前是离线状态的，则不修改状态
	if o.state == protocol.SUCCESS {
		o.state = protocol.FAILED
		o.lastStateTime = util.MlNow()
		GetEventMgr().PushEvent(&EventObject{ID: o.did, EventType: models.ET_OFFLINE, Time: o.lastStateTime})
		cacheState(o.did, o.lastStateTime.Format("2006-01-02 15:04:05"), o.state)
		//检查是否要缓存数据
		o.checkSaveData()
	}
}

func (o *YMLampController) handleLampEvent(t time.Time, lampOn float64) {
	//开关灯状态
	if o.lampOn == 0xFF {
		if lampOn > 0 {
			o.lampOn = 1
		} else {
			o.lampOn = 0
		}
		return
	}
	if lampOn > 0 && o.lampOn == 0 { //熄灯状态到开灯状态
		GetEventMgr().PushEvent(&EventObject{ID: o.did, EventType: models.ET_ONLAMP, Time: t, Value: lampOn})
	} else if lampOn == 0 && o.lampOn == 1 { //开灯状态到熄灯状态
		GetEventMgr().PushEvent(&EventObject{ID: o.did, EventType: models.ET_OFFLAMP, Time: t, Value: lampOn})
	}
	//更新状态
	if lampOn > 0 {
		o.lampOn = 1
	} else {
		o.lampOn = 0
	}
}

func (o *YMLampController) handleStateChange(t time.Time) {
	//最新状态
	state := uint8(0)
	if o.errCnt >= 10 {
		state = 1
	} else if o.errCnt == 0 {
		state = 0
	} else {
		return
	}

	//状态处理
	if o.lastStateTime.IsZero() {
		t0, s0, err := getState(o.did)
		if err != nil {
			o.state = state
			o.lastStateTime = t
			return
		}
		o.state = s0
		o.lastStateTime = t0
	}
	if o.state == 0 && state == 1 { //在线->离线
		GetEventMgr().PushEvent(&EventObject{ID: o.did, EventType: models.ET_OFFLINE, Time: t})
	} else if o.state == 1 && state == 0 { //离线->在线
		GetEventMgr().PushEvent(&EventObject{ID: o.did, EventType: models.ET_ONLINE, Time: t})
	}
	o.state = state
	o.lastStateTime = t
}

func (o *YMLampController) HandleAlarm(alarm *protocol.LampAlarm) {
	if o.mapAlarm == nil {
		o.mapAlarm = make(map[string]int64)
	}
	key := o.did + "@" + alarm.StartTime
	if alarm.EndTime != "" { //告警结束
		if t, err := util.MlParseTime(alarm.EndTime); err == nil {
			aid, ok := o.mapAlarm[key]
			if !ok {
				if str, err := redisCltRawData.HGet(DeviceAlarmId, key).Result(); err == nil {
					if id, err := strconv.Atoi(str); err == nil {
						aid = int64(id)
					}
				}
			}
			if aid > 0 {
				oo := models.DeviceAlarm{ID: aid, TEnd: t, EValue: float32(alarm.Brightness)}
				if err := oo.Update(); err != nil {
					logrus.Errorf("更新告警[%d]信息失败:%s", aid, err.Error())
				}
				delete(o.mapAlarm, key)
				if err := redisCltRawData.HDel(DeviceAlarmId, key).Err(); err != nil {
					logrus.Errorf("更新告警[%d]信息失败:%s", aid, err.Error())
				}
			}
		}
	} else { //告警开始
		t, err := util.MlParseTime(alarm.StartTime)
		if err != nil {
			return
		}
		oo := models.DeviceAlarm{
			DID:       alarm.DID,
			TStart:    t,
			Threshold: 0,
			SValue:    float32(alarm.AlarmBrightness),
			Content:   ConvertExcept(alarm.AlarmType),
			AlarmType: alarm.AlarmType,
			Level:     1,
		}
		if err := models.G_db.Create(&oo).Error; err != nil {
			logrus.Errorf("告警信息[%v]入库失败:%s", alarm, err.Error())
		} else {
			o.mapAlarm[key] = oo.ID
			if err := redisCltRawData.HSet(DeviceAlarmId, key, oo.ID).Err(); err != nil {
				logrus.Errorf("设备[%s]告警数据[%d]缓存失败:%s", o.did, oo.ID, err.Error())
			}
		}
	}
}

func (o *YMLampController) checkSaveData() {
	if o.nextHourTime.Before(o.lastDataTime) {
		//判断小时是否一样，不一样则以数据时间为准
		if !util.New(o.nextHourTime).BeginningOfHour().Equal(util.New(o.lastDataTime).BeginningOfHour()) {
			o.nextHourTime = util.New(o.lastDataTime).BeginningOfHour()
		}
		if len(o.mapData) > 0 {
			var datas []models.DeviceHourData
			for k, v := range o.mapData {
				o := models.DeviceHourData{ID: o.did, Sid: k, Val: float32(v), Time: o.nextHourTime, CreatedAt: time.Now()}
				datas = append(datas, o)
			}
			if err := models.MultiInsertDeviceHourData(datas); err != nil {
				//TODO 是否考虑备份数据
				logrus.Errorf("设备[%s]小时[%s]数据插入数据库失败:%s", o.did, o.nextHourTime.Format("2006-01-02 15:04:05"), err.Error())
			}
		}
		o.nextHourTime = o.nextHourTime.Add(time.Hour)
		o.mapData = make(map[uint16]float64)
	}
}