123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690 |
- package map_service
- import (
- "archive/zip"
- "context"
- "encoding/json"
- "fmt"
- "github.com/cloudwego/hertz/pkg/app"
- "github.com/cloudwego/hertz/pkg/protocol/consts"
- uuid "github.com/satori/go.uuid"
- "io"
- "io/ioutil"
- "net/http"
- "os"
- "path/filepath"
- "pji_desktop_http/biz/dal/mysql"
- "pji_desktop_http/biz/model"
- "pji_desktop_http/common/config"
- "pji_desktop_http/common/config/c_log"
- "pji_desktop_http/common/config/c_pji"
- "pji_desktop_http/common/entity"
- "pji_desktop_http/common/util"
- "pji_desktop_http/pji_client"
- "strconv"
- "strings"
- )
- // CheckMapBufConsistency 检查请求id对应的mapBuf文件夹的一致性
- // @router /map/checkmapbuf [POST]
- func CheckMapBufConsistency(ctx context.Context, c *app.RequestContext) {
- var req []string
- err := c.BindAndValidate(&req)
- if err != nil {
- c.String(consts.StatusBadRequest, err.Error())
- return
- }
- fmt.Println(req)
- if len(req) == 0 { // 请求数据为空
- c.JSON(consts.StatusBadRequest, entity.HttpResult{Status: false, Code: "", Message: "请求数据为空。"})
- } else if len(req) == 1 { // 只有一条数据则直接返回
- c.JSON(consts.StatusOK, entity.HttpResult{Status: true, Code: "", Message: "mapBuf文件夹数据一致。"})
- } else {
- var firstValue int
- for i, id := range req {
- // 根据id获取对应的oss文件列表
- fileList, err := util.GetExactedMapFileById(id)
- // 过滤特定后缀的文件列表
- fileList = util.FilterBySuffixes(fileList, config.MapBufFiltersuffixes...)
- //fmt.Println("Filtered Strings:", fileList)
- if err != nil {
- c.String(consts.StatusBadRequest, err.Error())
- return
- }
- // 获取文件列表的总大小
- totalSize := calculateTotalFileSize(fileList)
- fmt.Println("Total Size:", totalSize)
- // 判断不同文件列表(mapBuf)中文件的总大小是否一致
- if i == 0 {
- firstValue = totalSize
- } else {
- if totalSize != firstValue {
- c.JSON(consts.StatusOK, entity.HttpResult{Status: false, Code: "", Message: "mapBuf文件夹数据不一致。"})
- return
- }
- }
- }
- c.JSON(consts.StatusOK, entity.HttpResult{Status: true, Code: "", Message: "mapBuf文件夹数据一致。"})
- }
- }
- // DownloadOSSFile 根据objectKey下载指定的oss文件
- // @router /map/downloadossfile [GET]
- func DownloadOSSFile(ctx context.Context, c *app.RequestContext) {
- objectKey := c.Query("objectKey")
- // 从OSS下载文件
- reader, err := config.OssBucket.GetObject(objectKey)
- if err != nil {
- c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
- return
- }
- defer reader.Close()
- // 设置响应头
- c.Response.Header.Set("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, filepath.Base(objectKey)))
- c.Response.Header.Set("Content-Type", "binary/octet-stream")
- // 将文件流式传输回客户端
- data, err := io.ReadAll(reader)
- if err != nil {
- panic(err)
- }
- if _, err := c.Write(data); err != nil {
- c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
- return
- }
- }
- // DownloadMapZipFile 根据请求id从oss拉取并打包下载建图所需要的文件
- // @router /map/download/map/zip [GET]
- func DownloadMapZipFile(ctx context.Context, c *app.RequestContext) {
- id := c.Query("id")
- fmt.Println("id", id)
- ids := strings.Split(id, ",")
- fmt.Println("ids", ids)
- // 根据id生成用于地图更新的压缩包
- filePath, tmpDir, err := generateMapZipById(ids)
- if err != nil {
- c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
- }
- fmt.Println("filePath", filePath)
- // 检查文件是否存在
- if _, err := os.Stat(filePath); os.IsNotExist(err) {
- c.JSON(http.StatusNotFound, map[string]string{"error": "File not found"})
- return
- }
- // 打开文件
- f, err := os.Open(filePath)
- if err != nil {
- c.JSON(http.StatusInternalServerError, map[string]string{"error": "Failed to open file"})
- return
- }
- defer f.Close()
- // 设置响应头
- c.Response.Header.Set("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, filepath.Base(filePath)))
- c.Response.Header.Set("Content-Type", "binary/octet-stream")
- // 将文件流式传输回客户端
- data, err := io.ReadAll(f)
- if err != nil {
- panic(err)
- }
- if _, err := c.Write(data); err != nil {
- c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
- return
- }
- defer os.RemoveAll(tmpDir)
- }
- // DownloadMapBagFile 根据请求id从oss拉取并下载解析后的map.bag文件
- // @router /map/downloadmapbagfile [GET]
- func DownloadMapBagFile(ctx context.Context, c *app.RequestContext) {
- id := c.Query("id")
- fmt.Println("id: ", id)
- // 根据id获取对应的oss文件列表
- fileList, err := util.GetExactedMapFileById(id)
- // 过滤特定后缀的文件列表
- fileList = util.FilterBySuffixes(fileList, config.MapBagFiltersuffixes...)
- //fmt.Println("fileList", fileList)
- objectKey := fileList[0]
- // 从OSS下载文件
- reader, err := config.OssBucket.GetObject(objectKey)
- if err != nil {
- c.JSON(consts.StatusInternalServerError, entity.HttpResult{Status: false, Code: "", Message: "解析地图Bag下载失败。"})
- return
- }
- defer reader.Close()
- // 设置响应头
- c.Response.Header.Set("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, filepath.Base(objectKey)))
- c.Response.Header.Set("Content-Type", "binary/octet-stream")
- // 将文件流式传输回客户端
- data, err := io.ReadAll(reader)
- if err != nil {
- panic(err)
- }
- if _, err := c.Write(data); err != nil {
- c.JSON(consts.StatusInternalServerError, entity.HttpResult{Status: false, Code: "", Message: "解析地图Bag下载失败。"})
- return
- }
- c.JSON(consts.StatusOK, entity.HttpResult{Status: true, Code: "", Message: "解析地图Bag下载成功。"})
- }
- // 根据id生成用于地图更新的压缩包
- func generateMapZipById(ids []string) (file string, tmpDir string, err error) {
- // 根据id获取对应的oss文件列表
- allFileList, err := util.GetExactedMapFileByIds(ids)
- fmt.Println("allFileList", allFileList)
- // 过滤特定后缀的文件列表
- mapBufFileList := util.FilterBySuffixes(allFileList, config.MapBufFiltersuffixes...)
- //fmt.Println("Filtered Strings:", fileList)
- if err != nil {
- return
- }
- // 创建临时文件夹
- tmpDir, err = os.MkdirTemp("", "temp-download-*")
- fmt.Println("tmpDir:", tmpDir)
- if err != nil {
- fmt.Println("Error creating temporary directory:", err)
- return "", "", err
- }
- c_log.GlobalLogger.Info("创建下载-临时文件夹:", tmpDir)
- // 创建根文件夹(文件打包的根目录)
- baseDir := filepath.Join(tmpDir, "data")
- if err := os.Mkdir(baseDir, 0755); err != nil {
- fmt.Println("Error creating subdirectory:", err)
- return "", "", err
- }
- c_log.GlobalLogger.Info("创建文件打包根目录:", baseDir)
- // 根目录创建子文件夹 bag_folder
- bagFolderDir := filepath.Join(baseDir, "bag_folder")
- if err := os.Mkdir(bagFolderDir, 0755); err != nil {
- fmt.Println("Error creating subdirectory:", err)
- return "", "", err
- }
- c_log.GlobalLogger.Info("根目录下创建bag_folder子文件夹:", bagFolderDir)
- // 根目录创建子文件夹 origin_map_folder
- originMapFolderDir := filepath.Join(baseDir, "origin_map_folder")
- if err := os.Mkdir(originMapFolderDir, 0755); err != nil {
- fmt.Println("Error creating subdirectory:", err)
- return "", "", err
- }
- c_log.GlobalLogger.Info("根目录下创建origin_map_folder子文件夹:", originMapFolderDir)
- // 子文件夹 bag_folder 创建子文件夹 mapBuf
- mapBufDir := filepath.Join(bagFolderDir, "mapBuf")
- if err := os.Mkdir(mapBufDir, 0755); err != nil {
- fmt.Println("Error creating subdirectory:", err)
- return "", "", err
- }
- c_log.GlobalLogger.Info("bag_folder创建mapBuf子文件夹 :", mapBufDir)
- // 从oss下载build_map.bag文件到 bag_folder 文件夹
- // 过滤特定后缀的文件列表
- buildMapBagFileList := util.FilterBySuffixes(allFileList, config.BuildMapBagFiltersuffixes...)
- fmt.Println("buildMapBagFileList", buildMapBagFileList)
- //buildMapBagFile := buildMapBagFileList[0]
- //err = config.OssBucket.GetObjectToFile(buildMapBagFile, filepath.Join(bagFolderDir, filepath.Base(buildMapBagFile)))
- for i, file := range buildMapBagFileList {
- fileName := "build_map" + "-" + strconv.Itoa(i) + ".bag"
- err = config.OssBucket.GetObjectToFile(file, filepath.Join(bagFolderDir, fileName))
- if err != nil {
- fmt.Println("Error downloading build map file:", err)
- return "", "", err
- }
- }
- if err != nil {
- fmt.Println("Error downloading Bag file:", err)
- return "", "", err
- }
- c_log.GlobalLogger.Info("下载文件到bag_folder文件夹 - 成功")
- // 从oss下载文件到 mapBuf 文件夹
- for _, file := range mapBufFileList {
- err = config.OssBucket.GetObjectToFile(file, filepath.Join(mapBufDir, filepath.Base(file)))
- if err != nil {
- fmt.Println("Error downloading mapBuf file:", err)
- return "", "", err
- }
- }
- c_log.GlobalLogger.Info("下载文件到mapBuf子文件夹 - 成功")
- // 从oss下载bag文件到 origin_map_folder 文件夹
- // 过滤特定后缀的文件列表
- originMapFileList := util.FilterBySuffixes(allFileList, config.OriginMapFiltersuffixes...)
- for _, file := range originMapFileList {
- err = config.OssBucket.GetObjectToFile(file, filepath.Join(originMapFolderDir, filepath.Base(file)))
- if err != nil {
- fmt.Println("Error downloading origin_map file:", err)
- return "", "", err
- }
- }
- c_log.GlobalLogger.Info("下载文件到origin_map_folder文件夹 - 成功")
- // 创建压缩文件
- zipPath := filepath.Join(tmpDir, "mapFile-"+ids[0]+".zip")
- zipFile, err := os.Create(zipPath)
- if err != nil {
- fmt.Println("Error creating ZIP file:", err)
- return "", "", err
- }
- defer zipFile.Close()
- zipWriter := zip.NewWriter(zipFile)
- defer zipWriter.Close()
- // 压缩文件夹
- if err := util.AddDirToZip(baseDir, zipWriter); err != nil {
- fmt.Println("Error adding directory to ZIP:", err)
- return "", "", err
- }
- fmt.Println("ZIP file created successfully.")
- c_log.GlobalLogger.Info("创建压缩文件 - 成功")
- return zipPath, tmpDir, nil
- }
- // UploadMapFile 将地图更新相关文件上传到oss
- // @router /map/upload/map [GET]
- func UploadMapFile(ctx context.Context, c *app.RequestContext) {
- equipmentNo := c.Query("equipmentNo")
- fmt.Println("equipmentNo", equipmentNo)
- mapId := c.Query("mapId")
- fmt.Println("mapId", mapId)
- timeStamp := c.Query("timeStamp")
- fmt.Println("timeStamp", timeStamp)
- header, err := c.FormFile("file")
- if err != nil {
- c.String(http.StatusBadRequest, fmt.Sprintf("get form err: %s", err.Error()))
- return
- }
- fileName := header.Filename
- fmt.Println("filename", fileName)
- ossObjectKey := config.UpdateMapOssBasePrefix + "/" + equipmentNo + "/" + mapId + "/" + timeStamp + "/" + fileName
- fmt.Println("ossObjectKey", ossObjectKey)
- f, _ := header.Open()
- defer f.Close()
- config.OssMutex.Lock()
- err = config.OssBucket.PutObject(ossObjectKey, f)
- config.OssMutex.Unlock()
- if err != nil {
- c_log.GlobalLogger.Error("程序异常退出。上传文件", fileName, "->", ossObjectKey, "出错:", err)
- c.JSON(consts.StatusOK, entity.HttpResult{Status: false, Code: "", Message: "上传文件失败", Details: ""})
- return
- }
- c_log.GlobalLogger.Info("上传文件", fileName, "->", ossObjectKey, "成功。")
- c.JSON(consts.StatusOK, entity.HttpResult{Status: true, Code: "", Message: "上传文件成功", Details: ossObjectKey})
- }
- // AddMapUpdateRecord 添加仿真测试记录
- // @router /map/add/record [GET]
- func AddMapUpdateRecord(ctx context.Context, c *app.RequestContext) {
- var record model.MapUpdate
- err := c.BindAndValidate(&record)
- record.ID = uuid.NewV1().String()
- fmt.Println("record", record)
- if err != nil {
- c.String(http.StatusBadRequest, fmt.Sprintf("get form err: %s", err.Error()))
- return
- }
- err = mysql.AddMapUpdateOneRecord(ctx, record)
- if err != nil {
- c.JSON(consts.StatusOK, entity.HttpResult{Status: false, Code: "", Message: "地图更新记录添加失败"})
- return
- }
- c.JSON(consts.StatusOK, entity.HttpResult{Status: true, Code: "", Message: "地图更新记录添加成功"})
- }
- // QueryMapUpdateRecord 根据条件查询地图更新记录
- // @router /map/query/update/record [GET]
- func QueryMapUpdateRecord(ctx context.Context, c *app.RequestContext) {
- var record model.MapUpdate
- err := c.BindAndValidate(&record)
- fmt.Println("record", record)
- var pageFlag bool
- if c.Query("page") != "" && c.Query("pageSize") != "" {
- pageFlag = true
- } else {
- pageFlag = false
- }
- page, _ := strconv.Atoi(c.Query("page"))
- pageSize, _ := strconv.Atoi(c.Query("pageSize"))
- records, count, err := mysql.QueryMapUpdateRecords(ctx, &record, pageFlag, page, pageSize)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "地图更新记录查询失败", Total: 0})
- return
- }
- output, err := json.Marshal(records)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "地图更新记录查询失败", Total: 0})
- return
- }
- c.JSON(consts.StatusOK, entity.Response{Status: true, Code: "", Message: "地图更新查询成功", Data: string(output), Total: int(count)})
- }
- // TestUploadUpdateMap 测试朴津地图上传接口
- // @router /map/upload/update/map [GET]
- func TestUploadUpdateMap(ctx context.Context, c *app.RequestContext) {
- header, err := c.FormFile("file")
- if err != nil {
- c.String(http.StatusBadRequest, fmt.Sprintf("get form err: %s", err.Error()))
- return
- }
- fileName := header.Filename
- fmt.Println("filename", fileName)
- f, _ := header.Open()
- defer f.Close()
- paramMap := make(map[string]interface{})
- // Map<String, String> paramMap=new HashMap<>();
- // paramMap.put("mapId","2247aeb31ad34wwddfd9c3ba24536e52859");
- // paramMap.put("mapName","合肥演示");
- // paramMap.put("buildId"," 7b4e8d01c08f472fabf7ccff2b2336s");
- // paramMap.put("floorId"," 695abbcfdf99471sdfff3b9b1d0818164e7");
- // paramMap.put("updateType"," 7b4e8d01c08f472fabf7ccff2b2336s");
- // paramMap.put("floor","1");
- // paramMap.put("buildName"," 2");
- // paramMap.put("customAreaId"," 355");
- // paramMap.put("snCode","P1YTYD1M233M00224");
- // paramMap.put("fileName","1702284791385137152.zip");
- // paramMap.put("mapType","0");
- paramMap["snCode"] = "P1YTXS1M22AM00001"
- paramMap["mapId"] = "553d9d47f8404734997e11175f0e63f5"
- paramMap["updateType"] = "0"
- paramMap["mapType"] = "0"
- paramMap["mapName"] = "北京办公区20230224"
- bytes, err := io.ReadAll(f)
- if err != nil {
- return
- }
- // 调用 朴津地图上传 接口
- resp, err := pji_client.ApiClient.UploadRequest(c_pji.PjiApiSitBaseUrl+"mapUpload", paramMap, pji_client.MapSecretId, bytes)
- fmt.Println(string(resp.Body()))
- c.JSON(consts.StatusOK, entity.HttpResult{Status: true, Code: "", Message: "上传地图测试"})
- }
- // UpdateDeviceMapById 根据场景id维护设备地图表
- // @router /map/update/deviceMap/record [GET]
- func UpdateDeviceMapById(ctx context.Context, c *app.RequestContext) {
- sceneId := c.Query("sceneId")
- fmt.Println("sceneId", sceneId)
- deviceNo := c.Query("deviceNo")
- fmt.Println("deviceNo", deviceNo)
- deviceName := c.Query("deviceName")
- fmt.Println("deviceName", deviceName)
- deviceType := c.Query("deviceType")
- fmt.Println("deviceType", deviceType)
- // 根据场景id查询地图id
- mapId, err := getMapIdById(sceneId)
- if err != nil || mapId == "" {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "获取mapId失败"})
- return
- }
- fmt.Println("mapId", mapId)
- // 查询朴津地图列表
- mapInfoList, err := getMapInfoList()
- if err != nil || mapInfoList == nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "获取地图列表数据失败"})
- return
- }
- fmt.Println("mapInfoList", mapInfoList)
- // 判断map id是否存在于朴津地图列表中
- mapId = "fee84db56d254d79b85971cb4ab2e7f1" // 存在
- //mapId = "553d9d47f8404734997e11175f0e63f5" // 不存在
- exist, record := checkMapIdExist(mapId, mapInfoList)
- fmt.Println("exist", exist)
- fmt.Println("record", record)
- if !exist { // map id不存在于朴津地图列表中
- // 查询地图设备表中是否存在对应map id且有效的记录
- records, err := mysql.QueryValidDeviceMapByMapId(ctx, mapId)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: " 查询设备地图失败"})
- return
- }
- fmt.Println("records", records)
- if len(records) > 0 { // 存在,
- fmt.Println("设置无效")
- // 将对应的map id的记录设置为无效
- update, err := mysql.UpdateDeviceMapValidFlagByMapId(ctx, mapId, config.MAP_NOT_VALID_FLAG)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "更新设备地图状态失败"})
- return
- }
- fmt.Println("update", update)
- }
- c.JSON(consts.StatusOK, entity.Response{Status: true, Code: "1001", Message: "map id不存在地图列表中"})
- } else { // map id存在于朴津地图列表中
- snCodeList := record.SnCodeList
- for _, snCode := range snCodeList { // snCodeList 代表配置了对应map id地图的机器人sn码列表, 上限为机器人数量
- fmt.Println("snCode", snCode)
- deviceMap := model.DeviceMap{DeviceNo: snCode, DeviceName: deviceName,
- DeviceType: deviceType, MapID: record.MapId, MapType: record.MapType, MapName: record.MapName, FloorID: record.FloorId,
- Floor: strconv.Itoa(record.Floor), BuildID: record.BuildId, BuildName: record.BuildName, MapCreateTime: record.CreateTime, MapUpdateTime: record.UpdateTime,
- MapVersion: strconv.Itoa(record.Version), MapValidFlag: config.MAP_VALID_FLAG, PjiMapZipURL: record.ZipUrl, CustomAreaID: int32(record.CustomAreaId)}
- fmt.Println("deviceMap", deviceMap)
- deviceMap.MapVersion = "123456"
- // 查询[对应设备]是否存在对应map id且有效的记录
- records, err := mysql.QueryValidDeviceMapByDevice(ctx, mapId, snCode)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: " 查询设备地图失败"})
- return
- }
- fmt.Println("records", records)
- if len(records) == 0 { // 对应设备不存在对应map id且有效的记录, 说明该设备部署了新的地图(重采或新的区域地图)
- // 添加记录
- deviceMap.ID = uuid.NewV1().String()
- err = mysql.AddDeviceMapOneRecord(ctx, deviceMap)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "添加设备地图记录失败"})
- return
- }
- } else if len(records) == 1 { // 对应设备存在1个对应map id且有效的记录, 说明该设备之前部署过该地图, 需要检测版本号来判断更新记录(版本号一致)还是添加记录(版本号不一致)
- fmt.Println("record", record)
- if deviceMap.MapVersion == records[0].MapVersion { // 地图列表查询到的版本与设备地图表中版本一致, 则更新记录
- // 更新记录
- deviceMap.ID = records[0].ID
- update, err := mysql.UpdateDeviceMapOneRecord(ctx, deviceMap)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "更新设备地图记录失败"})
- return
- }
- fmt.Println("update", update)
- } else { // 地图列表查询到的版本与设备地图表中版本不一致, 则先将之前的记录状态设置为无效, 然后添加记录
- // 将对应设备存在的对应map id的记录设置为无效
- update, err := mysql.UpdateDeviceMapValidFlagByDevice(ctx, mapId, snCode, config.MAP_NOT_VALID_FLAG)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "更新设备地图状态失败"})
- return
- }
- fmt.Println("update", update)
- // 添加记录
- deviceMap.ID = uuid.NewV1().String()
- err = mysql.AddDeviceMapOneRecord(ctx, deviceMap)
- if err != nil {
- c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "添加设备地图记录失败"})
- return
- }
- }
- }
- //if len(records) > 0 { // 对应设备存在对应map id且有效的记录
- // // 设置之前记录地图标志为无效
- // fmt.Println("设置无效")
- // // 将对应设备存在的对应map id的记录设置为无效
- // update, err := mysql.UpdateDeviceMapValidFlagByDevice(ctx, mapId, snCode, config.MAP_NOT_VALID_FLAG)
- // if err != nil {
- // c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "更新设备地图状态失败"})
- // return
- // }
- // fmt.Println("update", update)
- // // 更新记录
- // mysql.UpdateDeviceMapOneRecord(ctx, deviceMap)
- //} else { // 对应设备不存在对应map id且有效的记录
- // // 添加记录
- // deviceMap.ID = uuid.NewV1().String()
- // err = mysql.AddDeviceMapOneRecord(ctx, deviceMap)
- // if err != nil {
- // c.JSON(consts.StatusOK, entity.Response{Status: false, Code: "", Message: "添加设备地图记录失败"})
- // return
- // }
- //}
- }
- c.JSON(consts.StatusOK, entity.Response{Status: true, Code: "", Message: "设备地图数据更新成功"})
- }
- }
- // 计算oss中文件列表的总大小
- func calculateTotalFileSize(fileList []string) int {
- var totalSize int
- for _, file := range fileList {
- size, err := util.GetOSSFileSize(config.OssBucket, file) // 获取oss中单个文件的大小
- if err != nil {
- return 0
- }
- totalSize += size
- }
- return totalSize
- }
- // 调用 朴津地图列表查询 接口
- func getMapInfoList() ([]model.MapInfo, error) {
- paramMap := make(map[string]interface{})
- paramMap["mapType"] = "0"
- resp, err := pji_client.ApiClient.GetRequest(c_pji.PjiApiSitBaseUrl+"mapList?mapType=0", paramMap, pji_client.MapSecretId)
- if err != nil {
- return nil, err
- }
- //fmt.Println(string(resp.Body()))
- // 解析json响应
- body := resp.Body()
- var mapRes model.MapInfoRes
- err = json.Unmarshal(body, &mapRes)
- if err != nil || mapRes.Code != 200 {
- return nil, err
- }
- //fmt.Println("mapRes", mapRes)
- return mapRes.Data, nil
- }
- // 查询地图id是否存在于地图列表中
- func checkMapIdExist(mapId string, mapInfoList []model.MapInfo) (bool, model.MapInfo) {
- // 转为map
- IdMap := make(map[string]model.MapInfo)
- for _, info := range mapInfoList {
- IdMap[info.MapId] = info
- }
- record, exist := IdMap[mapId]
- return exist, record
- }
- // 根据场景id获取数据采集时的地图id
- func getMapIdById(id string) (string, error) {
- var mapId string
- // 下载map.json
- // 根据id获取对应的oss文件列表
- allFileList, err := util.GetExactedMapFileById(id)
- fmt.Println("allFileList", allFileList)
- // 过滤特定后缀的文件列表
- fileList := util.FilterBySuffixes(allFileList, config.MapJsonFiltersuffixes...)
- fmt.Println("fileList", fileList)
- // 创建临时文件夹
- tmpDir, err := os.MkdirTemp("", "temp-download-*")
- fmt.Println("tmpDir:", tmpDir)
- if err != nil {
- fmt.Println("Error creating temporary directory:", err)
- return "", err
- }
- c_log.GlobalLogger.Info("创建下载-临时文件夹:", tmpDir)
- for _, file := range fileList {
- path := filepath.Join(tmpDir, filepath.Base(file))
- err = config.OssBucket.GetObjectToFile(file, path)
- if err != nil {
- fmt.Println("Error downloading file:", err)
- return "", err
- }
- c_log.GlobalLogger.Info("下载map.json文件 - 成功")
- // 读取json文件 - mapId
- // 打开文件
- file, err := os.Open(path)
- if err != nil {
- fmt.Println("Error opening file:", err)
- return "", err
- }
- defer file.Close()
- // 读取文件内容
- fileData, err := ioutil.ReadAll(file)
- if err != nil {
- fmt.Println("Error reading file:", err)
- return "", err
- }
- // Json转换为map
- var mapData map[string]interface{}
- err = json.Unmarshal(fileData, &mapData)
- data := mapData["map"].(map[string]interface{})
- // 读取mapId
- mapId = data["mapId"].(string)
- fmt.Println("mapId:", mapId)
- }
- return mapId, err
- }
|