feat: incr sync version.
This commit is contained in:
Gordon
103
go/chao-sdk-core/pkg/network/http_client.go
Normal file
103
go/chao-sdk-core/pkg/network/http_client.go
Normal file
@@ -0,0 +1,103 @@
|
||||
// Copyright © 2023 OpenIM SDK. All rights reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package network
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"net"
|
||||
|
||||
"github.com/openimsdk/openim-sdk-core/v3/pkg/utils"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
func get(url string) (response []byte, err error) {
|
||||
client := http.Client{Timeout: 5 * time.Second}
|
||||
resp, err := client.Get(url)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return body, nil
|
||||
}
|
||||
func retry(url string, data interface{}, token string, attempts int, sleep time.Duration, fn func(string, interface{}, string) ([]byte, error)) ([]byte, error) {
|
||||
b, err := fn(url, data, token)
|
||||
if err != nil {
|
||||
if attempts--; attempts > 0 {
|
||||
time.Sleep(sleep)
|
||||
return retry(url, data, token, attempts, 2*sleep, fn)
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return b, nil
|
||||
}
|
||||
|
||||
// application/json; charset=utf-8
|
||||
func Post2Api(url string, data interface{}, token string) (content []byte, err error) {
|
||||
c, err := postLogic(url, data, token)
|
||||
return c, utils.Wrap(err, " post")
|
||||
return retry(url, data, token, 1, 10*time.Second, postLogic)
|
||||
}
|
||||
|
||||
func Post2ApiForRead(url string, data interface{}, token string) (content []byte, err error) {
|
||||
return retry(url, data, token, 3, 10*time.Second, postLogic)
|
||||
}
|
||||
|
||||
func postLogic(url string, data interface{}, token string) (content []byte, err error) {
|
||||
jsonStr, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
return nil, utils.Wrap(err, "marshal failed, url")
|
||||
}
|
||||
|
||||
req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonStr))
|
||||
if err != nil {
|
||||
return nil, utils.Wrap(err, "newRequest failed, url")
|
||||
}
|
||||
req.Close = true
|
||||
req.Header.Add("content-type", "application/json")
|
||||
req.Header.Add("token", token)
|
||||
req.Header.Add("OperationID", utils.OperationIDGenerator())
|
||||
tp := &http.Transport{
|
||||
DialContext: (&net.Dialer{
|
||||
KeepAlive: 10 * time.Minute,
|
||||
}).DialContext,
|
||||
ResponseHeaderTimeout: 60 * time.Second,
|
||||
MaxIdleConnsPerHost: 100,
|
||||
IdleConnTimeout: 60 * time.Second,
|
||||
TLSHandshakeTimeout: 10 * time.Second,
|
||||
}
|
||||
client := &http.Client{Timeout: 60 * time.Second, Transport: tp}
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
return nil, utils.Wrap(err, "client.Do failed, url")
|
||||
}
|
||||
if resp.StatusCode != 200 {
|
||||
return nil, utils.Wrap(errors.New(resp.Status), "status code failed "+url)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
result, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, utils.Wrap(err, "ioutil.ReadAll failed, url")
|
||||
}
|
||||
// fmt.Println(url, "Marshal data: ", string(jsonStr), string(result))
|
||||
return result, nil
|
||||
}
|
||||
165
go/chao-sdk-core/pkg/network/new_http.go
Normal file
165
go/chao-sdk-core/pkg/network/new_http.go
Normal file
@@ -0,0 +1,165 @@
|
||||
// Copyright © 2023 OpenIM SDK. All rights reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package network
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"github.com/openimsdk/openim-sdk-core/v3/pkg/utils"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"time"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
type HttpCli struct {
|
||||
httpClient *http.Client
|
||||
httpRequest *http.Request
|
||||
Error error
|
||||
}
|
||||
|
||||
func newHttpClient() *http.Client {
|
||||
return &http.Client{Timeout: 30 * time.Second}
|
||||
}
|
||||
|
||||
func PostWithTimeOut(url string, data interface{}, token string, timeout time.Duration) (content []byte, err error) {
|
||||
return Post(url).BodyWithJson(data).SetTimeOut(timeout).SetHeader("token", token).ToBytes()
|
||||
}
|
||||
|
||||
func Get(url string) *HttpCli {
|
||||
request, err := http.NewRequest("GET", url, nil)
|
||||
return &HttpCli{
|
||||
httpClient: newHttpClient(),
|
||||
httpRequest: request,
|
||||
Error: err,
|
||||
}
|
||||
}
|
||||
|
||||
func Post(url string) *HttpCli {
|
||||
request, err := http.NewRequest("POST", url, nil)
|
||||
return &HttpCli{
|
||||
httpClient: newHttpClient(),
|
||||
httpRequest: request,
|
||||
Error: utils.Wrap(err, "newRequest failed, url"),
|
||||
}
|
||||
}
|
||||
|
||||
func (c *HttpCli) SetTimeOut(timeout time.Duration) *HttpCli {
|
||||
c.httpClient.Timeout = timeout
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *HttpCli) SetHeader(key, value string) *HttpCli {
|
||||
c.httpRequest.Header.Set(key, value)
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *HttpCli) BodyWithJson(obj interface{}) *HttpCli {
|
||||
if c.Error != nil {
|
||||
return c
|
||||
}
|
||||
|
||||
buf, err := json.Marshal(obj)
|
||||
if err != nil {
|
||||
c.Error = utils.Wrap(err, "marshal failed, url")
|
||||
return c
|
||||
}
|
||||
c.httpRequest.Body = ioutil.NopCloser(bytes.NewReader(buf))
|
||||
c.httpRequest.ContentLength = int64(len(buf))
|
||||
c.httpRequest.Header.Set("Content-Type", "application/json")
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *HttpCli) BodyWithBytes(buf []byte) *HttpCli {
|
||||
if c.Error != nil {
|
||||
return c
|
||||
}
|
||||
|
||||
c.httpRequest.Body = ioutil.NopCloser(bytes.NewReader(buf))
|
||||
c.httpRequest.ContentLength = int64(len(buf))
|
||||
return c
|
||||
}
|
||||
|
||||
func (c *HttpCli) BodyWithForm(form map[string]string) *HttpCli {
|
||||
if c.Error != nil {
|
||||
return c
|
||||
}
|
||||
|
||||
var value url.Values = make(map[string][]string, len(form))
|
||||
for k, v := range form {
|
||||
value.Add(k, v)
|
||||
}
|
||||
buf := Str2bytes(value.Encode())
|
||||
|
||||
c.httpRequest.Body = ioutil.NopCloser(bytes.NewReader(buf))
|
||||
c.httpRequest.ContentLength = int64(len(buf))
|
||||
c.httpRequest.Header.Add("Content-Type", "application/x-www-form-urlencoded")
|
||||
return c
|
||||
}
|
||||
|
||||
func Str2bytes(s string) []byte {
|
||||
x := (*[2]uintptr)(unsafe.Pointer(&s))
|
||||
h := [3]uintptr{x[0], x[1], x[1]}
|
||||
return *(*[]byte)(unsafe.Pointer(&h))
|
||||
}
|
||||
|
||||
func (c *HttpCli) ToBytes() (content []byte, err error) {
|
||||
if c.Error != nil {
|
||||
return nil, c.Error
|
||||
}
|
||||
|
||||
resp, err := c.httpClient.Do(c.httpRequest)
|
||||
if err != nil {
|
||||
return nil, utils.Wrap(err, "client.Do failed, url")
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return nil, utils.Wrap(errors.New(resp.Status), "status code failed ")
|
||||
}
|
||||
buf, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, utils.Wrap(err, "ioutil.ReadAll failed, url")
|
||||
}
|
||||
|
||||
return buf, nil
|
||||
}
|
||||
|
||||
func (c *HttpCli) ToJson(obj interface{}) error {
|
||||
if c.Error != nil {
|
||||
return c.Error
|
||||
}
|
||||
|
||||
resp, err := c.httpClient.Do(c.httpRequest)
|
||||
if err != nil {
|
||||
return utils.Wrap(err, "client.Do failed, url")
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return utils.Wrap(errors.New(resp.Status), "status code failed ")
|
||||
}
|
||||
buf, err := ioutil.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return utils.Wrap(err, "ioutil.ReadAll failed, url")
|
||||
}
|
||||
err = json.Unmarshal(buf, obj)
|
||||
if err != nil {
|
||||
return utils.Wrap(err, "marshal failed, url")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user