diff --git a/src/framework/utils/fetch/fetch.go b/src/framework/utils/fetch/fetch.go index be7cc07c..d0075b0f 100644 --- a/src/framework/utils/fetch/fetch.go +++ b/src/framework/utils/fetch/fetch.go @@ -12,8 +12,13 @@ import ( "os" "strings" "time" + + libGlobal "be.ems/lib/global" ) +// userAgent 自定义 User-Agent +var userAgent = fmt.Sprintf("OMC/%s", libGlobal.Version) + // Get 发送 GET 请求 // timeout 超时时间(毫秒) func Get(url string, headers map[string]string, timeout int) ([]byte, error) { @@ -29,6 +34,8 @@ func Get(url string, headers map[string]string, timeout int) ([]byte, error) { return nil, err } + req.Header.Set("User-Agent", userAgent) + req.Header.Set("Content-Type", "application/json;charset=UTF-8") for key, value := range headers { req.Header.Set(key, value) } @@ -60,8 +67,8 @@ func Post(url string, data url.Values, headers map[string]string) ([]byte, error return nil, err } + req.Header.Set("User-Agent", userAgent) req.Header.Set("Content-Type", "application/x-www-form-urlencoded") - for key, value := range headers { req.Header.Set(key, value) } @@ -100,8 +107,8 @@ func PostJSON(url string, data any, headers map[string]string) ([]byte, error) { return nil, err } - req.Header.Set("Content-Type", "application/json") - + req.Header.Set("User-Agent", userAgent) + req.Header.Set("Content-Type", "application/json;charset=UTF-8") for key, value := range headers { req.Header.Set(key, value) } @@ -156,6 +163,7 @@ func PostUploadFile(url string, params map[string]string, file *os.File) ([]byte return nil, fmt.Errorf("failed to create HTTP request: %v", err) } + req.Header.Set("User-Agent", userAgent) req.Header.Set("Content-Type", writer.FormDataContentType()) client := &http.Client{} @@ -193,6 +201,8 @@ func PutJSON(url string, data any, headers map[string]string) ([]byte, error) { return nil, err } + req.Header.Set("User-Agent", userAgent) + req.Header.Set("Content-Type", "application/json;charset=UTF-8") for key, value := range headers { req.Header.Set(key, value) } @@ -224,6 +234,8 @@ func Delete(url string, headers map[string]string) ([]byte, error) { return nil, err } + req.Header.Set("User-Agent", userAgent) + req.Header.Set("Content-Type", "application/json;charset=UTF-8") for key, value := range headers { req.Header.Set(key, value) }