package bzgorm

import (
	"database/sql"
	"gorm.io/gorm"
)

func DB(db *gorm.DB) (*sql.DB, error) {
	return db.DB()
}

func DBPing(db *gorm.DB) error {
	s, err := DB(db)
	if err != nil {
		return err
	}
	return s.Ping()
}

func DBClose(db *gorm.DB) error {
	s, err := DB(db)
	if err != nil {
		return err
	}
	return s.Close()
}

func DBStats(db *gorm.DB) (sql.DBStats, error) {
	s, err := DB(db)
	if err != nil {
		return sql.DBStats{}, err
	}

	// 返回数据库统计信息
	return s.Stats(), nil
}