From de1f9c1e9419f40a2055a17f00e03a46de0c9408 Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Thu, 28 Sep 2023 15:57:30 +0800
Subject: [PATCH] remove upstream count

---
 main.go      | 13 +++----------
 structure.go |  9 ++-------
 2 files changed, 5 insertions(+), 17 deletions(-)

diff --git a/main.go b/main.go
index a76d20f..4b64d98 100644
--- a/main.go
+++ b/main.go
@@ -67,9 +67,9 @@ func main() {
 	if *listMode {
 		result := make([]OPENAI_UPSTREAM, 0)
 		db.Find(&result)
-		fmt.Println("SK\tEndpoint\tSuccess\tFailed\tLast Success Time")
+		fmt.Println("SK\tEndpoint")
 		for _, upstream := range result {
-			fmt.Println(upstream.SK, upstream.Endpoint, upstream.SuccessCount, upstream.FailedCount, upstream.LastCallSuccessTime)
+			fmt.Println(upstream.SK, upstream.Endpoint)
 		}
 		return
 	}
@@ -227,16 +227,9 @@ func main() {
 			// send notification
 			upstreams := []OPENAI_UPSTREAM{}
 			db.Find(&upstreams)
-			upstreamDescriptions := make([]string, 0)
-			for _, upstream := range upstreams {
-				upstreamDescriptions = append(upstreamDescriptions, fmt.Sprintf("ID: %d, %s: %s 成功次数: %d, 失败次数: %d, 最后成功调用: %s",
-					upstream.ID, upstream.SK, upstream.Endpoint, upstream.SuccessCount, upstream.FailedCount, upstream.LastCallSuccessTime,
-				))
-			}
-			content := fmt.Sprintf("[%s] OpenAI 转发出错 ID: %d... 密钥: [%s] 上游: [%s] 错误: %s\n---\n%s",
+			content := fmt.Sprintf("[%s] OpenAI 转发出错 ID: %d... 密钥: [%s] 上游: [%s] 错误: %s",
 				c.ClientIP(),
 				upstream.ID, upstream.SK, upstream.Endpoint, err.Error(),
-				strings.Join(upstreamDescriptions, "\n"),
 			)
 			go sendMatrixMessage(content)
 			if err.Error() != "context canceled" && r.Response.StatusCode != 400 {
diff --git a/structure.go b/structure.go
index ca29e9b..08509ca 100644
--- a/structure.go
+++ b/structure.go
@@ -1,17 +1,12 @@
 package main
 
 import (
-	"time"
-
 	"gorm.io/gorm"
 )
 
 // one openai upstream contain a pair of key and endpoint
 type OPENAI_UPSTREAM struct {
 	gorm.Model
-	SK                  string `gorm:"index:idx_sk_endpoint,unique"` // key
-	Endpoint            string `gorm:"index:idx_sk_endpoint,unique"` // endpoint
-	SuccessCount        int64
-	FailedCount         int64
-	LastCallSuccessTime time.Time
+	SK       string `gorm:"index:idx_sk_endpoint,unique"` // key
+	Endpoint string `gorm:"index:idx_sk_endpoint,unique"` // endpoint
 }