完善楼层的bug

zhangzengfei

2024-04-20 a25d8a11e6955d44bb51ac73799ab82acd2c62a7

完善楼层的bug

 main.go

@@ -4,7 +4,6 @@
    "data_msg_push_server/config"
    "data_msg_push_server/service"

    "fmt"
    "time"

    "basic.com/valib/logger.git"

 model/model.go

@@ -11,3 +11,20 @@
    PicMaxImages: nil,
    PicSmImages:  nil,
}

var PushDataInfoV2 = &pushDataInfoV2{
    PicMaxImages: nil,
    PicSmImages:  nil,
    CameraFloor:  "",
    DataSource:   "smart-ai",
}

type pushDataInfoV2 struct {
    CameraId     string   `json:"cameraId"`
    CameraFloor  string   `json:"cameraFloor"` //摄像机楼层
    PicDate      string   `json:"picDate"`
    PicId        string   `json:"picId"`
    DataSource   string   `json:"dataSource"`
    PicMaxImages [][]byte `json:"picMaxImages"`
    PicSmImages  [][]byte `json:"picSmImages"`
}

 service/server.go

@@ -12,6 +12,8 @@
    "fmt"
    "io/ioutil"
    "net/http"
    "regexp"
    "strings"
)

func ConnectControl() bool {
@@ -145,13 +147,41 @@
    }

    // 上报机构名称
    source[0]["orgName"] = config.Options.OrgName
    //source[0]["orgName"] = config.Options.OrgName

    model.PushDataInfo.SourceData = source[0]
    model.PushDataInfo.PicMaxImages = picMaxImages
    model.PushDataInfo.PicSmImages = picSmImages
    cameraAddr := source[0]["cameraAddr"].(string)
    if len(cameraAddr) > 6 {
        if cameraAddr[len(cameraAddr)-3:] == "楼" {
            cameraAddr = cameraAddr[:len(cameraAddr)-3] + "F"
        }
        if cameraAddr[len(cameraAddr)-3:] == "层" {
            cameraAddr = cameraAddr[:len(cameraAddr)-3] + "F"
        }
    }

    if strings.HasSuffix(cameraAddr, "F") {
        model.PushDataInfoV2.CameraFloor = extractFloor(cameraAddr)
    } else {
        model.PushDataInfoV2.CameraFloor = ""
    }

    model.PushDataInfoV2.CameraId = source[0]["cameraId"].(string)
    model.PushDataInfoV2.PicDate = source[0]["picDate"].(string)
    model.PushDataInfoV2.PicId = source[0]["id"].(string)
    model.PushDataInfoV2.PicMaxImages = picMaxImages
    model.PushDataInfoV2.PicSmImages = picSmImages
    //fmt.Println(model.PushDataInfo.SourceData)
    return model.PushDataInfo, nil
    return model.PushDataInfoV2, nil
}

func extractFloor(s string) string {
    // 使用正则表达式匹配楼层信息
    re := regexp.MustCompile(`[-\d]+F`)
    matches := re.FindStringSubmatch(s)
    if len(matches) >= 1 {
        return matches[0] // 返回匹配到的楼层信息
    }
    return "" // 如果没有匹配到，返回空字符串
}

func SendData(pushDataInfo interface{}, url string) (id string, err error) {