go-db/function/testConnection.go

package function

import (
	"fmt"
	"log"

	"github.com/jmoiron/sqlx"
)

// TestConnection 测试数据库连接
func TestConnection(db *sqlx.DB, dbType string) error {
	var query string
	switch dbType {
	case "mysql", "postgres", "sqlserver":
		query = "SELECT 1"
	case "oracle":
		query = "SELECT 1 FROM DUAL"
	default:
		query = "SELECT 1"
	}

	var result int
	err := db.QueryRow(query).Scan(&result)
	if err != nil {
		return err
	}

	if result != 1 {
		return fmt.Errorf("%s unexpected test result: %d", dbType, result)
	}

	log.Printf("test Connection  database is success. %s", dbType)
	return nil
}