package service
|
|
import (
|
"basic.com/pubsub/esutil.git"
|
"basic.com/valib/logger.git"
|
"bytes"
|
"data_msg_push_server/config"
|
"data_msg_push_server/model"
|
"data_msg_push_server/util"
|
"encoding/json"
|
"errors"
|
"fmt"
|
"io/ioutil"
|
"net/http"
|
"regexp"
|
)
|
|
func ConnectControl() bool {
|
url := config.Options.ServerUrl
|
resp, err := http.Get(url)
|
if err != nil {
|
fmt.Println("连接失败", err)
|
return false
|
}
|
defer resp.Body.Close()
|
|
fmt.Println("连接成功!")
|
return true
|
}
|
|
func GetTotal() (total int, err error) {
|
url := "http://" + config.Options.EsUrl + "/ai_face_ocean/_search"
|
queryDSL := `{
|
"size": "1",
|
"query": {
|
"bool": {
|
"filter": [
|
{
|
"term": {
|
"isDelete": false
|
}
|
}
|
]
|
}
|
},
|
"sort": [
|
{
|
"updateTime": {
|
"order": "asc"
|
}
|
}
|
]
|
}`
|
buf, err := esutil.EsReq("POST", url, []byte(queryDSL))
|
if err != nil {
|
return 0, err
|
}
|
totals, err := util.SourceTotal(buf)
|
if err != nil {
|
return 0, err
|
}
|
if totals == 0 {
|
return 0, nil
|
}
|
return totals, nil
|
}
|
|
func GetData() (*model.PushDataInfoV2, error) {
|
//fmt.Println(config.Options.EsUrl)
|
//fmt.Println(config.Options.ServerUrl)
|
url := "http://" + config.Options.EsUrl + "/ai_face_ocean/_search"
|
queryDSL := `{
|
"size": "1",
|
"query": {
|
"bool": {
|
"filter": [
|
{
|
"term": {
|
"isDelete": false
|
}
|
}
|
]
|
}
|
},
|
"sort": [
|
{
|
"updateTime": {
|
"order": "asc"
|
}
|
}
|
]
|
}`
|
|
buf, err := esutil.EsReq("POST", url, []byte(queryDSL))
|
if err != nil {
|
return nil, err
|
}
|
source, err := util.Sourcelist(buf)
|
if err != nil {
|
return nil, err
|
}
|
if len(source) == 0 {
|
return nil, nil
|
}
|
|
b, _ := json.Marshal(source[0])
|
|
var esInfoV2 model.ESInfoV2
|
err = json.Unmarshal(b, &esInfoV2)
|
if err != nil {
|
return nil, err
|
}
|
|
picMaxImages := make([]byte, 0) // 存储图片数据的数组
|
if esInfoV2.PicMaxUrl != "" {
|
picMaxImages, err = GetImageData("http://" + esInfoV2.PicMaxUrl)
|
if err != nil {
|
logger.Warn("获取大图数据失败:", err.Error())
|
}
|
}
|
|
if len(picMaxImages) == 0 {
|
return nil, errors.New("获取大图失败, id:" + esInfoV2.TargetInfo.PicSmUrl)
|
}
|
|
//fmt.Println("图片数据数组大小:", len(picMaxImages))
|
|
picSmImages := make([]byte, 0) // 存储图片数据的数组
|
if esInfoV2.TargetInfo.PicSmUrl != "" {
|
picSmImages, err = GetImageData("http://" + esInfoV2.TargetInfo.PicSmUrl)
|
if err != nil {
|
fmt.Println("获取小图数据失败:", err)
|
}
|
}
|
|
if len(picSmImages) == 0 {
|
return nil, errors.New("获取小图数据失败, id:" + esInfoV2.Id)
|
}
|
|
var pushData = new(model.PushDataInfoV2)
|
pushData.CameraId = esInfoV2.CameraId
|
pushData.PicDate = esInfoV2.PicDate
|
pushData.PicId = esInfoV2.Id
|
pushData.CameraFloor = esInfoV2.CameraLocation.Floor
|
pushData.Direction = esInfoV2.CameraLocation.Direction
|
pushData.PicMaxImages = append(pushData.PicMaxImages, picMaxImages)
|
pushData.PicSmImages = append(pushData.PicSmImages, picSmImages)
|
//fmt.Println(model.PushDataInfo.SourceData)
|
|
return pushData, 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) {
|
id = ""
|
payload, err := json.Marshal(pushDataInfo)
|
if err != nil {
|
return id, err
|
}
|
|
req, err := http.NewRequest("POST", url, bytes.NewBuffer(payload))
|
if err != nil {
|
return id, err
|
}
|
|
client := &http.Client{}
|
resp, err := client.Do(req)
|
if err != nil {
|
return id, err
|
}
|
defer resp.Body.Close()
|
|
var responseData map[string]interface{}
|
err = json.NewDecoder(resp.Body).Decode(&responseData)
|
if err != nil {
|
return id, err
|
}
|
id = responseData["data"].(string)
|
// 处理响应
|
// 这里可以根据实际需求进行处理，例如读取响应内容或检查状态码等
|
|
return id, nil
|
|
}
|
|
func DeleteData(id string) bool {
|
url := "http://" + config.Options.EsUrl + "/ai_face_ocean/_delete_by_query?refresh=true"
|
deleteDSL := `{
|
"query":{
|
"bool":{
|
"filter":[{
|
"term":{
|
"id":"` + id + `"
|
}
|
}]
|
}
|
}
|
}`
|
buf, err := esutil.EsReq("POST", url, []byte(deleteDSL))
|
if err != nil {
|
fmt.Println(err)
|
}
|
deleted, err := util.SourceDeleted(buf)
|
if err != nil {
|
fmt.Println(err)
|
}
|
if deleted >= 1 {
|
return true
|
}
|
return false
|
}
|
|
func MarkData(id string) bool {
|
url := "http://" + config.Options.EsUrl + "/ai_face_ocean/_update_by_query?refresh=true"
|
markDSL := `{
|
"script": {
|
"source": "ctx._source.isDelete=true"
|
},
|
"query": {
|
"term": {
|
"id": "` + id + `"
|
}
|
}
|
}`
|
buf, err := esutil.EsReq("POST", url, []byte(markDSL))
|
if err != nil {
|
fmt.Println(err)
|
return false
|
}
|
total, err := util.SourceUpdated(buf)
|
if err != nil {
|
fmt.Println(err)
|
return false
|
}
|
fmt.Println(total)
|
return true
|
}
|
|
func GetImageData(url string) ([]byte, error) {
|
// 发起HTTP GET请求
|
response, err := http.Get(url)
|
if err != nil {
|
return nil, err
|
}
|
defer response.Body.Close()
|
|
// 读取图片数据
|
imageData, err := ioutil.ReadAll(response.Body)
|
if err != nil {
|
return nil, err
|
}
|
|
return imageData, nil
|
}
|
