package model

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

var _ NhTaskNftStakeRewardModel = (*customNhTaskNftStakeRewardModel)(nil)

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

	customNhTaskNftStakeRewardModel struct {
		*defaultNhTaskNftStakeRewardModel
	}
)

// NewNhTaskNftStakeRewardModel returns a model for the database table.
func NewNhTaskNftStakeRewardModel(conn sqlx.SqlConn) NhTaskNftStakeRewardModel {
	return &customNhTaskNftStakeRewardModel{
		defaultNhTaskNftStakeRewardModel: newNhTaskNftStakeRewardModel(conn),
	}
}

func (m *customNhTaskNftStakeRewardModel) withSession(session sqlx.Session) NhTaskNftStakeRewardModel {
	return NewNhTaskNftStakeRewardModel(sqlx.NewSqlConnFromSession(session))
}