package functions

import (
	"fmt"

	"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("unexpected test result: %d", result)
	}

	fmt.Println("test Connection  database is success.")
	return nil
}