初次提交

backend/main.go

@@ -0,0 +1,88 @@
+// 本文件功能为根据分页参数从数据库(用了gorm)中查询gpt对话历史记录，并通过http返回给前端json
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+	"log"
+	"net/http"
+	"strconv"
+)
+
+type Request struct {
+	gorm.Model
+	Ip                 string // 来访者IP
+	Url                string // 请求的URL
+	Stream             bool   // 是否是流式请求
+	APIModel           string // 使用的 openapi 模型
+	TokenKey           string `json:"TokenKey"` // api key
+	ResponseCode       int    // 返回的状态码
+	Error              string `json:"Error"`        // 错误信息
+	RequestData        string `json:"RequestData"`  // 请求内容
+	ResponseData       string `json:"ResponseData"` // 返回内容
+	CompletionDuration int64  `json:"duration"`     // 请求耗时
+}
+
+func main() {
+	dsn := "host=172.17.0.2 user=postgres password=mysecretpassword dbname=openai port=5432 sslmode=disable TimeZone=Asia/Shanghai"
+	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
+	if err != nil {
+		log.Fatalf("Error open db: %v", err)
+	}
+	db.AutoMigrate(&Request{})
+	http.HandleFunc("/requests/api", func(w http.ResponseWriter, r *http.Request) {
+
+		// 获取参数
+		page, _ := strconv.Atoi(r.URL.Query().Get("page"))     // 页数
+		pageSize, _ := strconv.Atoi(r.URL.Query().Get("size")) // 分页大小
+
+		// 参数小于零置为1
+		if page <= 0 {
+			page = 1
+		}
+		if pageSize <= 0 {
+			pageSize = 10
+		}
+
+		// 计算偏移量
+		offset := (page - 1) * pageSize
+
+		var requests []Request
+
+		// 根据偏移量查记录
+		db.Offset(offset).Limit(pageSize).Find(&requests)
+		w.Header().Set("Content-Type", "application/json")
+
+		// 计算总数
+		var totalRecords int64
+		db.Model(&Request{}).Count(&totalRecords)
+
+		// 计算总页数
+		totalPages := (totalRecords + int64(pageSize) - 1) / int64(pageSize)
+
+		// 返回的分页数据
+		pagination := map[string]interface{}{
+			"page":         page,
+			"pageSize":     pageSize,
+			"totalRecords": totalRecords,
+			"totalPages":   totalPages,
+		}
+		fmt.Println(requests)
+		// 返回的数据
+		responseDataWithPagination, err := json.Marshal(map[string]interface{}{
+			"data":       requests,
+			"pagination": pagination,
+		})
+		if err != nil {
+			http.Error(w, "Failed to marshal JSON", http.StatusInternalServerError)
+			return
+		}
+		w.Header().Set("Access-Control-Allow-Origin", "*")
+		w.Header().Set("Access-Control-Allow-Methods", "GET, OPTIONS")
+		w.WriteHeader(http.StatusOK)
+		w.Write(responseDataWithPagination)
+	})
+	http.ListenAndServe(":3002", nil)
+}