增加下载上传功能

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"fmt"
+	"io"
 	"math"
 	"net/http"
 	"sort"
@@ -1199,11 +1200,18 @@ func initwellParaUnitMap() {
 }
 
 func fileServerG() {
-	// http.Handle("/file", http.FileServer(http.Dir("E:/Code/old/deesCloud/deescloud/authfile")))
 	http.Handle("/file/", http.StripPrefix("/file/", http.FileServer(http.Dir(modelePath+"/authfile"))))
 	err := http.ListenAndServe("0.0.0.0:8881", nil)
 	if err != nil {
-		fmt.Println("服务器错误8884")
+		fmt.Println("服务器错误8881")
+	}
+}
+
+func uploadFile() {
+	http.Handle("/upload/", http.StripPrefix("/upload/", http.FileServer(http.Dir(modelePath+"/upload"))))
+	err := http.ListenAndServe("0.0.0.0:8882", nil)
+	if err != nil {
+		fmt.Print(err)
 	}
 }
 
@@ -1316,6 +1324,7 @@ func main() {
 
 	//fileserver
 	go fileServerG()
+	go uploadFile()
 
 	http.HandleFunc("/test", test)
 
@@ -1535,6 +1544,10 @@ func main() {
 
 	http.HandleFunc("/deescloud/get_tool_WarningCsv", get_tool_WarningCsv)
 	http.HandleFunc("/deescloud/getInstrumentMess", getInstrumentMess)
+	http.HandleFunc("/deescloud/addQualityData", addQualityData)
+	http.HandleFunc("/deescloud/getQualityData", getQualityData)
+	http.HandleFunc("/deescloud/upload", upload)
+	http.HandleFunc("/deescloud/download_ds_file", download_ds_file)
 
 	//监听服务
 	fmt.Println("listen server ...")
@@ -3599,7 +3612,7 @@ func GenerateInstrumentPDF(fullData []byte, resp []Tool_warning) (filename strin
 		pdf.SetX(margin)
 		pdf.SetY(y)
 		reportTime := time.Now().Format("2006-01-02 15:04:05")
-		pdf.Cell(nil, fmt.Sprintf("井号：%s\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t报告时间：%s", wellName, reportTime))
+		pdf.Cell(nil, fmt.Sprintf("井号：%s\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t报告时间：%s", wellName, reportTime))
 		y += lineHeight * 1.2
 		pdf.Line(margin, y, pageWidth-margin, y)
 		y += 10
@@ -30072,7 +30085,7 @@ func getInstrumentMessUtil(req InstrumentMess) (*InstrumentMessRespdata, error)
 	}
 	fmt.Println("")
 	//上井情况汇总
-	resp.WorkMessList = workMess(wellInfos, req.Flag, req.Content, req.WellName)
+	resp.WorkMessList = workMess(wellInfos, req.Flag, req.Content)
 	//维保情况
 	resp.GetLcmRespData = getMaintenanceMess(req.Series, req.Instrument, req.Opuser)
 	//仪器所在部门
@@ -30082,7 +30095,7 @@ func getInstrumentMessUtil(req InstrumentMess) (*InstrumentMessRespdata, error)
 	return resp, nil
 }
 
-func workMess(wellInfos []get_vibration_resp, flag string, content string, WellName string) []WellMessRespdata {
+func workMess(wellInfos []get_vibration_resp, flag string, content string) []WellMessRespdata {
 
 	var results []WellMessRespdata
 	if len(wellInfos) == 0 {
@@ -30095,6 +30108,7 @@ func workMess(wellInfos []get_vibration_resp, flag string, content string, WellN
 	// 循环打印每个 WellNameSource
 	fmt.Println("开始打印 WellNameSource 列表:")
 	workMessSql := `SELECT 
+				MAX(id) AS id,
 				[5200] AS WellName,
 				(SELECT TOP 1 [5203] 
 				FROM [dbo].[EQUIPMENTRECORDS] t2 
@@ -30135,6 +30149,7 @@ func workMess(wellInfos []get_vibration_resp, flag string, content string, WellN
 			continue
 		}
 		defer conn.Close()
+
 		rows, err := conn.Query(qsql)
 		if err != nil {
 			fmt.Println("查询工作状态出错", err.Error())
@@ -30142,28 +30157,66 @@ func workMess(wellInfos []get_vibration_resp, flag string, content string, WellN
 		}
 
 		defer rows.Close()
-
+		fmt.Println("查询数据库------------------", qsql)
 		for rows.Next() {
 			var item WellMessRespdata
-			var wellName, sampleField, valueInterval, personnelList, globalMax1, globalMax2 string
+			var id, wellName, sampleField, valueInterval, personnelList, globalMax1, globalMax2 string
-			err := rows.Scan(&wellName, &sampleField, &valueInterval, &globalMax1, &globalMax2, &personnelList)
+
-			if err != nil {
+			//先扫描工作数据
+			if err := rows.Scan(&id, &wellName, &sampleField, &valueInterval, &globalMax1, &globalMax2, &personnelList); err != nil {
 				fmt.Println("扫描数据失败", db, err.Error())
 				continue
 			}
-			// 映射到结构体
+
+			//查询上传文件数据
+			queryUploadSql := `SELECT id, file_name, create_user, create_time, well_file_id, type 
+                      FROM t_file 
+                      WHERE type='tool_detail' AND well_file_id=@wellFileId`
+
+			uploadRows, err := sqlConn.Query(queryUploadSql, sql.Named("wellFileId", id))
+			if err != nil {
+				fmt.Printf("查询上传数据失败: %v\n", err)
+				continue
+			}
+
+			var uploadData []uploadRes
+			for uploadRows.Next() {
+				var res uploadRes
+				if err := uploadRows.Scan(
+					&res.ID,
+					&res.FileName,
+					&res.CreateUser,
+					&res.CreateTime,
+					&res.WellFileId,
+					&res.Types,
+				); err != nil {
+					fmt.Printf("扫描上传数据行失败: %v\n", err)
+					continue
+				}
+				uploadData = append(uploadData, res)
+			}
+			uploadRows.Close()
+
+			// 3. 填充数据
+			fmt.Printf("扫描到的值 - ID: %s, WellName: %s, SampleField: %s, ValueInterval: %s, GlobalMax1: %s, GlobalMax2: %s, PersonnelList: %s\n",
+				id, wellName, sampleField, valueInterval, globalMax1, globalMax2, personnelList)
+
+			item.ID = id
 			item.WellName = wellName
 			item.SampleField = sampleField
 			item.ValueInterval = valueInterval
 			item.GlobalMax1 = globalMax1
 			item.GlobalMax2 = globalMax2
 			item.PersonnelList = personnelList
+			item.Data = uploadData // 添加上传数据
+
 			if flag == "0" {
 				item.Content = "正常"
 			} else {
 				item.Content = content
 			}
 			item.Flag = flag
+
 			results = append(results, item)
 		}
 	}
@@ -30295,6 +30348,296 @@ func get_DeptMess(series string, instrument string) string {
 	}
 }
 
+func addQualityData(response http.ResponseWriter, request *http.Request) {
+	beginTime := time.Now().UnixNano()
+	fmt.Println("addQualityData recv req begin", time.Now().Format("2006-01-02 15:04:05"))
+
+	reqdata, _ := ioutil.ReadAll(request.Body)
+	var req QualityData
+	json.Unmarshal(reqdata, &req)
+
+	// 确定要更新的字段名
+	var fieldName string
+	switch req.Field {
+	case "process_record":
+		fieldName = "process_record"
+	case "process_inspection":
+		fieldName = "process_inspection"
+	case "final_inspection":
+		fieldName = "final_inspection"
+	default:
+		// 如果字段不匹配，返回错误
+		response.WriteHeader(http.StatusBadRequest)
+		response.Write([]byte(`{"code": 1, "msg": "无效的字段类型"}`))
+		return
+	}
+
+	// 构建查询语句，检查记录是否存在
+	// 安全警告：当前代码存在SQL注入风险，建议使用参数化查询
+	querySQL := fmt.Sprintf(`SELECT id 
+                         FROM t_quality_inspection 
+                         WHERE well_name = '%s' AND series = '%s' AND instrument = '%s'`,
+		req.WellName, req.Series, req.Instrument)
+
+	fmt.Println("querySQL----", querySQL)
+
+	row := sqlConn.QueryRow(querySQL) // 使用QueryRow获取单行结果
+
+	var fieldValue sql.NullString // 使用NullString处理可能为NULL的情况
+	err := row.Scan(&fieldValue)
+
+	var exists bool
+	if err != nil {
+		if err == sql.ErrNoRows {
+			exists = false // 没有找到记录
+		} else {
+			fmt.Println("查询质量检验数据失败:", err)
+			response.WriteHeader(http.StatusInternalServerError)
+			response.Write([]byte(`{"code": 1, "msg": "数据库查询失败"}`))
+			return
+		}
+	} else {
+		// 检查字段是否有值（非NULL且非空字符串）
+		exists = fieldValue.Valid && fieldValue.String != ""
+	}
+
+	// 现在可以使用exists变量
+	fmt.Printf("字段 '%s' 是否有值: %t\n", req.Field, exists)
+	// 获取当前时间
+	currentTime := time.Now().Format("2006-01-02 15:04:05")
+
+	if exists {
+		// 记录存在，执行更新操作
+		updateSQL := fmt.Sprintf(`UPDATE t_quality_inspection 
+		                          SET %s = '%s', update_time = '%s', update_user = '%s' 
+		                          WHERE well_name = '%s' AND series = '%s' AND instrument = '%s'`,
+			fieldName, req.Value, currentTime, req.OpUser, req.WellName, req.Series, req.Instrument)
+		fmt.Println("updateSQL----", updateSQL)
+		_, err := sqlConn.Exec(updateSQL)
+		if err != nil {
+			fmt.Println("更新质量检验数据失败:", err)
+			response.WriteHeader(http.StatusInternalServerError)
+			response.Write([]byte(`{"code": 1, "msg": "数据更新失败"}`))
+			return
+		}
+	} else {
+		// 记录不存在，执行插入操作
+		insertSQL := fmt.Sprintf(`INSERT INTO t_quality_inspection 
+		                          (create_time, update_time, create_user, update_user, %s, well_name, series, instrument) 
+		                          VALUES ('%s', '%s', '%s', '%s', '%s', '%s', '%s', '%s')`,
+			fieldName, currentTime, currentTime, req.OpUser, req.OpUser, req.Value, req.WellName, req.Series, req.Instrument)
+		fmt.Println("insertSQL----", insertSQL)
+		_, err := sqlConn.Exec(insertSQL)
+		if err != nil {
+			fmt.Println("插入质量检验数据失败:", err)
+			response.WriteHeader(http.StatusInternalServerError)
+			response.Write([]byte(`{"code": 1, "msg": "数据插入失败"}`))
+			return
+		}
+	}
+
+	// 返回成功响应
+	responseData := map[string]interface{}{
+		"code": 0,
+		"msg":  "操作成功",
+		"data": nil,
+	}
+
+	jsonResponse, _ := json.Marshal(responseData)
+	response.Write(jsonResponse)
+
+	fmt.Println("addQualityData recv req end", time.Now().Format("2006-01-02 15:04:05"), "cost:", (time.Now().UnixNano()-beginTime)/1e6, "ms")
+}
+
+func getQualityData(response http.ResponseWriter, request *http.Request) {
+	beginTime := time.Now().UnixNano()
+	fmt.Println("getQualityData recv req begin", time.Now().Format("2006-01-02 15:04:05"))
+
+	// 读取并解析请求体
+	reqdata, err := ioutil.ReadAll(request.Body)
+	if err != nil {
+		response.WriteHeader(http.StatusBadRequest)
+		response.Write([]byte(`{"code": 1, "msg": "读取请求体失败"}`))
+		return
+	}
+
+	var req QualityData
+	if err := json.Unmarshal(reqdata, &req); err != nil {
+		response.WriteHeader(http.StatusBadRequest)
+		response.Write([]byte(`{"code": 1, "msg": "解析请求体失败"}`))
+		return
+	}
+
+	// 验证必填字段
+	if req.Series == "" || req.Instrument == "" || req.Field == "" {
+		response.WriteHeader(http.StatusBadRequest)
+		response.Write([]byte(`{"code": 1, "msg": "缺少必要参数"}`))
+		return
+	}
+
+	// 查询数据库
+	querySQL := fmt.Sprintf(`SELECT COALESCE(process_record, '') as process_record ,COALESCE(process_inspection, '') as process_inspection ,COALESCE(final_inspection, '') as final_inspection
+                         FROM t_quality_inspection 
+                         WHERE well_name = '%s' AND series = '%s' AND instrument = '%s'`,
+		req.WellName, req.Series, req.Instrument)
+	fmt.Println("querySQL----", querySQL)
+	var resp QualityDataResp
+	err = sqlConn.QueryRow(querySQL).Scan(
+		&resp.ProcessRecord,
+		&resp.ProcessInspection,
+		&resp.FinalInspection,
+	)
+
+	if err != nil {
+		if err == sql.ErrNoRows {
+			fmt.Println("未查询到数据:", err)
+			// 没有找到记录，返回空值
+			resp = QualityDataResp{}
+		} else {
+			// 数据库查询出错
+			fmt.Println("查询质量检验数据失败:", err)
+			response.WriteHeader(http.StatusInternalServerError)
+			response.Write([]byte(`{"code": 1, "msg": "数据库查询失败"}`))
+			return
+		}
+	}
+
+	fmt.Println("查询质量检验数据成功:", resp)
+
+	// 构造响应
+	responseData := map[string]interface{}{
+		"code": 0,
+		"msg":  "操作成功",
+		"data": resp,
+	}
+
+	jsonResponse, err := json.Marshal(responseData)
+	if err != nil {
+		response.WriteHeader(http.StatusInternalServerError)
+		response.Write([]byte(`{"code": 1, "msg": "生成响应失败"}`))
+		return
+	}
+
+	response.Header().Set("Content-Type", "application/json")
+	response.Write(jsonResponse)
+
+	fmt.Println("getQualityData recv req end", time.Now().Format("2006-01-02 15:04:05"), "cost:", (time.Now().UnixNano()-beginTime)/1e6, "ms")
+}
+
+func upload(response http.ResponseWriter, request *http.Request) {
+	beginTime := time.Now().UnixNano()
+	fmt.Println("upload recv req begin", time.Now().Format("2006-01-02 15:04:05"))
+
+	// 解析 multipart 表单，最大内存为 32MB
+	err := request.ParseMultipartForm(32 << 20)
+	if err != nil {
+		logs.Error("解析表单失败:", err)
+		response.WriteHeader(http.StatusBadRequest)
+		response.Write([]byte(`{"code": 1, "msg": "解析表单失败"}`))
+		return
+	}
+
+	// 获取文件句柄和文件头
+	file, handler, err := request.FormFile("file")
+	if err != nil {
+		logs.Error("获取上传文件失败:", err)
+		response.WriteHeader(http.StatusBadRequest)
+		response.Write([]byte(`{"code": 1, "msg": "获取上传文件失败"}`))
+		return
+	}
+	defer file.Close()
+
+	// 获取其他表单字段
+	id := request.FormValue("id")
+	wellName := request.FormValue("wellName")
+	opuser := request.FormValue("opuser")
+	types := request.FormValue("type")
+
+	// 创建保存文件的目录（如果不存在）
+	saveDir := "./upload"
+	if _, err := os.Stat(saveDir); os.IsNotExist(err) {
+		os.MkdirAll(saveDir, 0755)
+	}
+
+	// 构造文件保存路径
+	filename := fmt.Sprintf("%s/%s", saveDir, handler.Filename)
+
+	// 创建文件
+	f, err := os.Create(filename)
+	if err != nil {
+		logs.Error("创建文件失败:", err)
+		response.WriteHeader(http.StatusInternalServerError)
+		response.Write([]byte(`{"code": 1, "msg": "创建文件失败"}`))
+		return
+	}
+	defer f.Close()
+
+	// 将上传的文件内容复制到新创建的文件中
+	_, err = io.Copy(f, file)
+	if err != nil {
+		logs.Error("保存文件失败:", err)
+		response.WriteHeader(http.StatusInternalServerError)
+		response.Write([]byte(`{"code": 1, "msg": "保存文件失败"}`))
+		return
+	}
+
+	// 返回成功响应
+	responseData := map[string]interface{}{
+		"code": 0,
+		"msg":  "文件上传成功",
+		"data": map[string]string{
+			"id":       id,
+			"wellName": wellName,
+			"opuser":   opuser,
+			"type":     types,
+			"filename": filename,
+			"fileSize": fmt.Sprintf("%d", handler.Size),
+		},
+	}
+
+	insert_sql := fmt.Sprintf(`INSERT INTO [t_file]
+           ([well_file_id]
+           ,[file_name]
+           ,[create_user]
+           ,[type]
+		   ,[create_time])
+     VALUES
+           ('%s'
+           ,'%s'
+           ,'%s','%s','%s')`, id, filename, opuser, types, time.Now().Format("2006-01-02 15:04:05"))
+	_, err = sqlConn.Exec(insert_sql)
+	if err != nil {
+		fmt.Println(fmt.Sprintf(`%s err:%s`, insert_sql, err.Error()))
+	}
+
+	jsonResponse, _ := json.Marshal(responseData)
+	response.Write(jsonResponse)
+
+	fmt.Println("upload recv req end", time.Now().Format("2006-01-02 15:04:05"), "cost:", (time.Now().UnixNano()-beginTime)/1e6, "ms")
+}
+
+func download_ds_file(rw http.ResponseWriter, r *http.Request) {
+	//获取请求参数
+
+	filename := r.FormValue("id")
+	fmt.Println("filename------", filename)
+	base_file_path := modelePath + "/upload/"
+
+	if filename != "" {
+		//设置响应头
+		header := rw.Header()
+		header.Add("Content-Type", "application/octet-stream")
+		//使用ioutil包读取文件
+		filepath := base_file_path + filename
+		fmt.Println(filepath)
+		b, _ := ioutil.ReadFile(filepath)
+
+		//写入到响应流中
+		rw.Write(b)
+	}
+
+}
+
 // 跨域测试
 func test(response http.ResponseWriter, request *http.Request) {
 	fmt.Println("recv test req")

req.go

@@ -666,3 +666,13 @@ type InstrumentMess struct {
 	Flag        string `json:"flag"`
 	Content     string `json:"content"`
 }
+
+type QualityData struct {
+	Field       string `json:"field"` //类型
+	Series      string `json:"series"`
+	Instrument  string `json:"instrument"`
+	OpUser      string `json:"opuser"`
+	Opuser_uuid string `json:"opuser_uuid"`
+	WellName    string `json:"wellName"`
+	Value       string `json:"value"`
+}

res.go

@@ -745,6 +745,7 @@ type InstrumentMessRespdata struct {
 	ID             string             `json:"id"`
 	WellName       string             `json:"wellName"`   //井名称
 	Time           string             `json:"time"`       //出厂日期
+	HouseDate      string             `json:"houseDate"`  //入库日期
 	Series         string             `json:"series"`     //系列号
 	Instrument     string             `json:"instrument"` //仪器号
 	Version        string             `json:"version"`    //版本号
@@ -759,6 +760,7 @@ type InstrumentMessRespdata struct {
 
 // 上井情况汇总
 type WellMessRespdata struct {
+	ID            string      `json:"id"`            //唯一标识
 	WellName      string      `json:"wellName"`      //井名称
 	SampleField   string      `json:"sampleField"`   //入井状态
 	ValueInterval string      `json:"valueInterval"` //入井井深
@@ -767,4 +769,21 @@ type WellMessRespdata struct {
 	PersonnelList string      `json:"personnelList"` //累计工作时间
 	Content       string      `json:"content"`       //仪器工作情况
 	Flag          string      `json:"flag"`          //仪器工作情况标识
+	Data          []uploadRes `json:"data"`
+}
+
+type QualityDataResp struct {
+	ProcessRecord     string `json:"process_record"`
+	ProcessInspection string `json:"process_inspection"`
+	FinalInspection   string `json:"final_inspection"`
+}
+
+type uploadRes struct {
+	ID           string `json:"id"`
+	FileName     string `json:"fileName"`
+	RealFileName string `json:"realFileName"`
+	CreateUser   string `json:"createUser"`
+	CreateTime   string `json:"createTime"`
+	WellFileId   string `json:"wellFileId"`
+	Types        string `json:"type"`
 }