golang 模拟客户端发送http请求上传文件demo
package main
import (
"bytes"
"encoding/json"
"fmt"
"github.com/google/uuid"
"io"
"log"
"mime/multipart"
"net/http"
"os"
"path/filepath"
"strings"
"time"
)
type Response struct {
Msg string `json:"msg"`
Code int `json:"code"`
}
func main() {
var total int
var response Response
fmt.Println("是否进行相似度对比,请输入0或者1: 0:不进行对比 | 1:进行对比")
var isdHash, isNeedChecker, root string
fmt.Scanln(&isdHash)
if isdHash == "0" || isdHash == "1" {
fmt.Println("参数正确:", isdHash)
} else {
fmt.Println("参数错误,请重新启动程序")
time.Sleep(time.Second)
return
}
fmt.Println("是否需要人工审核: 1:需要 | 2:不需要 ")
fmt.Scanln(&isNeedChecker)
if isNeedChecker == "1" || isNeedChecker == "2" {
fmt.Println("参数正确:", isNeedChecker)
} else {
fmt.Println("参数错误,请重新启动程序")
time.Sleep(time.Second)
return
}
fmt.Println("请输入文件目录:")
fmt.Scanln(&root)
fmt.Println(root)
imgFiles := make(map[string]string, 0)
jsonFiles := make(map[string]string, 0)
err := getAllFiles(root, imgFiles, jsonFiles)
if err != nil {
fmt.Println(err)
return
}
if len(imgFiles) == 0 {
fmt.Println("该目录下无图片文件！")
return
}
id := uuid.NewString()
imgLen := len(imgFiles)
fmt.Println("共有图片:", imgLen, " | json文件:", len(jsonFiles))
extraParams := map[string]string{
"isdHash": isdHash,
"isNeedChecker": isNeedChecker,
"id": id,
}
for k, v := range imgFiles {
request, err := newFileUploadRequest("http://192.168.1.11:8899/dataImport/v1/upload", extraParams, "imgFile", v, "jsonFile", jsonFiles[k])
if err != nil {
log.Fatal(err)
}
client := &http.Client{}
resp, err := client.Do(request)
if err != nil {
log.Fatal(err)
return
} else {
body := &bytes.Buffer{}
_, err := body.ReadFrom(resp.Body)
if err != nil {
log.Fatal(err)
}
json.Unmarshal(body.Bytes(), &response)
resp.Body.Close()
if response.Msg == "存在相同图片" {
total++
fmt.Println(response.Msg, total)
}
if response.Code != http.StatusOK {
fmt.Println(body)
}
imgLen--
fmt.Println("还剩:", imgLen, " 张图片")
}
}
fmt.Println()
fmt.Println("------------------")
fmt.Println("完成，5秒后退出程序")
time.Sleep(time.Second * 5)
}
func getAllFiles(root string, imgFiles, jsonFiles map[string]string) (err error) {
dirs, err := os.ReadDir(root)
if err != nil {
return err
}
for _, v := range dirs {
if v.IsDir() {
root2 := root + `\` + v.Name()
err = getAllFiles(root2, imgFiles, jsonFiles)
} else {
switch filepath.Ext(v.Name()) {
case ".jpg":
imgFiles[strings.TrimRight(v.Name(), filepath.Ext(v.Name()))] = root + `\` + v.Name()
case ".json":
jsonFiles[strings.TrimRight(v.Name(), filepath.Ext(v.Name()))] = root + `\` + v.Name()
default:
return
}
}
}
return
}
// Creates a new file upload http request with optional extra params
func newFileUploadRequest(uri string, params map[string]string, imgFiled, imgPath, jsonFiled, jsonPath string) (*http.Request, error) {
file, err := os.Open(imgPath)
if err != nil {
return nil, err
}
defer file.Close()
body := &bytes.Buffer{}
writer := multipart.NewWriter(body)
part, err := writer.CreateFormFile(imgFiled, filepath.Base(imgPath))
if err != nil {
return nil, err
}
_, err = io.Copy(part, file)
if err != nil {
return nil, err
}
if jsonPath != "" { //有json文件
file2, err := os.Open(jsonPath)
if err != nil {
return nil, err
}
defer file2.Close()
part2, err := writer.CreateFormFile(jsonFiled, filepath.Base(jsonPath))
if err != nil {
return nil, err
}
_, err = io.Copy(part2, file2)
if err != nil {
return nil, err
}
}
for key, val := range params {
_ = writer.WriteField(key, val)
}
err = writer.Close()
if err != nil {
return nil, err
}
req, err := http.NewRequest("POST", uri, body)
//origin := req.Header.Get("Origin")
//req.Header.Set("Access-Control-Allow-Origin", origin)
//req.Header.Set("Access-Control-Allow-Headers", "Content-Type,AccessToken,X-CSRF-Token, Authorization, Token,X-Token,X-User-Id")
//req.Header.Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS,DELETE,PUT")
//req.Header.Set("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type, New-Token, New-Expires-At")
//req.Header.Set("Access-Control-Allow-Credentials", "true")
req.Header.Add("Content-Type", writer.FormDataContentType())
return req, err
}