api2gpt-mid/api/server.go

package api

import (
	"api2gpt-mid/model"
	"bytes"
	"encoding/json"
	"io"
	"net/http"
	"os"
)

// BalanceConsumption 余额消费
func BalanceConsumption(key string, modelStr string, prompt_tokens int, completion_tokens int, total_tokens int, msg_id string) (string, error) {
	var data = model.Consumption{
		SecretKey:        key,
		Model:            modelStr,
		MsgId:            msg_id,
		PromptTokens:     prompt_tokens,
		CompletionTokens: completion_tokens,
		TotalTokens:      total_tokens,
	}

	jsonData, err := json.Marshal(data)
	// 构造post请求的body
	reqBody := bytes.NewBuffer(jsonData)

	url := os.Getenv("SERVER_API_USAGE_RECORD_STRING")
	if url == "" {
		url = "http://172.17.0.1:8080/other/usageRecord"
	}
	req2, err := http.NewRequest("POST", url, reqBody)

	// 设置http请求的header
	req2.Header.Set("Content-Type", "application/json")

	client := &http.Client{}
	resp, err := client.Do(req2)
	if err != nil {
		return "", err
	}
	defer func(Body io.ReadCloser) {
		err := Body.Close()
		if err != nil {

		}
	}(resp.Body)

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return "", err
	}
	return string(body), nil
}