// Code generated by gowebx, DO AVOID EDIT.
package audioencodedb

import (
	"context"
	"easyvqd/internal/core/vqd"
	"git.lnton.com/lnton/pkg/orm"
)

var _ vqd.VqdTaskStorer = VqdTask{}

// VqdTask Related business namespaces
type VqdTask DB

// FindAll implements vqd.VqdTaskStorer.
func (d VqdTask) FindAll(bs *[]*vqd.VqdTask) (int64, error) {
	db := d.db.Model(&vqd.VqdTask{})
	var total int64
	if err := db.Count(&total).Error; err != nil || total <= 0 {
		// 如果统计失败或者数量为0，则返回错误
		return 0, err
	}
	return total, db.Find(bs).Error
}

// Find implements vqd.VqdTaskStorer.
func (d VqdTask) Find(ctx context.Context, bs *[]*vqd.VqdTask, page orm.Pager, opts ...orm.QueryOption) (int64, error) {
	return orm.FindWithContext(ctx, d.db, bs, page, opts...)
}

// Get implements vqd.VqdTaskStorer.
func (d VqdTask) Get(ctx context.Context, model *vqd.VqdTask, opts ...orm.QueryOption) error {
	return orm.FirstWithContext(ctx, d.db, model, opts...)
}

// Add implements vqd.VqdTaskStorer.
func (d VqdTask) Add(ctx context.Context, model *vqd.VqdTask) error {
	return d.db.WithContext(ctx).Create(model).Error
}

// Edit implements vqd.VqdTaskStorer.
func (d VqdTask) Edit(ctx context.Context, model *vqd.VqdTask, changeFn func(*vqd.VqdTask), opts ...orm.QueryOption) error {
	return orm.UpdateWithContext(ctx, d.db, model, changeFn, opts...)
}

// Del implements vqd.VqdTaskStorer.
func (d VqdTask) Del(ctx context.Context, model *vqd.VqdTask, opts ...orm.QueryOption) error {
	return orm.DeleteWithContext(ctx, d.db, model, opts...)
}

// EditStatus implements vqd.VqdTaskStorer.
func (d VqdTask) EditStatus(status int, id int) error {
	if err := d.db.Model(&vqd.VqdTask{}).Where(`id = ?`, id).Update("task_status", status).Error; err != nil {
		return err
	}
	return nil
}

// EditStatusError implements vqd.VqdTaskStorer.
func (d VqdTask) EditStatusError(id, status int, s string) error {
	if err := d.db.Model(&vqd.VqdTask{}).Where(`id = ?`, id).Updates(map[string]any{
		"task_status": status,
		"error_msg":   s,
	}).Error; err != nil {
		return err
	}
	return nil
}
