console/plugin/api/gateway/helper.go

163 lines
3.8 KiB
Go

/* Copyright © INFINI Ltd. All rights reserved.
* web: https://infinilabs.com
* mail: hello#infini.ltd */
package gateway
import (
"bytes"
"crypto/tls"
"github.com/segmentio/encoding/json"
"infini.sh/console/model/gateway"
"infini.sh/framework/lib/fasthttp"
"io"
"time"
)
//import (
// "fmt"
// "infini.sh/console/model/gateway"
// "infini.sh/framework/core/orm"
// "infini.sh/framework/core/util"
//)
//
//func fetchInstanceGroup(instanceID string) (string, error){
// // fetch gateway instance group
// q := orm.Query{}
// q.RawQuery = []byte(fmt.Sprintf(`{"size": 1, "query":{"term":{"instance_id":{"value":"%s"}}}}`, instanceID))
// err, res := orm.Search(&gateway.InstanceGroup{}, &q)
// if err != nil {
// return "", err
// }
// if len(res.Result) > 0 {
// if rowMap, ok := res.Result[0].(map[string]interface{}); ok {
// return rowMap["group_id"].(string), nil
// }
// }
// return "", nil
//}
//
//func fetchInstanceGroupByID(instanceIDs []interface{})([]interface{}, error){
// if len(instanceIDs) == 0 {
// return nil, nil
// }
// // fetch gateway instance groups
// esQuery := util.MapStr{
// "query": util.MapStr{
// "terms": util.MapStr{
// "instance_id": instanceIDs,
// },
// },
// }
// q := orm.Query{}
// q.RawQuery = util.MustToJSONBytes(esQuery)
// err, res := orm.Search(&gateway.InstanceGroup{}, &q)
// return res.Result, err
//}
//func fetchGroupByID(groupIDs []interface{})([]interface{}, error){
// if len(groupIDs) == 0 {
// return nil, nil
// }
// // fetch gateway groups
// esQuery := util.MapStr{
// "query": util.MapStr{
// "terms": util.MapStr{
// "_id": groupIDs,
// },
// },
// }
// q := orm.Query{}
// q.RawQuery = util.MustToJSONBytes(esQuery)
// err, res := orm.Search(&gateway.Group{}, &q)
// return res.Result, err
//}
//
//func pickElasticsearchColumnValues(result []interface{}, columnName string) []interface{}{
// if len(result) == 0 {
// return nil
// }
// columnValues := make([]interface{}, 0, len(result))
// for _, row := range result {
// if rowMap, ok := row.(map[string]interface{}); ok {
// columnValues = append(columnValues, rowMap[columnName])
// }
// }
// return columnValues
//}
//
//func getRelationshipMap(result []interface{}, key string, value string) map[string]interface{}{
// if len(result) == 0 {
// return nil
// }
// resultMap := map[string]interface{}{}
// for _, row := range result {
// if rowMap, ok := row.(map[string]interface{}); ok {
// resultMap[rowMap[key].(string)] = rowMap[value]
// }
// }
// return resultMap
//}
type ProxyRequest struct {
Endpoint string
Path string
Method string
BasicAuth gateway.BasicAuth
Body interface{}
}
type ProxyResponse struct {
Body []byte
StatusCode int
}
func doGatewayRequest(req *ProxyRequest) (*ProxyResponse, error){
var (
freq = fasthttp.AcquireRequest()
fres = fasthttp.AcquireResponse()
)
defer func() {
fasthttp.ReleaseRequest(freq)
fasthttp.ReleaseResponse(fres)
}()
freq.SetRequestURI(req.Endpoint+ req.Path)
freq.Header.SetMethod(req.Method)
if req.BasicAuth.Username != ""{
freq.SetBasicAuth(req.BasicAuth.Username, req.BasicAuth.Password)
}
if req.Body != nil {
switch req.Body.(type) {
case []byte:
freq.SetBody(req.Body.([]byte))
case string:
freq.SetBody([]byte(req.Body.(string)))
case io.Reader:
freq.SetBodyStream(req.Body.(io.Reader), -1)
default:
rw := &bytes.Buffer{}
enc := json.NewEncoder(rw)
err := enc.Encode(req.Body)
if err != nil {
return nil, err
}
freq.SetBody(rw.Bytes())
}
}
client := &fasthttp.Client{
MaxConnsPerHost: 1000,
TLSConfig: &tls.Config{InsecureSkipVerify: true},
ReadTimeout: time.Second * 5,
}
err := client.Do(freq, fres)
if err != nil {
return nil, err
}
return &ProxyResponse{
Body: fres.Body(),
StatusCode: fres.StatusCode(),
}, nil
}