protected SearchContext buildSearchContext(
long companyId,
long[] groupIds,
String title,
long[] parentCategoryIds,
long[] vocabularyIds,
int start,
int end,
Sort sort) {
SearchContext searchContext = new SearchContext();
Map<String, Serializable> attributes = new HashMap<>();
attributes.put(Field.ASSET_PARENT_CATEGORY_IDS, parentCategoryIds);
attributes.put(Field.ASSET_VOCABULARY_IDS, vocabularyIds);
attributes.put(Field.TITLE, title);
searchContext.setAttributes(attributes);
searchContext.setCompanyId(companyId);
searchContext.setEnd(end);
searchContext.setGroupIds(groupIds);
searchContext.setKeywords(title);
searchContext.setSorts(sort);
searchContext.setStart(start);
QueryConfig queryConfig = searchContext.getQueryConfig();
queryConfig.setHighlightEnabled(false);
queryConfig.setScoreEnabled(false);
return searchContext;
}
protected int searchBaseModelsCount(Class<?> clazz, long groupId, SearchContext searchContext)
throws Exception {
Indexer indexer = IndexerRegistryUtil.getIndexer(clazz);
searchContext.setGroupIds(new long[] {groupId});
Hits results = indexer.search(searchContext);
return results.getLength();
}
protected void search(
FileEntry fileEntry, boolean rootFolder, String keywords, boolean assertTrue)
throws Exception {
SearchContext searchContext = new SearchContext();
searchContext.setAttribute("paginationType", "regular");
searchContext.setCompanyId(fileEntry.getCompanyId());
searchContext.setFolderIds(new long[] {fileEntry.getFolderId()});
searchContext.setGroupIds(new long[] {fileEntry.getRepositoryId()});
searchContext.setKeywords(keywords);
QueryConfig queryConfig = new QueryConfig();
queryConfig.setHighlightEnabled(false);
queryConfig.setScoreEnabled(false);
searchContext.setQueryConfig(queryConfig);
Indexer indexer = IndexerRegistryUtil.getIndexer(DLFileEntryConstants.getClassName());
Hits hits = indexer.search(searchContext);
List<Document> documents = hits.toList();
boolean found = false;
for (Document document : documents) {
long fileEntryId = GetterUtil.getLong(document.get(Field.ENTRY_CLASS_PK));
if (fileEntryId == fileEntry.getFileEntryId()) {
found = true;
break;
}
}
String message = "Search engine could not find ";
if (rootFolder) {
message += "root file entry by " + keywords;
} else {
message += "file entry by " + keywords;
}
message += " using query " + hits.getQuery();
if (assertTrue) {
Assert.assertTrue(message, found);
} else {
Assert.assertFalse(message, found);
}
}
public Hits search(
long companyId,
long[] groupIds,
long userId,
String className,
String keywords,
int status,
int start,
int end)
throws SystemException {
try {
SearchContext searchContext = new SearchContext();
Facet assetEntriesFacet = new AssetEntriesFacet(searchContext);
assetEntriesFacet.setStatic(true);
searchContext.addFacet(assetEntriesFacet);
Facet scopeFacet = new ScopeFacet(searchContext);
scopeFacet.setStatic(true);
searchContext.addFacet(scopeFacet);
searchContext.setAttribute("paginationType", "regular");
searchContext.setAttribute("status", status);
searchContext.setCompanyId(companyId);
searchContext.setEnd(end);
searchContext.setEntryClassNames(getClassNames(className));
searchContext.setGroupIds(groupIds);
searchContext.setKeywords(keywords);
QueryConfig queryConfig = new QueryConfig();
queryConfig.setHighlightEnabled(false);
queryConfig.setScoreEnabled(false);
searchContext.setQueryConfig(queryConfig);
searchContext.setStart(start);
searchContext.setUserId(userId);
Indexer indexer = FacetedSearcher.getInstance();
return indexer.search(searchContext);
} catch (Exception e) {
throw new SystemException(e);
}
}
protected static AssetSearcher getAssetSearcher(
SearchContext searchContext, AssetEntryQuery assetEntryQuery, int start, int end)
throws Exception {
Indexer<?> searcher = AssetSearcher.getInstance();
AssetSearcher assetSearcher = (AssetSearcher) searcher;
assetSearcher.setAssetEntryQuery(assetEntryQuery);
Layout layout = assetEntryQuery.getLayout();
if (layout != null) {
searchContext.setAttribute(Field.LAYOUT_UUID, layout.getUuid());
}
String ddmStructureFieldName = (String) assetEntryQuery.getAttribute("ddmStructureFieldName");
Serializable ddmStructureFieldValue = assetEntryQuery.getAttribute("ddmStructureFieldValue");
if (Validator.isNotNull(ddmStructureFieldName) && Validator.isNotNull(ddmStructureFieldValue)) {
searchContext.setAttribute("ddmStructureFieldName", ddmStructureFieldName);
searchContext.setAttribute("ddmStructureFieldValue", ddmStructureFieldValue);
}
String paginationType = GetterUtil.getString(assetEntryQuery.getPaginationType(), "more");
if (!paginationType.equals("none") && !paginationType.equals("simple")) {
searchContext.setAttribute("paginationType", paginationType);
}
searchContext.setClassTypeIds(assetEntryQuery.getClassTypeIds());
searchContext.setEnd(end);
searchContext.setGroupIds(assetEntryQuery.getGroupIds());
if (Validator.isNotNull(assetEntryQuery.getKeywords())) {
searchContext.setLike(true);
}
searchContext.setSorts(getSorts(assetEntryQuery, searchContext.getLocale()));
searchContext.setStart(start);
return assetSearcher;
}
@Override
public Hits search(
long groupId,
long userId,
long creatorUserId,
long startDate,
long endDate,
int status,
int start,
int end)
throws PortalException {
Indexer<MBThread> indexer = IndexerRegistryUtil.getIndexer(MBThread.class.getName());
SearchContext searchContext = new SearchContext();
searchContext.setAttribute(Field.STATUS, status);
if (endDate > 0) {
searchContext.setAttribute("endDate", endDate);
}
searchContext.setAttribute("paginationType", "none");
if (creatorUserId > 0) {
searchContext.setAttribute("participantUserId", String.valueOf(creatorUserId));
}
if (startDate > 0) {
searchContext.setAttribute("startDate", startDate);
}
Group group = groupLocalService.getGroup(groupId);
searchContext.setCompanyId(group.getCompanyId());
searchContext.setEnd(end);
searchContext.setGroupIds(new long[] {groupId});
searchContext.setSorts(new Sort("lastPostDate", true));
searchContext.setStart(start);
searchContext.setUserId(userId);
return indexer.search(searchContext);
}
protected SearchContext buildSearchContext(
long companyId, long groupId, String title, int start, int end) {
SearchContext searchContext = new SearchContext();
searchContext.setAttribute(Field.TITLE, title);
searchContext.setCompanyId(companyId);
searchContext.setEnd(end);
searchContext.setGroupIds(new long[] {groupId});
searchContext.setKeywords(title);
searchContext.setStart(start);
QueryConfig queryConfig = searchContext.getQueryConfig();
queryConfig.setHighlightEnabled(false);
queryConfig.setScoreEnabled(false);
return searchContext;
}
protected void search(boolean rootFolder, String keywords) throws Exception {
SearchContext searchContext = new SearchContext();
searchContext.setCompanyId(_fileEntry.getCompanyId());
searchContext.setFolderIds(new long[] {_fileEntry.getFolderId()});
searchContext.setGroupIds(new long[] {_fileEntry.getRepositoryId()});
searchContext.setKeywords(keywords);
QueryConfig queryConfig = new QueryConfig();
queryConfig.setHighlightEnabled(false);
queryConfig.setScoreEnabled(false);
searchContext.setQueryConfig(queryConfig);
Indexer indexer = IndexerRegistryUtil.getIndexer(DLFileEntry.class);
List<Document> documents = indexer.search(searchContext).toList();
boolean found = false;
for (Document document : documents) {
long fileEntryId = GetterUtil.getLong(document.get(Field.ENTRY_CLASS_PK));
if (fileEntryId == _fileEntry.getFileEntryId()) {
found = true;
break;
}
}
String message = "Search engine could not find ";
if (rootFolder) {
message += "root file entry by " + keywords;
} else {
message += "file entry by " + keywords;
}
assertTrue(message, found);
}
public Hits search(
long companyId,
long userId,
String portletId,
long groupId,
long[] repositoryIds,
String keywords,
int start,
int end) {
try {
SearchContext searchContext = new SearchContext();
searchContext.setCompanyId(companyId);
searchContext.setEnd(end);
searchContext.setEntryClassNames(new String[] {DLFileEntryConstants.getClassName()});
searchContext.setGroupIds(new long[] {groupId});
Indexer indexer = IndexerRegistryUtil.getIndexer(DLFileEntryConstants.getClassName());
searchContext.setSearchEngineId(indexer.getSearchEngineId());
searchContext.setStart(start);
searchContext.setUserId(userId);
BooleanQuery contextQuery = BooleanQueryFactoryUtil.create(searchContext);
contextQuery.addRequiredTerm(Field.PORTLET_ID, portletId);
if (groupId > 0) {
Group group = groupLocalService.getGroup(groupId);
if (group.isLayout()) {
contextQuery.addRequiredTerm(Field.SCOPE_GROUP_ID, groupId);
groupId = group.getParentGroupId();
}
contextQuery.addRequiredTerm(Field.GROUP_ID, groupId);
}
if (ArrayUtil.isNotEmpty(repositoryIds)) {
BooleanQuery repositoryIdsQuery = BooleanQueryFactoryUtil.create(searchContext);
for (long repositoryId : repositoryIds) {
try {
if (userId > 0) {
PermissionChecker permissionChecker = PermissionThreadLocal.getPermissionChecker();
DLFolderPermission.check(permissionChecker, groupId, repositoryId, ActionKeys.VIEW);
}
if (repositoryId == DLFolderConstants.DEFAULT_PARENT_FOLDER_ID) {
repositoryId = groupId;
}
TermQuery termQuery =
TermQueryFactoryUtil.create(searchContext, "repositoryId", repositoryId);
repositoryIdsQuery.add(termQuery, BooleanClauseOccur.SHOULD);
} catch (Exception e) {
}
}
contextQuery.add(repositoryIdsQuery, BooleanClauseOccur.MUST);
}
BooleanQuery searchQuery = BooleanQueryFactoryUtil.create(searchContext);
searchQuery.addTerms(_KEYWORDS_FIELDS, keywords);
BooleanQuery fullQuery = BooleanQueryFactoryUtil.create(searchContext);
fullQuery.add(contextQuery, BooleanClauseOccur.MUST);
List<BooleanClause> clauses = searchQuery.clauses();
if (!clauses.isEmpty()) {
fullQuery.add(searchQuery, BooleanClauseOccur.MUST);
}
return SearchEngineUtil.search(searchContext, fullQuery);
} catch (Exception e) {
throw new SystemException(e);
}
}
public Hits search(
long companyId,
long[] groupIds,
long userId,
String className,
String userName,
String title,
String description,
String assetCategoryIds,
String assetTagNames,
int status,
boolean andSearch,
int start,
int end)
throws SystemException {
try {
SearchContext searchContext = new SearchContext();
Facet assetEntriesFacet = new AssetEntriesFacet(searchContext);
assetEntriesFacet.setStatic(true);
searchContext.addFacet(assetEntriesFacet);
Facet scopeFacet = new ScopeFacet(searchContext);
scopeFacet.setStatic(true);
searchContext.addFacet(scopeFacet);
searchContext.setAndSearch(andSearch);
searchContext.setAssetCategoryIds(StringUtil.split(assetCategoryIds, 0L));
searchContext.setAssetTagNames(StringUtil.split(assetTagNames));
searchContext.setAttribute(Field.DESCRIPTION, description);
searchContext.setAttribute(Field.TITLE, title);
searchContext.setAttribute(Field.USER_NAME, userName);
searchContext.setAttribute("paginationType", "regular");
searchContext.setAttribute("status", status);
searchContext.setCompanyId(companyId);
searchContext.setEnd(end);
searchContext.setEntryClassNames(getClassNames(className));
searchContext.setGroupIds(groupIds);
QueryConfig queryConfig = new QueryConfig();
queryConfig.setHighlightEnabled(false);
queryConfig.setScoreEnabled(false);
searchContext.setQueryConfig(queryConfig);
searchContext.setStart(start);
searchContext.setUserId(userId);
Indexer indexer = FacetedSearcher.getInstance();
return indexer.search(searchContext);
} catch (Exception e) {
throw new SystemException(e);
}
}