465 lines
12 KiB
Go
465 lines
12 KiB
Go
|
package webodm
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"context"
|
|||
|
|
"encoding/base64"
|
|||
|
|
"encoding/json"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/go-resty/resty/v2"
|
|||
|
|
"github.com/gogf/gf/v2/frame/g"
|
|||
|
|
"github.com/gogf/gf/v2/os/gctx"
|
|||
|
|
"github.com/tidwall/gjson"
|
|||
|
|
"github.com/tiger1103/gfast/v3/api/v1/common/coryCommon"
|
|||
|
|
"io"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"mime/multipart"
|
|||
|
|
"net/http"
|
|||
|
|
"net/url"
|
|||
|
|
"os"
|
|||
|
|
fpath "path/filepath"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 获取项目列表及其对应的任务列表
|
|||
|
|
func (w WebOdmApi) ProjectList(ctx context.Context, req *ProjectListReq) (res *ProjectListRes, err error) {
|
|||
|
|
res, err = ProjectListFunc()
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
func ProjectListFunc() (res *ProjectListRes, err error) {
|
|||
|
|
res = new(ProjectListRes)
|
|||
|
|
url := fmt.Sprintf("http://%s:%s/api/projects/?ordering=-created_at&page=1", Host, Port)
|
|||
|
|
|
|||
|
|
// 创建基本身份验证信息
|
|||
|
|
auth := Username + ":" + Password
|
|||
|
|
authEncoded := base64.StdEncoding.EncodeToString([]byte(auth))
|
|||
|
|
|
|||
|
|
// 创建请求
|
|||
|
|
request, err := http.NewRequest("GET", url, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error creating request:", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加基本身份验证信息到请求头
|
|||
|
|
request.Header.Set("Authorization", "Basic "+authEncoded)
|
|||
|
|
|
|||
|
|
// 发送请求
|
|||
|
|
client := &http.Client{}
|
|||
|
|
resp, err := client.Do(request)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error sending request:", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
// 读取响应体
|
|||
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error reading response body:", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析 JSON 响应
|
|||
|
|
var projectsResponse ProjectsResponse
|
|||
|
|
err = json.Unmarshal(body, &projectsResponse)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error decoding JSON:", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置响应结果
|
|||
|
|
res.ProjectsResponse = projectsResponse
|
|||
|
|
return res, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建任务
|
|||
|
|
func (w WebOdmApi) TaskCreate(ctx context.Context, req *TaskCreateReq) (res *TaskCreateRes, err error) {
|
|||
|
|
res = new(TaskCreateRes)
|
|||
|
|
res, err = TaskCreateFunc(ctx, req)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
func TaskCreateFunc(ctx context.Context, req *TaskCreateReq) (res *TaskCreateRes, err error) {
|
|||
|
|
res = new(TaskCreateRes)
|
|||
|
|
// 准备图片 URL 列表
|
|||
|
|
imageURLs := req.UrlList
|
|||
|
|
|
|||
|
|
// 准备一个 buffer 来保存 multipart form 的内容
|
|||
|
|
body := &bytes.Buffer{}
|
|||
|
|
writer := multipart.NewWriter(body)
|
|||
|
|
|
|||
|
|
// 遍历图片 URL 列表并从 URL 获取图片并添加到 multipart form 中
|
|||
|
|
for i, imageURL := range imageURLs {
|
|||
|
|
// 从 URL 获取图片
|
|||
|
|
resp, err := http.Get(imageURL)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to fetch image from URL:", err)
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
// 准备一个 multipart form 的 part，将图片数据写入
|
|||
|
|
part, err := writer.CreateFormFile("file"+strconv.Itoa(i+1), fpath.Base(imageURL))
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to create form file:", err)
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
_, err = io.Copy(part, resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to write image data:", err)
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 写入项目ID到 multipart form 中
|
|||
|
|
writer.WriteField("project_id", req.ProjectId)
|
|||
|
|
|
|||
|
|
// 设置任务参数，以生成合并后的tif图片
|
|||
|
|
writer.WriteField("merge", "true")
|
|||
|
|
|
|||
|
|
// 关闭 multipart writer
|
|||
|
|
writer.Close()
|
|||
|
|
|
|||
|
|
// 创建包含基本身份验证的请求
|
|||
|
|
url := fmt.Sprintf("http://%s:%s/api/projects/%s/tasks/", Host, Port, req.ProjectId)
|
|||
|
|
request, err := http.NewRequest("POST", url, body)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to create request:", err)
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加基本身份验证信息
|
|||
|
|
auth := Username + ":" + Password
|
|||
|
|
authEncoded := base64.StdEncoding.EncodeToString([]byte(auth))
|
|||
|
|
request.Header.Set("Authorization", "Basic "+authEncoded)
|
|||
|
|
|
|||
|
|
// 设置请求头中的 Content-Type 为 multipart/form-data
|
|||
|
|
request.Header.Set("Content-Type", writer.FormDataContentType())
|
|||
|
|
|
|||
|
|
// 发送请求
|
|||
|
|
client := &http.Client{}
|
|||
|
|
resp, err := client.Do(request)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to send request:", err)
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
// 读取响应体
|
|||
|
|
respBody, err := io.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to read response body:", err)
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析 JSON 响应
|
|||
|
|
var jsonResponse map[string]interface{}
|
|||
|
|
err = json.Unmarshal(respBody, &jsonResponse)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Failed to parse JSON response:", err)
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 提取任务ID
|
|||
|
|
taskID, ok := jsonResponse["id"].(string)
|
|||
|
|
if !ok {
|
|||
|
|
fmt.Println("Failed to extract task ID from response")
|
|||
|
|
return res, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置任务ID到响应结构体中
|
|||
|
|
res.TaskId = taskID
|
|||
|
|
return res, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取任务进度
|
|||
|
|
func (w WebOdmApi) TaskProcess(ctx context.Context, req *TaskProcessReq) (*TaskProcessRes, error) {
|
|||
|
|
ressss, err := TaskProcessFunc(ctx, req)
|
|||
|
|
return ressss, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TaskProcessFunc(ctx context.Context, req *TaskProcessReq) (res *TaskProcessRes, err error) {
|
|||
|
|
res = new(TaskProcessRes)
|
|||
|
|
|
|||
|
|
// 构建请求 URL
|
|||
|
|
url := fmt.Sprintf("http://%s:%s/api/projects/"+req.ProjectId+"/tasks/%s/", Host, Port, req.TaskId)
|
|||
|
|
|
|||
|
|
// 创建基本身份验证信息
|
|||
|
|
auth := Username + ":" + Password
|
|||
|
|
authEncoded := base64.StdEncoding.EncodeToString([]byte(auth))
|
|||
|
|
|
|||
|
|
// 创建请求
|
|||
|
|
request, err := http.NewRequest("GET", url, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error creating request:", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加基本身份验证信息到请求头
|
|||
|
|
request.Header.Set("Authorization", "Basic "+authEncoded)
|
|||
|
|
|
|||
|
|
// 发送请求
|
|||
|
|
client := &http.Client{}
|
|||
|
|
resp, err := client.Do(request)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error sending request:", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
// 读取响应体
|
|||
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error reading response body:", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析 JSON 响应
|
|||
|
|
err = json.Unmarshal(body, res)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println("Error decoding JSON:", err)
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return res, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 下载任务得到的结果
|
|||
|
|
func (w WebOdmApi) DownloadTask(ctx context.Context, req *TaskDownloadReq) (res *TaskDownloadRes, err error) {
|
|||
|
|
res, err = DownloadTask(ctx, req)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func DownloadTask(ctx context.Context, req *TaskDownloadReq) (res *TaskDownloadRes, err error) {
|
|||
|
|
res = new(TaskDownloadRes)
|
|||
|
|
|
|||
|
|
// 构建请求 URL
|
|||
|
|
requestURL := fmt.Sprintf("http://%s:%s/api/projects/"+req.ProjectId+"/tasks/%s/orthophoto/export", Host, Port, req.TaskId)
|
|||
|
|
|
|||
|
|
// 创建基本身份验证信息
|
|||
|
|
auth := Username + ":" + Password
|
|||
|
|
authEncoded := base64.StdEncoding.EncodeToString([]byte(auth))
|
|||
|
|
|
|||
|
|
// 获取任务的 epsg
|
|||
|
|
epsg, err := GetEpsgByTaskID(req.ProjectId, req.TaskId)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建请求体
|
|||
|
|
form := url.Values{}
|
|||
|
|
form.Set("format", req.Format)
|
|||
|
|
form.Set("epsg", epsg)
|
|||
|
|
|
|||
|
|
// 创建请求
|
|||
|
|
httpRequest, err := http.NewRequest("POST", requestURL, strings.NewReader(form.Encode()))
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置基本身份验证头部
|
|||
|
|
httpRequest.Header.Set("Authorization", "Basic "+authEncoded)
|
|||
|
|
httpRequest.Header.Set("Content-Type", "application/x-www-form-urlencoded; charset=UTF-8")
|
|||
|
|
|
|||
|
|
// 执行请求
|
|||
|
|
client := &http.Client{}
|
|||
|
|
httpResponse, err := client.Do(httpRequest)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
defer httpResponse.Body.Close()
|
|||
|
|
|
|||
|
|
// 读取响应体
|
|||
|
|
responseBody, err := ioutil.ReadAll(httpResponse.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var TaskExportRes struct {
|
|||
|
|
CeleryTaskID string `json:"celery_task_id"`
|
|||
|
|
Filename string `json:"filename"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析 JSON 响应
|
|||
|
|
err = json.Unmarshal(responseBody, &TaskExportRes)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
celeryTaskID := TaskExportRes.CeleryTaskID
|
|||
|
|
filename := TaskExportRes.Filename
|
|||
|
|
|
|||
|
|
type WorkerCheckRes struct {
|
|||
|
|
Ready bool `json:"ready"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义 worker 检查结果的变量
|
|||
|
|
var workerCheckRes WorkerCheckRes
|
|||
|
|
|
|||
|
|
// 循环检查任务状态，直到 Ready 为 true
|
|||
|
|
for {
|
|||
|
|
// 发起请求到 http://%s:%s/api/workers/check/celeryTaskID
|
|||
|
|
workerCheckURL := fmt.Sprintf("http://%s:%s/api/workers/check/%s", Host, Port, celeryTaskID)
|
|||
|
|
|
|||
|
|
// 创建新的 GET 请求
|
|||
|
|
workerCheckRequest, err := http.NewRequest("GET", workerCheckURL, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置基本身份验证头部
|
|||
|
|
workerCheckRequest.Header.Set("Authorization", "Basic "+authEncoded)
|
|||
|
|
workerCheckRequest.Header.Set("Accept", "application/json")
|
|||
|
|
|
|||
|
|
// 执行 GET 请求
|
|||
|
|
workerCheckResponse, err := client.Do(workerCheckRequest)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
defer workerCheckResponse.Body.Close()
|
|||
|
|
|
|||
|
|
// 读取 GET 响应体
|
|||
|
|
workerCheckResponseBody, err := ioutil.ReadAll(workerCheckResponse.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析 GET 响应
|
|||
|
|
err = json.Unmarshal(workerCheckResponseBody, &workerCheckRes)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 输出 worker 检查结果
|
|||
|
|
fmt.Println("Worker Check Result:", workerCheckRes)
|
|||
|
|
|
|||
|
|
// 如果 Ready 为 true，则退出循环
|
|||
|
|
if workerCheckRes.Ready {
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果 Ready 为 false，则等待 1 秒后重新检查
|
|||
|
|
time.Sleep(1 * time.Second)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建下载 URL
|
|||
|
|
downloadURL := fmt.Sprintf("http://%s:%s/api/workers/get/%s?filename=%s", Host, Port, celeryTaskID, filename)
|
|||
|
|
|
|||
|
|
// 发送 GET 请求并获取响应
|
|||
|
|
resp, err := http.Get(downloadURL)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
// 检查响应状态码
|
|||
|
|
if resp.StatusCode != http.StatusOK {
|
|||
|
|
return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 读取图片数据
|
|||
|
|
imageData, err := ioutil.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建目标文件夹路径
|
|||
|
|
if err := os.MkdirAll(coryCommon.UavMerge, os.ModePerm); err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构建文件名，添加当前时间的年月日时分秒作为前缀
|
|||
|
|
currentTime := time.Now().Format("20060102150405")
|
|||
|
|
fileNameWithTime := fmt.Sprintf("%s%s", currentTime, filename)
|
|||
|
|
|
|||
|
|
// 构建文件路径
|
|||
|
|
filepath := fpath.Join(coryCommon.UavMerge, fileNameWithTime)
|
|||
|
|
|
|||
|
|
// 写入图片数据到文件
|
|||
|
|
err = ioutil.WriteFile(filepath, imageData, 0644)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 打印下载文件的信息
|
|||
|
|
fmt.Printf("Downloaded image saved as: %s\n", filepath)
|
|||
|
|
|
|||
|
|
// 获取IP和端口
|
|||
|
|
//host := g.RequestFromCtx(ctx).Host
|
|||
|
|
res.FileUrl = fmt.Sprintf("/%s/%s", coryCommon.UavMerge, fileNameWithTime)
|
|||
|
|
return res, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetEpsgByTaskID 根据任务ID 获取 epsg
|
|||
|
|
func GetEpsgByTaskID(projectID, taskID string) (string, error) {
|
|||
|
|
client := &http.Client{}
|
|||
|
|
|
|||
|
|
auth := Username + ":" + Password
|
|||
|
|
authEncoded := base64.StdEncoding.EncodeToString([]byte(auth))
|
|||
|
|
|
|||
|
|
url := fmt.Sprintf("http://%s:%s/api/projects/%s/tasks/?ordering=-created_at", Host, Port, projectID)
|
|||
|
|
|
|||
|
|
request, err := http.NewRequest("GET", url, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
request.Header.Set("Authorization", "Basic "+authEncoded)
|
|||
|
|
|
|||
|
|
resp, err := client.Do(request)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
defer resp.Body.Close()
|
|||
|
|
|
|||
|
|
body, err := io.ReadAll(resp.Body)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
result := gjson.Parse(string(body))
|
|||
|
|
for _, value := range result.Array() {
|
|||
|
|
if value.Get("id").String() == taskID {
|
|||
|
|
return value.Get("epsg").String(), nil
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return "", nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// @Title DelUuidFunc 2024/8/13 11:40:00
|
|||
|
|
// @Description 根据项目ID及任务ID 找到资源并删除
|
|||
|
|
// @Auth Cory
|
|||
|
|
// @param projectID ---> "webdom项目ID"
|
|||
|
|
// @Return taskID ---> "webdom任务ID"
|
|||
|
|
func DelUuidFunc(projectID, taskID string) (flag bool, err error) {
|
|||
|
|
client := resty.New()
|
|||
|
|
header := client.R().
|
|||
|
|
SetHeader("Content-Type", "application/json").
|
|||
|
|
SetHeader("Authorization", "Basic "+base64.StdEncoding.EncodeToString([]byte(Username+":"+Password)))
|
|||
|
|
//1、获取uuid
|
|||
|
|
url := fmt.Sprintf("http://%s:%s/api/projects/%s/tasks/%s", Host, Port, projectID, taskID)
|
|||
|
|
resp, err := header.Get(url)
|
|||
|
|
if err != nil {
|
|||
|
|
g.Log().Error(gctx.New(), err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
uuid := gjson.Get(string(resp.Body()), "uuid").String()
|
|||
|
|
if uuid == "" {
|
|||
|
|
err = errors.New("删除失败，无法获取uuid！")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
//2、根据uuid去删除对应的资源
|
|||
|
|
m := make(map[string]string)
|
|||
|
|
m["uuid"] = uuid
|
|||
|
|
url = url + "/remove/"
|
|||
|
|
post, err := header.SetBody(m).Post(url)
|
|||
|
|
if err != nil {
|
|||
|
|
g.Log().Error(gctx.New(), err)
|
|||
|
|
return false, err
|
|||
|
|
}
|
|||
|
|
return gjson.Get(string(post.Body()), "succeed").Bool(), nil
|
|||
|
|
}
|