package rdb

import (
	"gorm.io/gorm"
)

// RDB gorm关系型数据
type RDB struct {
	db *gorm.DB
}

func NewRDB(db *gorm.DB) *RDB {
	return &RDB{
		db: db,
	}
}

func (m *RDB) Init() (err error) {
	return
}

func (m *RDB) Insert(data any) (err error) {
	return m.db.Create(data).Error
}

// sql查询数据
func (m *RDB) Select(ret any, sql string, args ...any) (err error) {
	return m.db.Raw(sql, args...).Scan(ret).Error
}

// sql更新数据
func (m *RDB) Update(sql string, args ...any) (rowsAffected int64, err error) {
	tx := m.db.Exec(sql, args...)
	rowsAffected, err = tx.RowsAffected, tx.Error
	return
}

// sql更新数据
func (m *RDB) UpdateBy(data any) (rowsAffected int64, err error) {
	tx := m.db.Model(data).Updates(data)
	rowsAffected, err = tx.RowsAffected, tx.Error
	return
}