feat: add limit history testcase amount

Author: SamYSF

cmd/root.go

@@ -39,7 +39,7 @@ func NewRootCommand() (c *cobra.Command) {
 func (o *option) runE(c *cobra.Command, args []string) (err error) {
 	if o.version {
 		c.Println(version.GetVersion())
-		//c.Println(version.GetDate())
+		c.Println(version.GetDate())
 		return
 	}
 	remoteServer := pkg.NewRemoteServer()

go.mod

@@ -1,13 +1,14 @@
 module github.com/linuxsuren/atest-ext-store-orm
 
-go 1.22.2
+go 1.22.4
 
-toolchain go1.22.4
+toolchain go1.22.6
 
 require (
 	github.com/linuxsuren/api-testing v0.0.18-0.20240712143814-6ce9363d5a07
 	github.com/spf13/cobra v1.8.0
 	github.com/stretchr/testify v1.9.0
+	google.golang.org/protobuf v1.33.0
 	gorm.io/driver/mysql v1.5.2
 	gorm.io/driver/postgres v1.5.4
 	gorm.io/driver/sqlite v1.5.6
@@ -53,6 +54,7 @@ require (
 	github.com/jinzhu/now v1.1.5 // indirect
 	github.com/josharian/intern v1.0.0 // indirect
 	github.com/linuxsuren/go-fake-runtime v0.0.4 // indirect
+	github.com/linuxsuren/oauth-hub v0.0.0-20240809060240-e78c21b5d8d4 // indirect
 	github.com/linuxsuren/unstructured v0.0.1 // indirect
 	github.com/mailru/easyjson v0.7.7 // indirect
 	github.com/mattn/go-sqlite3 v1.14.22 // indirect
@@ -85,15 +87,15 @@ require (
 	go.uber.org/zap v1.27.0 // indirect
 	golang.org/x/crypto v0.21.0 // indirect
 	golang.org/x/net v0.23.0 // indirect
-	golang.org/x/oauth2 v0.18.0 // indirect
+	golang.org/x/oauth2 v0.22.0 // indirect
 	golang.org/x/sync v0.6.0 // indirect
 	golang.org/x/sys v0.18.0 // indirect
 	golang.org/x/text v0.14.0 // indirect
-	google.golang.org/appengine v1.6.8 // indirect
 	google.golang.org/genproto v0.0.0-20240123012728-ef4313101c80 // indirect
 	google.golang.org/genproto/googleapis/api v0.0.0-20240123012728-ef4313101c80 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20240123012728-ef4313101c80 // indirect
-	google.golang.org/protobuf v1.33.0 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
+
+replace github.com/linuxsuren/api-testing => github.com/SamYSF/api-testing v0.0.0-20240827074726-d0b782cb43d6

pkg/server.go

@@ -30,6 +30,7 @@ import (
 	"gorm.io/gorm"
 	"gorm.io/gorm/logger"
 	"log"
+	"strconv"
 	"strings"
 )
 
@@ -238,6 +239,22 @@ func (s *dbserver) CreateTestCaseHistory(ctx context.Context, historyTestResult
 		return
 	}
 
+	store := remote.GetStoreFromContext(ctx)
+	historyLimit := 10
+	if v, ok := store.Properties["historyLimit"]; ok {
+		if parsedHistoryLimit, parseErr := strconv.Atoi(v); parseErr == nil {
+			historyLimit = parsedHistoryLimit
+		}
+	}
+
+	var count int64
+	db.Model(&HistoryTestResult{}).Count(&count)
+
+	if count >= int64(historyLimit) {
+		fmt.Printf("Existing count: %d, limit: %d\nmaximum number of entries reached, cannot create new TestCaseHistory\n", count, historyLimit)
+		return
+	}
+
 	db.Create(ConvertToDBHistoryTestResult(historyTestResult))
 	return
 }
@@ -380,14 +397,14 @@ func (s *dbserver) Verify(ctx context.Context, in *server.Empty) (reply *server.
 	return
 }
 
-// func (s *dbserver) GetVersion(context.Context, *server.Empty) (ver *server.Version, err error) {
-// 	ver = &server.Version{
-// 		Version: version.GetVersion(),
-// 		Commit:  version.GetCommit(),
-// 		Date:    version.GetDate(),
-// 	}
-// 	return
-// }
+func (s *dbserver) GetVersion(context.Context, *server.Empty) (ver *server.Version, err error) {
+	ver = &server.Version{
+		Version: version.GetVersion(),
+		Commit:  version.GetCommit(),
+		Date:    version.GetDate(),
+	}
+	return
+}
 
 func (s *dbserver) PProf(ctx context.Context, in *server.PProfRequest) (data *server.PProfData, err error) {
 	log.Println("pprof", in.Name)