| @@ -565,22 +565,26 @@ func ReferenceDataset(ctx *context.Context) { | |||||
| func ReferenceDatasetData(ctx *context.Context) { | func ReferenceDatasetData(ctx *context.Context) { | ||||
| MustEnableDataset(ctx) | MustEnableDataset(ctx) | ||||
| datasetIds := models.GetDatasetIdsByRepoID(ctx.Repo.Repository.ID) | 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)) | ctx.JSON(http.StatusOK, repository.ConvertToDatasetWithStar(ctx, datasets)) | ||||