소스 검색

增加判断如果没有文件则不进行文件传输

Cheng Jian 2 년 전
부모
커밋
7eb7e78601
2개의 변경된 파일9개의 추가작업 그리고 4개의 파일을 삭제
  1. 2 0
      rpcx-gateway/converter.go
  2. 7 4
      rpcx-gateway/gateway.go

+ 2 - 0
rpcx-gateway/converter.go

@@ -148,6 +148,8 @@ func MultipartRequest2RpcxRequest(r *http.Request) (map[string]string, map[strin
 	}
 	formValues["authExclude"] = "false"
 
+	formValues["fileNum"] = string(len(form.File))
+
 	return formValues, form.File, nil
 }
 

+ 7 - 4
rpcx-gateway/gateway.go

@@ -110,10 +110,13 @@ func (g *Gateway) handler(r *http.Request, servicePath string) (meta map[string]
 			return nil, nil, err
 		}
 
-		//发送文件
-		err = sendFile(conn, formFile)
-		if err != nil {
-			return nil, nil, err
+		//判断是否需要传输文件
+		if len(formFile) > 0 {
+			//发送文件
+			err = sendFile(conn, formFile)
+			if err != nil {
+				return nil, nil, err
+			}
 		}
 
 		//获取反馈结果