OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

feat: 直连网元请求带UA标识OMC/version

Browse Source
This commit is contained in:
TsMask
2024-12-18 09:57:31 +08:00
parent edb77d96fb
commit fdf5a517ed
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/framework/utils/fetch/fetch.go
View File

@@ -12,8 +12,13 @@ import (
"os" "os"
"strings" "strings"
"time" "time"
libGlobal "be.ems/lib/global"
) )
// userAgent 自定义 User-Agent
var userAgent = fmt.Sprintf("OMC/%s", libGlobal.Version)
// Get 发送 GET 请求 // Get 发送 GET 请求
// timeout 超时时间（毫秒） // timeout 超时时间（毫秒）
func Get(url string, headers map[string]string, timeout int) ([]byte, error) { 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 return nil, err
} }
req.Header.Set("User-Agent", userAgent)
req.Header.Set("Content-Type", "application/json;charset=UTF-8")
for key, value := range headers { for key, value := range headers {
req.Header.Set(key, value) 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 return nil, err
} }
req.Header.Set("User-Agent", userAgent)
req.Header.Set("Content-Type", "application/x-www-form-urlencoded") req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
for key, value := range headers { for key, value := range headers {
req.Header.Set(key, value) req.Header.Set(key, value)
} }
@@ -100,8 +107,8 @@ func PostJSON(url string, data any, headers map[string]string) ([]byte, error) {
return nil, err 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 { for key, value := range headers {
req.Header.Set(key, value) 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) return nil, fmt.Errorf("failed to create HTTP request: %v", err)
} }
req.Header.Set("User-Agent", userAgent)
req.Header.Set("Content-Type", writer.FormDataContentType()) req.Header.Set("Content-Type", writer.FormDataContentType())
client := &http.Client{} client := &http.Client{}
@@ -193,6 +201,8 @@ func PutJSON(url string, data any, headers map[string]string) ([]byte, error) {
return nil, err return nil, err
} }
req.Header.Set("User-Agent", userAgent)
req.Header.Set("Content-Type", "application/json;charset=UTF-8")
for key, value := range headers { for key, value := range headers {
req.Header.Set(key, value) req.Header.Set(key, value)
} }
@@ -224,6 +234,8 @@ func Delete(url string, headers map[string]string) ([]byte, error) {
return nil, err return nil, err
} }
req.Header.Set("User-Agent", userAgent)
req.Header.Set("Content-Type", "application/json;charset=UTF-8")
for key, value := range headers { for key, value := range headers {
req.Header.Set(key, value) req.Header.Set(key, value)
} }