nbi_alarm/service/sysn_alarm_file.go

package service

import (
	"bytes"
	"encoding/binary"
	"encoding/json"
	"errors"
	"github.com/aceld/zinx/ziface"
	"omc/db"
	"omc/lib"
	"omc/model"
	"omc/omc"
	"time"
)

func GenFile(request ziface.IRequest, meta *lib.FileMeta, data []OmcAlarm) {
	//生成文件内容
	dataBuff := bytes.NewBuffer([]byte{})
	for _, v := range data {
		b, _ := json.Marshal(v)
		binary.Write(dataBuff, binary.BigEndian, b)
		binary.Write(dataBuff, binary.BigEndian, '\r')
		binary.Write(dataBuff, binary.BigEndian, '\n')
	}

	file, err := lib.GenFile(meta, dataBuff.Bytes())
	if err != nil {
		return
	}

	//发送文件同步信息
	ackBody := omc.MsgBody{
		MsgName: "ackSyncOmcAlarmFileResult",
		Msg:     make(map[string]string, 0),
	}
	ackBody.Msg["reqId"] = meta.ReqId
	ackBody.Msg["result"] = "succ"
	ackBody.Msg["fileName"] = file
	ackBody.Msg["resDesc"] = ""
	ackBody.Pack()
	request.GetConnection().SendMsg(omc.AckSyncAlarmFileResult, ackBody.RawData)
}

// GetAlarmOfAlarmSeq 获取告警信息
func GetAlarmOfAlarmSeq(neType, neId string, alarmSeq int) ([]OmcAlarm, error) {
	var alarms []model.Alarm
	var result []OmcAlarm

	query := db.Client.Model(&model.Alarm{}).Where("ne_type = ? and ne_id = ? and alarm_seq > ?", neType, neId, alarmSeq)
	if err := query.Order("alarm_seq asc").Find(&alarms).Error; err != nil {
		return nil, err
	}
	for _, v := range alarms {
		var item OmcAlarm
		item.AlarmSeq = int32(v.AlarmSeq)
		item.AlarmTitle = v.AlarmTitle
		item.AlarmStatus = int32(v.AlarmStatus)
		item.AlarmType = v.AlarmType
		item.OrigSeverity = omc.OrigSeverity(v.OrigSeverity)
		item.EventTime = v.EventTime.Format("2006-01-02 15:04:05")
		item.AlarmId = v.AlarmId
		item.SpecificProblemID = v.SpecificProblemID
		item.SpecificProblem = v.SpecificProblem
		item.NeUID = v.NeId
		item.NeName = v.NeName
		item.NeType = v.NeType
		item.ObjectUID = v.ObjectUID
		item.ObjectName = v.NeName
		item.ObjectType = v.ObjectType
		item.LocationInfo = v.LocationInfo
		item.AddInfo = v.AddInfo
		item.PVFlag = v.PVFlag
		item.Province = v.Province
		result = append(result, item)
	}
	return result, nil
}

//GetAlarm

/*
1   如果syncSource=1 &&  alarmSeq 为空:   从北向告警上报日志中(nbi_alarm_log)取数据ID，然后反查告警信息表(alarm)取出告警日志

2   其他情况:   从告警信息表中取数据， 数据来源为设备告警事件

*/

func GetAlarm(neType, neId, startTime, endTime, syncSource string, alarmSeq int) ([]OmcAlarm, error) {
	if syncSource == "0" {
		return GetAlarmOfEventTime(neType, neId, startTime, endTime)

	} else {
		if alarmSeq > 0 {
			return GetAlarmOfAlarmSeq(neType, neId, alarmSeq)
		} else {
			return GetAlarmOfLog(neType, neId, startTime, endTime)
		}
	}
}

// GetAlarmOfEventTime 获取告警信息
func GetAlarmOfEventTime(neType, neId, startTime, endTime string) ([]OmcAlarm, error) {
	var alarms []model.Alarm
	var result []OmcAlarm

	if startTime == "" && endTime == "" {
		return result, nil
	}
	query := db.Client.Model(&model.Alarm{}).Where("ne_type = ? and ne_id = ?", neType, neId)
	if startTime != "" {
		t1, err := time.Parse("2006-01-02 15:04:05", startTime)
		if err != nil {
			return nil, errors.New("startTime invalid")
		}
		query = query.Where("event_time > ?", t1)
	}
	if endTime != "" {
		t2, err := time.Parse("2006-01-02 15:04:05", endTime)
		if err != nil {
			return nil, errors.New("endTime invalid")
		}
		query = query.Where("event_time < ?", t2)
	}
	if err := query.Order("alarm_seq asc").Find(&alarms).Error; err != nil {
		return nil, err
	}
	for _, v := range alarms {
		var item OmcAlarm
		item.AlarmSeq = int32(v.AlarmSeq)
		item.AlarmTitle = v.AlarmTitle
		item.AlarmStatus = int32(v.AlarmStatus)
		item.AlarmType = v.AlarmType
		item.OrigSeverity = omc.OrigSeverity(v.OrigSeverity)
		item.EventTime = v.EventTime.Format("2006-01-02 15:04:05")
		item.AlarmId = v.AlarmId
		item.SpecificProblemID = v.SpecificProblemID
		item.SpecificProblem = v.SpecificProblem
		item.NeUID = v.NeId
		item.NeName = v.NeName
		item.NeType = v.NeType
		item.ObjectUID = v.ObjectUID
		item.ObjectName = v.NeName
		item.ObjectType = v.ObjectType
		item.LocationInfo = v.LocationInfo
		item.AddInfo = v.AddInfo
		item.PVFlag = v.PVFlag
		item.Province = v.Province
		result = append(result, item)
	}
	return result, nil
}

// GetAlarmOfLog  获取告警信息
func GetAlarmOfLog(neType, neId, startTime, endTime string) ([]OmcAlarm, error) {
	var alarms []model.Alarm
	var result []OmcAlarm

	if startTime == "" && endTime == "" {
		return result, nil
	}
	var aIDs []int64
	query := db.Client.Model(&model.NbiAlarmLog{}).Select("distinct a_id").Where("ne_type = ? and ne_id = ?", neType, neId)
	if startTime != "" {
		t1, err := time.Parse("2006-01-02 15:04:05", startTime)
		if err != nil {
			return nil, errors.New("startTime invalid")
		}
		query = query.Where("log_time >= ?", t1)
	}
	if endTime != "" {
		t2, err := time.Parse("2006-01-02 15:04:05", endTime)
		if err != nil {
			return nil, errors.New("endTime invalid")
		}
		query = query.Where("log_time <= ?", t2)
	}
	if err := query.Order("alarm_seq asc").Find(&aIDs).Error; err != nil {
		return nil, err
	}

	if err := db.Client.Model(&model.Alarm{}).Where("id in (?)", aIDs).Find(&alarms).Error; err != nil {
		return nil, err
	}
	for _, v := range alarms {
		var item OmcAlarm
		item.AlarmSeq = int32(v.AlarmSeq)
		item.AlarmTitle = v.AlarmTitle
		item.AlarmStatus = int32(v.AlarmStatus)
		item.AlarmType = v.AlarmType
		item.OrigSeverity = omc.OrigSeverity(v.OrigSeverity)
		item.EventTime = v.EventTime.Format("2006-01-02 15:04:05")
		item.AlarmId = v.AlarmId
		item.SpecificProblemID = v.SpecificProblemID
		item.SpecificProblem = v.SpecificProblem
		item.NeUID = v.NeId
		item.NeName = v.NeName
		item.NeType = v.NeType
		item.ObjectUID = v.ObjectUID
		item.ObjectName = v.NeName
		item.ObjectType = v.ObjectType
		item.LocationInfo = v.LocationInfo
		item.AddInfo = v.AddInfo
		item.PVFlag = v.PVFlag
		item.Province = v.Province
		result = append(result, item)
	}
	return result, nil
}