src/dochelper/URLRecord.java

Go to the documentation of this file.

/*
 * TableRecord.java
 *
 * Created on March 16, 2009, 08:32 AM
 *
 * Copyright 2009 Online Nederland Breedband B.V.
 * See the COPYRIGHT file for redistribution and use restrictions.
 * Authors: Guillermo Christen <guillermo.christen@is.online.nl>
 */
package dochelper;

import java.net.URI;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.HashMap;

import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.util.EntityUtils;

import dochelper.exceptions.InvalidValueException;
public class URLRecord implements Record {
        

    protected URI url;

    protected String value;    
    
    protected String name;
    
    protected HashMap<String, String> _params;    
    
    static final HttpClient httpClient = new DefaultHttpClient();
           
    public URLRecord(String url, String name) throws InvalidValueException {
        try {
            this.url = new URI(url);
            this.name = name;        
        }
        catch(URISyntaxException use) {
            throw new InvalidValueException();
        }
    }
    
    public ResultNode execute(ArrayList<DocPattern> patterns) {

        String value = this.getValue();

        ResultNode root = new ResultNode(null, _params, name);

        //match all the patterns
        if (patterns != null) {
            for (DocPattern pattern : patterns) {
                root.addChild(pattern.execute(value));
            }
        }

        return root;
        
    }
    
    public URI getURL() {
      return url;
    }
    
    public String getValue() {
        
        try {
            if(value==null || value.length()<2) {
                //establish the HTTP connection
                HttpGet request = new HttpGet(this.url);        

                //fetch the value
                HttpResponse response = httpClient.execute(request);        

                //return as a string and kill connection
                if(response.getStatusLine().getStatusCode()== 200) {
                    value = EntityUtils.toString(response.getEntity());
                }
                else {
                    //TODO: Log error or Throw
                    value = "";
                }
                
                response.getEntity().consumeContent();
            }
            
        }
        catch(Exception e) {
            //TODO: handle properly later
            e.printStackTrace();
        }
        
        return value;
    }

    public boolean equals(Object record) {
      
      if(record instanceof URLRecord) 
        return this.getURL().equals( ((URLRecord)record).getURL());
      
      return false;
      
    }
    
    
}

---