复制项目

pkg/common/imapi/call.go

@@ -0,0 +1,160 @@
+// Copyright © 2023 OpenIM open source community. 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 imapi
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"io"
+	"net/http"
+	"net/url"
+	"time"
+
+	"git.imall.cloud/openim/chat/pkg/common/constant"
+	constantpb "git.imall.cloud/openim/protocol/constant"
+	"github.com/openimsdk/tools/errs"
+	"github.com/openimsdk/tools/log"
+	"gorm.io/gorm/utils"
+)
+
+type baseApiResponse[T any] struct {
+	ErrCode int    `json:"errCode"`
+	ErrMsg  string `json:"errMsg"`
+	ErrDlt  string `json:"errDlt"`
+	Data    *T     `json:"data"`
+}
+
+var client = &http.Client{
+	Timeout: time.Second * 10,
+}
+
+type ApiCaller[Req, Resp any] interface {
+	Call(ctx context.Context, apiPrefix string, req *Req) (*Resp, error)
+	CallWithQuery(ctx context.Context, apiPrefix string, req *Req, queryParams map[string]string) (*Resp, error)
+}
+
+func NewApiCaller[Req, Resp any](api string) ApiCaller[Req, Resp] {
+	return &caller[Req, Resp]{
+		api: api,
+	}
+}
+
+type caller[Req, Resp any] struct {
+	api string
+}
+
+func (a caller[Req, Resp]) Call(ctx context.Context, apiPrefix string, req *Req) (*Resp, error) {
+	start := time.Now()
+	resp, err := a.call(ctx, apiPrefix, req)
+	if err != nil {
+		log.ZError(ctx, "api caller failed", err, "api", a.api, "duration", time.Since(start), "req", req, "resp", resp)
+		return nil, err
+	}
+	log.ZInfo(ctx, "api caller success resp", "api", a.api, "duration", time.Since(start), "req", req, "resp", resp)
+	return resp, nil
+}
+
+func (a caller[Req, Resp]) call(ctx context.Context, apiPrefix string, req *Req) (*Resp, error) {
+	url := apiPrefix + a.api
+	reqBody, err := json.Marshal(req)
+	if err != nil {
+		return nil, err
+	}
+	request, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(reqBody))
+	if err != nil {
+		return nil, err
+	}
+	operationID := utils.ToString(ctx.Value(constantpb.OperationID))
+	request.Header.Set(constantpb.OperationID, operationID)
+	if token, _ := ctx.Value(constant.CtxApiToken).(string); token != "" {
+		request.Header.Set(constantpb.Token, token)
+	}
+	response, err := client.Do(request)
+	if err != nil {
+		return nil, err
+	}
+	defer response.Body.Close()
+	data, err := io.ReadAll(response.Body)
+	if err != nil {
+		return nil, errs.WrapMsg(err, "read http response body", "url", url, "code", response.StatusCode)
+	}
+	var resp baseApiResponse[Resp]
+	if err := json.Unmarshal(data, &resp); err != nil {
+		return nil, errs.WrapMsg(err, string(data))
+	}
+	if resp.ErrCode != 0 {
+		return nil, errs.NewCodeError(resp.ErrCode, resp.ErrMsg).WithDetail(resp.ErrDlt).Wrap()
+	}
+	return resp.Data, nil
+}
+
+func (a caller[Req, Resp]) CallWithQuery(ctx context.Context, apiPrefix string, req *Req, queryParams map[string]string) (*Resp, error) {
+	start := time.Now()
+	resp, err := a.callWithQuery(ctx, apiPrefix, req, queryParams)
+	if err != nil {
+		log.ZError(ctx, "api caller failed", err, "api", a.api, "duration", time.Since(start), "req", req, "resp", resp)
+		return nil, err
+	}
+	log.ZInfo(ctx, "api caller success resp", "api", a.api, "duration", time.Since(start), "req", req, "resp", resp)
+	return resp, nil
+}
+
+func (a caller[Req, Resp]) callWithQuery(ctx context.Context, apiPrefix string, req *Req, queryParams map[string]string) (*Resp, error) {
+	fullURL := apiPrefix + a.api
+	parsedURL, err := url.Parse(fullURL)
+	if err != nil {
+		return nil, errs.WrapMsg(err, "failed to parse URL", fullURL)
+	}
+
+	query := parsedURL.Query()
+
+	for key, value := range queryParams {
+		query.Set(key, value)
+	}
+
+	parsedURL.RawQuery = query.Encode()
+	fullURL = parsedURL.String()
+	reqBody, err := json.Marshal(req)
+	if err != nil {
+		return nil, err
+	}
+	request, err := http.NewRequestWithContext(ctx, http.MethodPost, fullURL, bytes.NewReader(reqBody))
+	if err != nil {
+		return nil, err
+	}
+	operationID := utils.ToString(ctx.Value(constantpb.OperationID))
+	request.Header.Set(constantpb.OperationID, operationID)
+	if token, _ := ctx.Value(constant.CtxApiToken).(string); token != "" {
+		request.Header.Set(constantpb.Token, token)
+	}
+	response, err := client.Do(request)
+	if err != nil {
+		return nil, err
+	}
+	defer response.Body.Close()
+	data, err := io.ReadAll(response.Body)
+	if err != nil {
+		return nil, errs.WrapMsg(err, "read http response body", "fullUrl", fullURL, "code", response.StatusCode)
+	}
+	var resp baseApiResponse[Resp]
+	if err := json.Unmarshal(data, &resp); err != nil {
+		return nil, errs.WrapMsg(err, string(data))
+	}
+	if resp.ErrCode != 0 {
+		return nil, errs.NewCodeError(resp.ErrCode, resp.ErrMsg).WithDetail(resp.ErrDlt).Wrap()
+	}
+	return resp.Data, nil
+}