|
|
@@ -565,22 +565,26 @@ func ReferenceDataset(ctx *context.Context) { |
|
|
|
func ReferenceDatasetData(ctx *context.Context) { |
|
|
|
MustEnableDataset(ctx) |
|
|
|
datasetIds := models.GetDatasetIdsByRepoID(ctx.Repo.Repository.ID) |
|
|
|
opts := &models.SearchDatasetOptions{ |
|
|
|
DatasetIDs: datasetIds, |
|
|
|
NeedAttachment: false, |
|
|
|
CloudBrainType: models.TypeCloudBrainAll, |
|
|
|
ListOptions: models.ListOptions{ |
|
|
|
Page: 1, |
|
|
|
PageSize: setting.RepoMaxReferenceDatasetNum, |
|
|
|
}, |
|
|
|
SearchOrderBy: getSearchOrderByInValues(datasetIds), |
|
|
|
QueryReference: true, |
|
|
|
} |
|
|
|
datasets, _, err := models.SearchDataset(opts) |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
ctx.ServerError("SearchDatasets", err) |
|
|
|
return |
|
|
|
var datasets models.DatasetList |
|
|
|
var err error |
|
|
|
if len(datasetIds) > 0 { |
|
|
|
|
|
|
|
opts := &models.SearchDatasetOptions{ |
|
|
|
DatasetIDs: datasetIds, |
|
|
|
NeedAttachment: false, |
|
|
|
CloudBrainType: models.TypeCloudBrainAll, |
|
|
|
ListOptions: models.ListOptions{ |
|
|
|
Page: 1, |
|
|
|
PageSize: setting.RepoMaxReferenceDatasetNum, |
|
|
|
}, |
|
|
|
SearchOrderBy: getSearchOrderByInValues(datasetIds), |
|
|
|
QueryReference: true, |
|
|
|
} |
|
|
|
datasets, _, err = models.SearchDataset(opts) |
|
|
|
if err != nil { |
|
|
|
ctx.ServerError("SearchDatasets", err) |
|
|
|
return |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
ctx.JSON(http.StatusOK, repository.ConvertToDatasetWithStar(ctx, datasets)) |
|
|
|