package model

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

var _ CbServicerModel = (*customCbServicerModel)(nil)

type (
	// CbServicerModel is an interface to be customized, add more methods here,
	// and implement the added methods in customCbServicerModel.
	CbServicerModel interface {
		cbServicerModel
		GetServicer(openKfid, userid string) (d *CbServicer, err error)
		DeleteServicerByOpenKfid(openKfid string) (err error)
	}

	customCbServicerModel struct {
		*defaultCbServicerModel
	}
)

// NewCbServicerModel returns a model for the database table.
func NewCbServicerModel(conn sqlx.SqlConn) CbServicerModel {
	return &customCbServicerModel{
		defaultCbServicerModel: newCbServicerModel(conn),
	}
}

func (m *customCbServicerModel) GetServicer(openKfid, userid string) (d *CbServicer, err error) {
	query := fmt.Sprintf("select * from %s where `open_kfid` = ? AND `userid` = ? limit 1", m.table)
	var resp CbServicer
	err = m.conn.QueryRow(&resp, query, openKfid, userid)
	d = &resp
	return
}

func (m *customCbServicerModel) DeleteServicerByOpenKfid(openKfid string) (err error) {
	query := fmt.Sprintf("delete from %s where `open_kfid` = ?", m.table)
	_, err = m.conn.Exec(query, openKfid)
	return err
}