package com.gyee.impala.model.master;

import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.extension.activerecord.Model;
import com.gyee.impala.common.spring.InitialRunner;
import com.gyee.impala.model.ToData;
import com.gyee.impala.model.custom.AlertHistory;
import lombok.Data;
import lombok.EqualsAndHashCode;

import java.io.Serializable;

@Data
@EqualsAndHashCode(callSuper = false)
@TableName("casewarningscada")
public class Casewarningscada extends Model<Casewarningscada> implements ToData<Object> {

    private static final long serialVersionUID = 1L;

    private String id;
    private String stationen;
    private String windturbineid;
    private String warntype;
    private String warncode;
    private String warndes;
    private String warntime;
    private int warnrank;
    private String model;
    private String projectid;
    private String lineid;
    private String stationcn;
    private String windturbinename;
    private String remark;
    private String category;

    @Override
    protected Serializable pkVal() {
        return this.id;
    }

    @Override
    public Object toData(Object t) {
        AlertHistory obj = (AlertHistory) t;
        if (obj == null)
            return null;

        Casewarningscada warn = new Casewarningscada();
        warn.setId(String.valueOf(obj.getId()));
        warn.setStationcn(obj.getStationName());
        warn.setStationen(obj.getStationId());
        warn.setWindturbineid(obj.getWindturbineId());
        warn.setWindturbinename(obj.getWindturbineName());
        warn.setWarntype(InitialRunner.scadaWarnMap.containsKey(obj.getCategory2().toUpperCase()) ? InitialRunner.scadaWarnMap.get(obj.getCategory2().toUpperCase()) : "");
        warn.setWarncode(obj.getCategory2());
        warn.setWarndes(obj.getAlertText());
        warn.setWarntime(obj.getAlertTime());
        warn.setWarnrank(Integer.valueOf(obj.getRank()));
        warn.setModel(obj.getModelId());
        warn.setProjectid(obj.getProjectId());
        warn.setLineid(obj.getLineId());
        warn.setRemark(null);
        warn.setCategory(null);

        return warn;
    }
}