adm-dws/impl/task/task1.go

// Copyright 2019 getensh.com. All rights reserved.
// Use of this source code is governed by getensh.com.

package task

import (
	"adm-dws/apis"
	"adm-dws/consts"
	"adm-dws/model"
	"encoding/json"
	"strings"

	"github.com/tidwall/gjson"
	"gorm.io/gorm"
)

func isInDws2(db *gorm.DB,plateNo,plateType,vin string) (bool,error) {
	dws2 := &model.Dws2{}
	where := map[string]interface{}{"plate_no": plateNo,"vin": vin}
	err := dws2.Query(db, where)
	if err == nil {
		if dws2.PlateType == "" && plateType != ""{
			dws2.PlateType = plateType
			dws2.Update(db)
		}
		return true,nil
	}
	if err != gorm.ErrRecordNotFound {
		return false, err
	}

	return false,nil
}

func isInDws1(db *gorm.DB,dws1 *model.Dws1)(bool,error){
	// 判断是否存在于dws1
	where := map[string]interface{}{"plate_no": dws1.PlateNo,"vin":dws1.Vin}
	err := dws1.Query(db, where)
	if err == nil {
		// 有数据，直接返回
		return true,nil
	}else {
		// 没有数据
		if err == gorm.ErrRecordNotFound{
			return false,nil
		}else{
			// 数据库错误
			return false,err
		}
	}
}

func handlePlateTypeExist(dws1 *model.Dws1,db *gorm.DB) (bool,error) {
	// 删除dws4中数据
	dws4 := &model.Dws4{}
	where := map[string]interface{}{"plate_no": dws1.PlateNo,"vin":dws1.Vin}
	err := dws4.Delete(db,where)
	if err != nil {
		if err != gorm.ErrRecordNotFound {
			return false,err
		}
	}

	// 插入或更新dws1
	// 通过车牌查询
	where = map[string]interface{}{"plate_no": dws1.PlateNo,"plate_type":dws1.PlateType}
	oldDws1 := &model.Dws1{}
	err = oldDws1.Query(db, where)
	if err != nil {
		// 没有数据
		if err == gorm.ErrRecordNotFound { // 没有数据则插入
		    err = dws1.Insert(db)
		    if err != nil {
		    	if !strings.Contains(err.Error(), "Duplicate") {
		    		return  false,err
		        }
			}
		}
	} else { // 有数据，判断是否一样
		if oldDws1.Vin == dws1.Vin {
			return true,nil
		} else {
			// 更新时间大于消息时间表示是后面的消息，不处理
			if oldDws1.UpdatedAt > dws1.UpdatedAt{
				return true,nil
			}
			// vin 码不一样
			// 插入过期数据
			dws2 := &model.Dws2{}
			dws2.PlateNo = oldDws1.PlateNo
			dws2.PlateType = oldDws1.PlateType
			dws2.Vin = oldDws1.Vin
			err = dws2.Insert(db)
			if err != nil {
				if !strings.Contains(err.Error(), "Duplicate") {
					return false,err
				}
			}
			// 更新为新数据
			oldDws1.Vin = dws1.Vin
			oldDws1.UpdatedAt = dws1.UpdatedAt
			err = oldDws1.Update(db)
			if err != nil{
				if !strings.Contains(err.Error(), "Duplicate") {
					return false,err
				}
			}
		}
	}
	return false,nil
}


// 车牌和vin码对应关系(正确关系)
func Dws1Task(db *gorm.DB,dwsMessage *apis.DwsMessage, outputSourceCode string) (adsMsgList []*apis.AdsMessage, dwsMsgList []*apis.DwsMessage, err error) {
	dws1 := &model.Dws1{}
	err = json.Unmarshal([]byte(dwsMessage.Content), dws1)
	if err != nil {
		return nil, nil, nil
	}

	dws1.UpdatedAt = dwsMessage.Timestamp
	dws1.CreatedAt = dwsMessage.Timestamp
	// 参数判断
	if dws1.PlateNo == "" || dws1.Vin == ""{
		return nil ,nil ,nil
	}
	isExpire := gjson.Get(dwsMessage.Content, "is_expire").String()
	if isExpire != consts.EXPIREDATA {
		// 计算号牌种类
		if dws1.PlateType == ""{
			dws1.PlateType = calcPlateType(dws1.Long,dws1.ApprovedNumber,dws1.GrossMass)
			if dws1.PlateType == ""{
				dws1.PlateType = CalcPlateType(db,dws1.Vin)
			}
		}

		if dws1.PlateType == ""{
			// 号牌种类为空
			// 判断是否在dws1
			isInDws1 ,err := isInDws1(db,dws1)
			if err != nil{
				return nil,nil,err
			}

			if isInDws1{
				return nil,nil,nil
			}

			// 发送到dws4
			dwsMsg := NewDwsMessage(dwsMessage,outputSourceCode)
			msgByte, _ := json.Marshal(*dws1)
			dwsMsg.Content = string(msgByte)
			adsMsgs,_,err := Dws4Task(db,dwsMsg, consts.SOURCEDWS4)
			if err != nil{
				return nil,nil,err
			}
			adsMsgList = append(adsMsgList,adsMsgs...)
		}else{
			// 号牌种类不为空
			// 判断错误关系表中是否存在，存在直接返回
			isInDws2 ,err := isInDws2(db,dws1.PlateNo,dws1.PlateType,dws1.Vin)
			if err != nil{
				return nil,nil,err
			}
			if isInDws2 {
				return nil,nil,nil
			}

			isExist ,err := handlePlateTypeExist(dws1,db)
			if err != nil{
				return nil,nil,err
			}

			// 已经存在不同步到ads
			if isExist{
				return nil,nil,nil
			}

			// 发送到ads
			adsMsg := NewAdsMessage(dwsMessage, outputSourceCode, consts.ACTIONINSERT)
			msgByte, _ := json.Marshal(*dws1)
			adsMsg.Content = string(msgByte)
			adsMsgList = append(adsMsgList,adsMsg)
		}
	}
	return adsMsgList, nil, nil
}