package service

import (
	"context"
	"rag/dao"
	"rag/model/dto"

	"github.com/gogf/gf/v2/util/gconv"
)

var Dataset = new(datasetService)

type datasetService struct{}

// Create 创建数据集
func (s *datasetService) Create(ctx context.Context, req *dto.CreateDatasetReq) (res *dto.CreateDatasetRes, err error) {
	id, err := dao.Dataset.Insert(ctx, req)
	if err != nil {
		return
	}
	return &dto.CreateDatasetRes{Id: id}, nil
}

// Update 更新数据集
func (s *datasetService) Update(ctx context.Context, req *dto.UpdateDatasetReq) (err error) {
	_, err = dao.Dataset.Update(ctx, req)
	return
}

// Delete 删除数据集
func (s *datasetService) Delete(ctx context.Context, req *dto.DeleteDatasetReq) (err error) {
	_, err = dao.Dataset.Delete(ctx, req)
	return
}

// List 数据集列表
func (s *datasetService) List(ctx context.Context, req *dto.ListDatasetReq) (res *dto.ListDatasetRes, err error) {
	list, total, err := dao.Dataset.List(ctx, req)
	if err != nil {
		return
	}
	res = &dto.ListDatasetRes{
		Total: total,
	}
	err = gconv.Struct(list, &res.List)
	return
}