package controller

import (
	"context"

	"rag/model/dto"
	"rag/service"

	"gitea.com/red-future/common/beans"
	"github.com/gogf/gf/v2/frame/g"
)

type dataset struct{}

var Dataset = new(dataset)

// Create 创建数据集
func (c *dataset) Create(ctx context.Context, req *dto.CreateDatasetReq) (res *dto.CreateDatasetRes, err error) {
	res, err = service.Dataset.Create(ctx, req)
	return
}

// Update 更新数据集
func (c *dataset) Update(ctx context.Context, req *dto.UpdateDatasetReq) (res *beans.ResponseEmpty, err error) {
	err = service.Dataset.Update(ctx, req)
	return
}

// Delete 删除数据集
func (c *dataset) Delete(ctx context.Context, req *dto.DeleteDatasetReq) (res *beans.ResponseEmpty, err error) {
	err = service.Dataset.Delete(ctx, req)
	return
}

// List 数据集列表
func (c *dataset) List(ctx context.Context, req *dto.ListDatasetReq) (res *dto.ListDatasetRes, err error) {
	if !g.IsEmpty(req.Page) {
		req.Page = &beans.Page{PageNum: 1, PageSize: 20}
	}
	res, err = service.Dataset.List(ctx, req)
	return
}

// Search 搜索
//func (c *dataset) Search(ctx context.Context, req *dto.SearchReq) (res *dto.SearchRes, err error) {
//	res, err = service.Dataset.Search(ctx, req)
//	return
//}