diff --git a/models/dataset.go b/models/dataset.go index 04591c64e..6e34ee093 100755 --- a/models/dataset.go +++ b/models/dataset.go @@ -190,6 +190,7 @@ type SearchDatasetOptions struct { JustNeedZipFile bool NeedAttachment bool UploadAttachmentByMe bool + QueryReference bool } func CreateDataset(dataset *Dataset) (err error) { @@ -260,7 +261,7 @@ func SearchDatasetCondition(opts *SearchDatasetOptions) builder.Cond { } } if len(opts.DatasetIDs) > 0 { - if opts.StarByMe { + if opts.StarByMe || (opts.RepoID == 0 && opts.QueryReference) { cond = cond.And(builder.In("dataset.id", opts.DatasetIDs)) } else { subCon := builder.NewCond() diff --git a/routers/repo/dataset.go b/routers/repo/dataset.go index 51cb4893f..cbfed4666 100755 --- a/routers/repo/dataset.go +++ b/routers/repo/dataset.go @@ -573,7 +573,8 @@ func ReferenceDatasetData(ctx *context.Context) { Page: 1, PageSize: setting.RepoMaxReferenceDatasetNum, }, - SearchOrderBy: getSearchOrderByInValues(datasetIds), + SearchOrderBy: getSearchOrderByInValues(datasetIds), + QueryReference: true, } datasets, _, err := models.SearchDataset(opts)