Initial commit
This commit is contained in:
yan.y
@@ -0,0 +1,296 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"mime/multipart"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"pet-house.com/core/helper/str"
|
||||
)
|
||||
|
||||
var ErrInvalidDataStruct = errors.New("invalid response data struct")
|
||||
var ErrBaseAuthConfig = errors.New("base auth config is error")
|
||||
var ErrEmptyFileNameField = errors.New("must set field filename")
|
||||
|
||||
type client struct {
|
||||
config *Config
|
||||
cookie *http.Cookie
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
TimeOver int64
|
||||
TimeOut int64
|
||||
Headers map[string]string // request headers
|
||||
CookieName string
|
||||
Host string
|
||||
}
|
||||
|
||||
// BaseAuth
|
||||
type BaseAuth struct {
|
||||
Enable bool
|
||||
Account string
|
||||
Pwd string
|
||||
}
|
||||
|
||||
// NewClient
|
||||
func NewClient(configs ...*Config) *client {
|
||||
var config *Config
|
||||
if len(configs) == 0 || configs[0] == nil {
|
||||
config = &Config{TimeOut: 30, TimeOver: 5, Headers: map[string]string{}}
|
||||
} else {
|
||||
config = configs[0]
|
||||
}
|
||||
if config.TimeOut == 0 {
|
||||
config.TimeOut = 30
|
||||
}
|
||||
if config.TimeOver == 0 {
|
||||
config.TimeOver = 5
|
||||
}
|
||||
if config.Host == "" {
|
||||
config.Host = "http://127.0.0.1:7777"
|
||||
}
|
||||
return &client{config: config}
|
||||
}
|
||||
|
||||
// NewResponse
|
||||
func NewResponse(path string) *ServerResponse {
|
||||
return &ServerResponse{path: path}
|
||||
}
|
||||
|
||||
type ServerResponse struct {
|
||||
path string
|
||||
baseAuth *BaseAuth
|
||||
Data interface{} `json:"data"`
|
||||
body io.Reader
|
||||
fields map[string]string
|
||||
}
|
||||
|
||||
// SetBaseAuth
|
||||
func (sr *ServerResponse) SetBaseAuth(account, pwd string) {
|
||||
sr.baseAuth = &BaseAuth{
|
||||
Enable: true,
|
||||
Account: account,
|
||||
Pwd: pwd,
|
||||
}
|
||||
}
|
||||
|
||||
// BaseAuth
|
||||
func (sr *ServerResponse) BaseAuth() *BaseAuth {
|
||||
return sr.baseAuth
|
||||
}
|
||||
|
||||
// SetFields
|
||||
func (sr *ServerResponse) SetFields(fields map[string]string) {
|
||||
sr.fields = fields
|
||||
}
|
||||
|
||||
// GetFields
|
||||
func (sr *ServerResponse) GetFields() map[string]string {
|
||||
return sr.fields
|
||||
}
|
||||
|
||||
// SetUploadFile
|
||||
func (sr *ServerResponse) SetUploadFile(name string) {
|
||||
f, err := os.Open(name)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
sr.body = f
|
||||
}
|
||||
|
||||
// Close
|
||||
func (sr *ServerResponse) Close() {
|
||||
if f, ok := sr.body.(*os.File); ok && f != nil {
|
||||
f.Close()
|
||||
}
|
||||
}
|
||||
|
||||
func (n *client) GetCookie() *http.Cookie {
|
||||
return n.cookie
|
||||
}
|
||||
|
||||
// Check
|
||||
func (n *client) Check(sr *ServerResponse) error {
|
||||
ba := sr.BaseAuth()
|
||||
if ba == nil {
|
||||
return nil
|
||||
}
|
||||
if ba.Enable && (ba.Account == "" || ba.Pwd == "") {
|
||||
return ErrBaseAuthConfig
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// getFullPath
|
||||
func (n *client) getFullPath(path string) string {
|
||||
log.Println("fullpath:", path)
|
||||
return str.Join(n.config.Host, path)
|
||||
}
|
||||
|
||||
// Post
|
||||
func (n *client) Post(sr *ServerResponse, data string) ([]byte, error) {
|
||||
err := n.Check(sr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
result := n.request("POST", n.getFullPath(sr.path), sr.BaseAuth(), strings.NewReader(data))
|
||||
if len(result) == 0 {
|
||||
return result, fmt.Errorf("Post %s 没有返回数据", n.getFullPath(sr.path))
|
||||
}
|
||||
if !json.Valid(result) || sr.Data == nil {
|
||||
sr.Data = string(result)
|
||||
return result, nil
|
||||
}
|
||||
err = json.Unmarshal(result, sr.Data)
|
||||
if err != nil {
|
||||
return result, fmt.Errorf("执行解码失败: %s 错误：%w ,结果: %v", n.getFullPath(sr.path), err, string(result))
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// GetFile
|
||||
func (n *client) GetFile(sr *ServerResponse) error {
|
||||
err := n.Check(sr)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_ = n.request("GET", n.getFullPath(sr.path), sr.BaseAuth(), nil)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Upload 上传文件
|
||||
func (n *client) Upload(sr *ServerResponse) ([]byte, error) {
|
||||
err := n.Check(sr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if sr.fields == nil {
|
||||
return nil, ErrEmptyFileNameField
|
||||
}
|
||||
if filename, ok := sr.fields["filename"]; !ok || filename == "" {
|
||||
return nil, ErrEmptyFileNameField
|
||||
}
|
||||
|
||||
body := &bytes.Buffer{}
|
||||
writer := multipart.NewWriter(body)
|
||||
fw, err := writer.CreateFormFile("file", sr.fields["filename"])
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("create form file %v", err)
|
||||
}
|
||||
|
||||
_, err = io.Copy(fw, sr.body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("copying fileWriter %v", err)
|
||||
}
|
||||
|
||||
for k, v := range sr.fields {
|
||||
_ = writer.WriteField(k, v)
|
||||
}
|
||||
|
||||
err = writer.Close() // close writer before POST request
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("writerClose: %v", err)
|
||||
}
|
||||
|
||||
n.config.Headers["Content-Type"] = writer.FormDataContentType()
|
||||
|
||||
result := n.request("POST", n.getFullPath(sr.path), sr.BaseAuth(), body)
|
||||
if len(result) == 0 {
|
||||
return result, fmt.Errorf("Upload %s 没有返回数据", n.getFullPath(sr.path))
|
||||
}
|
||||
|
||||
if !json.Valid(result) || sr.Data == nil {
|
||||
sr.Data = string(result)
|
||||
return result, nil
|
||||
}
|
||||
|
||||
err = json.Unmarshal(result, sr.Data)
|
||||
if err != nil {
|
||||
return result, fmt.Errorf("执行解码失败: %s 错误：%w ,结果: %v", n.getFullPath(sr.path), err, string(result))
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// Get 获取数据
|
||||
func (n *client) Get(sr *ServerResponse) ([]byte, error) {
|
||||
err := n.Check(sr)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
result := n.request("GET", n.getFullPath(sr.path), sr.BaseAuth(), nil)
|
||||
if len(result) == 0 {
|
||||
return result, fmt.Errorf("Get %s 没有返回数据", n.getFullPath(sr.path))
|
||||
}
|
||||
if !json.Valid(result) || sr.Data == nil {
|
||||
sr.Data = string(result)
|
||||
return result, nil
|
||||
}
|
||||
err = json.Unmarshal(result, sr.Data)
|
||||
if err != nil {
|
||||
return result, fmt.Errorf("执行解码失败: %s 获取服务解析返回内容报错 %w : ,结果:[%s]", n.getFullPath(sr.path), err, string(result))
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (n *client) request(method, fullpath string, ba *BaseAuth, body io.Reader) []byte {
|
||||
result := make(chan []byte, 30)
|
||||
T := time.NewTicker(time.Duration(n.config.TimeOver) * time.Second)
|
||||
go func() {
|
||||
t := time.Duration(n.config.TimeOut) * time.Second
|
||||
Client := http.Client{Timeout: t}
|
||||
req, err := http.NewRequest(method, fullpath, body)
|
||||
if err != nil {
|
||||
result <- nil
|
||||
return
|
||||
}
|
||||
|
||||
if len(n.config.Headers) > 0 {
|
||||
for key, value := range n.config.Headers {
|
||||
req.Header.Set(key, value)
|
||||
}
|
||||
} else {
|
||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded; param=value")
|
||||
}
|
||||
if ba != nil && ba.Enable {
|
||||
req.SetBasicAuth(ba.Account, ba.Pwd)
|
||||
}
|
||||
var resp *http.Response
|
||||
resp, err = Client.Do(req)
|
||||
if err != nil {
|
||||
result <- nil
|
||||
return
|
||||
}
|
||||
if n.config.CookieName != "" && resp.Cookies() != nil {
|
||||
for _, cookie := range resp.Cookies() {
|
||||
if cookie.Name == n.config.CookieName {
|
||||
n.cookie = cookie
|
||||
}
|
||||
}
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
buf := bytes.NewBuffer(nil)
|
||||
io.Copy(buf, resp.Body)
|
||||
result <- buf.Bytes()
|
||||
|
||||
}()
|
||||
|
||||
for {
|
||||
select {
|
||||
case x := <-result:
|
||||
return x
|
||||
case <-T.C:
|
||||
return []byte("请求超时")
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,167 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func startGin() {
|
||||
r := gin.Default()
|
||||
gin.DebugPrintRouteFunc = func(httpMethod, absolutePath, handlerName string, nuHandlers int) {
|
||||
log.Printf("endpoint %v %v %v %v\n", httpMethod, absolutePath, handlerName, nuHandlers)
|
||||
}
|
||||
|
||||
r.POST("/foo", func(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, "foo")
|
||||
})
|
||||
|
||||
r.GET("/bar", func(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, "bar")
|
||||
})
|
||||
|
||||
r.GET("/status", func(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, "ok")
|
||||
})
|
||||
r.StaticFS("/txt", http.Dir("./txt"))
|
||||
|
||||
r.MaxMultipartMemory = 8 << 20 // 8 MiB
|
||||
r.POST("/upload", func(c *gin.Context) {
|
||||
// single file
|
||||
file, err := c.FormFile("file")
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, "upload failed!")
|
||||
return
|
||||
}
|
||||
|
||||
// Upload the file to specific dst.
|
||||
c.SaveUploadedFile(file, "./txt")
|
||||
|
||||
c.JSON(http.StatusOK, fmt.Sprintf("'%s' uploaded!", file.Filename))
|
||||
})
|
||||
|
||||
// Listen and Server in http://0.0.0.0:7777
|
||||
r.Run(":7777")
|
||||
}
|
||||
|
||||
func TestNewClient(t *testing.T) {
|
||||
|
||||
client := NewClient()
|
||||
t.Run("test new client", func(t *testing.T) {
|
||||
if client == nil {
|
||||
t.Error("client is nil")
|
||||
return
|
||||
}
|
||||
if client.config.TimeOut != 30 {
|
||||
t.Errorf("client default timeout want %d but get %d", 30, client.config.TimeOut)
|
||||
return
|
||||
}
|
||||
if client.config.TimeOver != 5 {
|
||||
t.Errorf("client default timeover want %d but get %d", 5, client.config.TimeOver)
|
||||
return
|
||||
}
|
||||
if client.config.Host != "http://127.0.0.1:7777" {
|
||||
t.Errorf("client default timeover want %s but get %s", "http://127.0.0.1:7777", client.config.Host)
|
||||
return
|
||||
}
|
||||
})
|
||||
|
||||
response := NewResponse("/foo")
|
||||
fullpath := client.getFullPath("/fullpath")
|
||||
if client.getFullPath("/fullpath") != "http://127.0.0.1:7777/fullpath" {
|
||||
t.Errorf("client default timeover want %s but get %s", "http://127.0.0.1:7777", fullpath)
|
||||
return
|
||||
}
|
||||
t.Run("test new response", func(t *testing.T) {
|
||||
if response == nil {
|
||||
t.Error("response is nil")
|
||||
return
|
||||
}
|
||||
if response.path != "/foo" {
|
||||
t.Errorf("response default path want %s but get %s", "/foo", response.path)
|
||||
}
|
||||
if response.Data != nil {
|
||||
t.Errorf("response default data is not nil")
|
||||
}
|
||||
ba := response.BaseAuth()
|
||||
if ba != nil {
|
||||
t.Errorf("response default baseauth is not nil")
|
||||
}
|
||||
response.SetBaseAuth("account", "pwd")
|
||||
ba = response.BaseAuth()
|
||||
if ba.Account != "account" {
|
||||
t.Errorf("response baseauth default accout is not account")
|
||||
}
|
||||
if ba.Pwd != "pwd" {
|
||||
t.Errorf("response baseauth default password is not pwd")
|
||||
}
|
||||
if !ba.Enable {
|
||||
t.Errorf("response baseauth default enable is not true")
|
||||
}
|
||||
fields := response.GetFields()
|
||||
if fields != nil {
|
||||
t.Errorf("response default fields is not nil")
|
||||
}
|
||||
f := map[string]string{
|
||||
"a": "a",
|
||||
"b": "b",
|
||||
}
|
||||
response.SetFields(f)
|
||||
fields = response.GetFields()
|
||||
if !reflect.DeepEqual(f, fields) {
|
||||
t.Error("fields is set failed")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("test get file", func(t *testing.T) {
|
||||
response := NewResponse("/txt/file.txt")
|
||||
err := client.GetFile(response)
|
||||
if err != nil {
|
||||
t.Error(err.Error())
|
||||
return
|
||||
}
|
||||
})
|
||||
t.Run("test upload file", func(t *testing.T) {
|
||||
response := NewResponse("/upload")
|
||||
response.SetUploadFile("./upload.txt")
|
||||
defer response.Close()
|
||||
response.SetFields(map[string]string{"filename": "upload.txt"})
|
||||
_, err := client.Upload(response)
|
||||
if err != nil {
|
||||
t.Error(err.Error())
|
||||
return
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("test get", func(t *testing.T) {
|
||||
response := NewResponse("/bar")
|
||||
_, err := client.Get(response)
|
||||
if err != nil {
|
||||
t.Error(err.Error())
|
||||
return
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("test post", func(t *testing.T) {
|
||||
response := NewResponse("/foo")
|
||||
b, err := json.Marshal(map[string]interface{}{
|
||||
"a": "a",
|
||||
"b": "b",
|
||||
})
|
||||
if err != nil {
|
||||
t.Error(err.Error())
|
||||
return
|
||||
}
|
||||
_, err = client.Post(response, string(b))
|
||||
if err != nil {
|
||||
t.Error(err.Error())
|
||||
return
|
||||
}
|
||||
})
|
||||
|
||||
}
|
||||
@@ -0,0 +1,12 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
go startGin()
|
||||
code := m.Run()
|
||||
os.Exit(code)
|
||||
}
|
||||
Reference in New Issue
Block a user