fix: the database verify is not working

Author: LinuxSuRen

cmd/root.go

@@ -1,26 +1,17 @@
 /*
-*
-MIT License
+Copyright 2023-2024 API Testing Authors.
 
-Copyright (c) 2023 API Testing Authors.
+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
 
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
+	http://www.apache.org/licenses/LICENSE-2.0
 
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+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 cmd
 

pkg/server.go

@@ -1,5 +1,5 @@
 /*
-Copyright 2023 API Testing Authors.
+Copyright 2023-2024 API Testing Authors.
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
@@ -52,7 +52,10 @@ func createDB(user, password, address, database, driver string) (db *gorm.DB, er
 		dialector = mysql.Open(dsn)
 	case "postgres":
 		obj := strings.Split(address, ":")
-		host, port := obj[0], obj[1]
+		host, port := obj[0], "5432"
+		if len(obj) > 1 {
+			port = obj[1]
+		}
 		dsn = fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%s sslmode=disable TimeZone=Asia/Shanghai", host, user, password, database, port)
 		dialector = postgres.Open(dsn)
 	default:
@@ -252,10 +255,10 @@ func (s *dbserver) DeleteTestCase(ctx context.Context, testcase *server.TestCase
 }
 
 func (s *dbserver) Verify(ctx context.Context, in *server.Empty) (reply *server.ExtensionStatus, err error) {
-	db, clientErr := s.getClient(ctx)
+	_, vErr := s.ListTestSuite(ctx, in)
 	reply = &server.ExtensionStatus{
-		Ready:   err == nil && db != nil,
-		Message: util.OKOrErrorMessage(clientErr),
+		Ready:   vErr == nil,
+		Message: util.OKOrErrorMessage(vErr),
 		Version: version.GetVersion(),
 	}
 	return