/*******************************************************************************
* Copyright (c) 2000, 2009 IBM Corporation and others.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* IBM Corporation - initial API and implementation
* James D Miles (IBM Corp.) - bug 181375, ArrayIndexOutOfBoundsException in SiteSearchCategory$Query
* James D Miles (IBM Corp.) - bug 191783, NullPointerException in FeatureDownloader
*******************************************************************************/
package org.eclipse.update.internal.search;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.update.core.ICategory;
import org.eclipse.update.core.IFeature;
import org.eclipse.update.core.ISite;
import org.eclipse.update.core.ISiteFeatureReference;
import org.eclipse.update.internal.core.ExtendedSite;
import org.eclipse.update.internal.core.LiteFeature;
import org.eclipse.update.internal.core.UpdateCore;
import org.eclipse.update.search.IQueryUpdateSiteAdapter;
import org.eclipse.update.search.IUpdateSearchFilter;
import org.eclipse.update.search.IUpdateSearchQuery;
import org.eclipse.update.search.IUpdateSearchResultCollector;
/**
* Searches an update site
*/
public class SiteSearchCategory extends BaseSearchCategory {
private IUpdateSearchQuery[] queries;
private boolean liteFeaturesAreOK;
private static final String CATEGORY_ID =
"org.eclipse.update.core.unified-search"; //$NON-NLS-1$
private static class Query implements IUpdateSearchQuery {
private boolean liteFeaturesAreOK;
public Query() {
liteFeaturesAreOK = false;
}
public Query(boolean liteFeaturesAreOK) {
this.liteFeaturesAreOK = liteFeaturesAreOK;
}
public void setLiteFeaturesAreOK(boolean liteFeaturesAreOK) {
this.liteFeaturesAreOK = liteFeaturesAreOK;
}
public void run(
ISite site,
String[] categoriesToSkip,
IUpdateSearchFilter filter,
IUpdateSearchResultCollector collector,
IProgressMonitor monitor) {
ISiteFeatureReference[] refs = site.getFeatureReferences();
HashSet ignores = new HashSet();
Map liteFeatures = new HashMap();
if (categoriesToSkip != null) {
for (int i = 0; i < categoriesToSkip.length; i++) {
ignores.add(categoriesToSkip[i]);
}
}
List siteFeatureReferences = new ArrayList(Arrays.asList(refs));
if (liteFeaturesAreOK && (site instanceof ExtendedSite) ) {
ExtendedSite extendedSite = (ExtendedSite)site;
LiteFeature[] liteFeaturesArray = extendedSite.getLiteFeatures();
if ( (liteFeaturesArray != null) && ( liteFeaturesArray.length != 0)) {
for(int i = 0; i < liteFeaturesArray.length; i++) {
liteFeatures.put(liteFeaturesArray[i].getVersionedIdentifier(), liteFeaturesArray[i]);
}
(new FeatureDownloader(siteFeatureReferences, collector, filter, ignores, monitor, true, liteFeatures)).run();
return;
} else {
liteFeaturesAreOK = false;
}
}
monitor.beginTask("", refs.length); //$NON-NLS-1$
ThreadGroup featureDownloaders = new ThreadGroup("FeatureDownloader"); //$NON-NLS-1$
int numberOfThreads = (refs.length > 5)? 5: refs.length;
Thread[] featureDownloader = new Thread[numberOfThreads];
for( int i = 0; i < numberOfThreads; i++) {
featureDownloader[i] = new Thread(featureDownloaders, new FeatureDownloader(siteFeatureReferences, collector, filter, ignores, monitor));
featureDownloader[i].start();
}
int i =0;
while(i < numberOfThreads){
if (monitor.isCanceled()) {
synchronized(siteFeatureReferences) {
siteFeatureReferences.clear();
}
}
try {
featureDownloader[i].join(250);
if(!featureDownloader[i].isAlive()){
i++;
}
} catch (InterruptedException ie) {
}
}
}
/* (non-Javadoc)
* @see org.eclipse.update.internal.ui.search.ISearchQuery#getSearchSite()
*/
public IQueryUpdateSiteAdapter getQuerySearchSite() {
return null;
}
}
public SiteSearchCategory() {
super(CATEGORY_ID);
queries = new IUpdateSearchQuery[] { new Query()};
}
public SiteSearchCategory(boolean liteFeaturesAreOK) {
this();
this.liteFeaturesAreOK = liteFeaturesAreOK;
queries = new IUpdateSearchQuery[] { new Query(liteFeaturesAreOK)};
}
public IUpdateSearchQuery[] getQueries() {
return queries;
}
private static class FeatureDownloader implements Runnable {
private List siteFeatureReferences;
private IProgressMonitor monitor;
private IUpdateSearchFilter filter;
private IUpdateSearchResultCollector collector;
private HashSet ignores;
private boolean liteFeaturesAreOK;
private Map liteFeatures;
public FeatureDownloader(List siteFeatureReferences, IUpdateSearchResultCollector collector, IUpdateSearchFilter filter, HashSet ignores, IProgressMonitor monitor) {
super();
this.collector = collector;
this.filter = filter;
this.ignores = ignores;
this.monitor = monitor;
this.siteFeatureReferences = siteFeatureReferences;
}
public FeatureDownloader(List siteFeatureReferences, IUpdateSearchResultCollector collector, IUpdateSearchFilter filter, HashSet ignores, IProgressMonitor monitor, boolean liteFeaturesAreOK, Map liteFeatures) {
this(siteFeatureReferences, collector, filter, ignores, monitor);
this.liteFeaturesAreOK = liteFeaturesAreOK && (liteFeatures != null);
this.liteFeatures = liteFeatures;
}
public void run() {
ISiteFeatureReference siteFeatureReference = null;
while (siteFeatureReferences.size() != 0) {
synchronized(siteFeatureReferences) {
try{
siteFeatureReference = (ISiteFeatureReference)siteFeatureReferences.remove(0);
}catch(IndexOutOfBoundsException e){
siteFeatureReference = null;
break;
}
}
if (siteFeatureReference != null) {
boolean skipFeature = false;
if (monitor.isCanceled())
break;
if (ignores.size() > 0) {
ICategory[] categories = siteFeatureReference.getCategories();
for (int j = 0; j < categories.length; j++) {
ICategory category = categories[j];
if (ignores.contains(category.getName())) {
skipFeature = true;
break;
}
}
}
try {
if (!skipFeature) {
if (filter.accept(siteFeatureReference)) {
IFeature feature = null;
if(liteFeaturesAreOK) {
feature = (IFeature)liteFeatures.get(siteFeatureReference.getVersionedIdentifier());
}
if (feature == null){
feature = siteFeatureReference.getFeature(null);
}
synchronized(siteFeatureReferences) {
if ( (feature != null) && (filter.accept(siteFeatureReference)) ) {
collector.accept(feature);
monitor.subTask(feature.getLabel());
}
}
}
}
} catch (CoreException e) {
UpdateCore.log(e);
} finally {
monitor.worked(1);
}
}
}
}
}
public boolean isLiteFeaturesAreOK() {
return liteFeaturesAreOK;
}
public void setLiteFeaturesAreOK(boolean liteFeaturesAreOK) {
this.liteFeaturesAreOK = liteFeaturesAreOK;
for( int i = 0; i < queries.length; i++) {
((Query)queries[i]).setLiteFeaturesAreOK(liteFeaturesAreOK);
}
}
}