package db

import (
	"context"
	"database/sql"
)

// DB 定义数据库连接接口
type DB interface {
	// 获取原始数据库连接
	GetDB() *sql.DB

	// 执行查询，返回多行结果
	Query(ctx context.Context, query string, args ...interface{}) (*sql.Rows, error)

	// 执行查询，返回单行结果
	QueryRow(ctx context.Context, query string, args ...interface{}) *sql.Row

	// 执行更新操作（INSERT、UPDATE、DELETE）
	Exec(ctx context.Context, query string, args ...interface{}) (sql.Result, error)

	// 开始事务
	BeginTx(ctx context.Context, opts *sql.TxOptions) (*sql.Tx, error)

	// 关闭数据库连接
	Close() error
}

// Config 数据库配置结构体
type Config struct {
	Type     string
	Host     string
	Port     int
	Username string
	Password string
	Database string
	SSLMode  string
	Charset  string
}