cz/vutbr/fit/dudka/SGVis/Visual/LookupWorker.java

Go to the documentation of this file.

package cz.vutbr.fit.dudka.SGVis.Visual;

import java.net.URL;
import java.util.HashSet;

import javax.swing.SwingWorker;

import cz.vutbr.fit.dudka.SGVis.Config;
import cz.vutbr.fit.dudka.SGVis.Data.RelationStorage;
import cz.vutbr.fit.dudka.SGVis.Lookup.Lookup;
import cz.vutbr.fit.dudka.SGVis.Lookup.Request;
import cz.vutbr.fit.dudka.SGVis.Lookup.Response;

public class LookupWorker extends SwingWorker<Void, Void> {
  private URL url;
  private RelationStorage storage;
  private boolean downloadOk;
  private GraphView caller;

  private static HashSet<LookupWorker> set = new HashSet<LookupWorker>();
  private static int cntTotal = 0;
  private static int cntSuccess = 0;
  private static int cntFailed = 0;
  
  public static synchronized void run(URL url, GraphView caller) {
    LookupWorker worker = new LookupWorker(url, caller);
    set.add(worker);
    cntTotal++;
    worker.execute();
  }
  
  public static synchronized void killAll() {
    HashSet<LookupWorker> setClone = new HashSet<LookupWorker>(set);
    for (LookupWorker worker: setClone)
      worker.cancel(true);
    set.clear();
  }
  
  public static synchronized int getTotalCount() {
    return cntTotal;
  }
  
  public static synchronized int getActiveCount() {
    return set.size();
  }
  
  public static synchronized int getSuccessCount() {
    return cntSuccess; 
  }
  
  public static synchronized int getFailedCount() {
    return cntFailed;
  }

  private static synchronized void removeFromSet(LookupWorker worker) {
    set.remove(worker);
  }

  private LookupWorker(URL url, GraphView caller) {
    super();
    this.url = url;
    this.storage = new RelationStorage();
    this.caller = caller;
    this.downloadOk = false;
  }

  @Override
  protected Void doInBackground() throws Exception {
    try {
      Lookup lookup = Config.createLookup();
      Request rq = Config.createRequest(url.toString());
      String responseText = lookup.lookup(rq);
      Response response = new Response(responseText);
      response.addTo(this.storage);
      this.downloadOk = true;
    }
    catch (Exception e) {
      this.downloadOk = false;
    }
    return null;
  }

  @Override
  protected void done() {
    if (downloadOk) {
      caller.handleResponse(this.storage, url);
      cntSuccess++;
    } else {
      caller.lookupError(url);
      cntFailed++;
    }
    removeFromSet(this);
  }
}

---