package model

import "github.com/zeromicro/go-zero/core/stores/sqlx"

var _ ImagesModel = (*customImagesModel)(nil)

type (
	// ImagesModel is an interface to be customized, add more methods here,
	// and implement the added methods in customImagesModel.
	ImagesModel interface {
		imagesModel
		withSession(session sqlx.Session) ImagesModel
	}

	customImagesModel struct {
		*defaultImagesModel
	}
)

// NewImagesModel returns a model for the database table.
func NewImagesModel(conn sqlx.SqlConn) ImagesModel {
	return &customImagesModel{
		defaultImagesModel: newImagesModel(conn),
	}
}

func (m *customImagesModel) withSession(session sqlx.Session) ImagesModel {
	return NewImagesModel(sqlx.NewSqlConnFromSession(session))
}