mirrors2/hk4e
1
0
Fork 0
mirror of https://github.com/FlourishingWorld/hk4e.git synced 2026-02-04 14:22:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

机器人http登录

Browse Source
This commit is contained in:
flswld
2023-02-14 12:28:31 +08:00
parent 65a3be68ca
commit c25473b070
5 changed files with 196 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
pkg/httpclient/http_client.go
View File

@@ -5,6 +5,7 @@ import (
"encoding/json"
"io"
"net/http"
"strings"
"time"
)
@@ -15,17 +16,17 @@ func init() {
Transport: &http.Transport{
DisableKeepAlives: true,
},
Timeout: time.Second * 60,
Timeout: time.Second * 10,
}
}
func Get[T any](url string, token string) (*T, error) {
func GetJson[T any](url string, authToken string) (*T, error) {
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return nil, err
}
if token != "" {
req.Header.Set("Authorization", "Bearer"+" "+token)
if authToken != "" {
req.Header.Set("Authorization", "Bearer"+" "+authToken)
}
rsp, err := httpClient.Do(req)
if err != nil {
@@ -44,7 +45,27 @@ func Get[T any](url string, token string) (*T, error) {
return responseData, nil
}
func Post[T any](url string, body any, token string) (*T, error) {
func GetRaw(url string, authToken string) (string, error) {
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return "", err
}
if authToken != "" {
req.Header.Set("Authorization", "Bearer"+" "+authToken)
}
rsp, err := httpClient.Do(req)
if err != nil {
return "", err
}
data, err := io.ReadAll(rsp.Body)
_ = rsp.Body.Close()
if err != nil {
return "", err
}
return string(data), nil
}
func PostJson[T any](url string, body any, authToken string) (*T, error) {
reqData, err := json.Marshal(body)
if err != nil {
return nil, err
@@ -54,8 +75,8 @@ func Post[T any](url string, body any, token string) (*T, error) {
return nil, err
}
req.Header.Set("Content-Type", "application/json")
if token != "" {
req.Header.Set("Authorization", "Bearer"+" "+token)
if authToken != "" {
req.Header.Set("Authorization", "Bearer"+" "+authToken)
}
rsp, err := httpClient.Do(req)
if err != nil {
@@ -73,3 +94,24 @@ func Post[T any](url string, body any, token string) (*T, error) {
}
return responseData, nil
}
func PostRaw(url string, body string, authToken string) (string, error) {
req, err := http.NewRequest("POST", url, strings.NewReader(body))
if err != nil {
return "", err
}
req.Header.Set("Content-Type", "application/json")
if authToken != "" {
req.Header.Set("Authorization", "Bearer"+" "+authToken)
}
rsp, err := httpClient.Do(req)
if err != nil {
return "", err
}
rspData, err := io.ReadAll(rsp.Body)
_ = rsp.Body.Close()
if err != nil {
return "", err
}
return string(rspData), nil
}