SolrQuery.php

<?
        
// Change this to your server

define('SOLR_META_QUERY', '127.0.0.1:8080');
        
// VERSION SolrQuery  0.1
// Written by Brian Lucas, use any which way you see fit.
//
// DESCRIPTION
// 
//   This class runs an external fulltext search on Solr
//      Example Usage:

/*
                $query = new SolrQuery;
        
                $query->language_id = $this->language_id;
                $query->lowerdate = $day_3neg;
                $query->upperdate = $day_3pos;
                $query->limit = 10;
                $query->group_id = 1;

                $results = $query->runQuery($titlekeywords);
                print_r($results);
*/              

class SolrQuery {
	
	// All of these init variables are yours to define, these are just examples
	var $language_id = 0;
	var $lowerdate = 0; // lower bound of date range
	var $upperdate = 0; // upper bound of date range
	var $limit = 100; // number of result to return
	var $min_score = 0.5; // only add these values if they are above this score
	var $httppost=false; // set to true to send via HTTP POST (form) instead of HTTP GET (url)
	var $debug=false; // turn on debugging mode (not fully complete)
	
	// Reset or initialize variables
	function init() {
		$this->language_id = 0;
		$this->lowerdate = 0;
		$this->upperdate = 0;
		$this->limit = 100;
		$this->min_score = 0.0; // only add these values if they are above this score
		$this->httppost=false;
	        
	} 
	// Example of a query -- different queries with their own logic should be made as separate functions
	function runQuery($titlekeywords) {
		$query = $this->setUpQuery();
		
		// remove spaces
		// change to mb_eregi_replace if using multibyte
		$keywords = preg_replace('@[\s]+@',' ', trim($titlekeywords));
		$string = "+(title:($titlekeywords)) ";
		$query.=$string;
		
		$query = trim($query);
		
		$data = $this->fetchResults($query);
		$results = $this->handleResponse($data);
		
		return $results;
	} // end function sortQuery
	
	// Use this to set up default logic that will apply to all queries
	function setUpQuery() {
		$query = "";
		
		// Set up your basic fields here that will be used in the query
		if ($this->language_id >0) $query.="+language_id:$this->language_id ";
		if ($this->group_id >0) $query.="+group_id:$this->group_id ";
		
		$lowerdate = "*";
		$upperdate = "*";
		                
		if ($this->lowerdate >0) $lowerdate = $this->lowerdate;
		if ($this->upperdate >0) $upperdate = $this->upperdate;
		
		if (($lowerdate!="*") || ($upperdate != "*")) {
			$query.="+lucene_date:[".$lowerdate." TO ".$upperdate."] ";
		}
		return $query;
	}
	
	// Handle the response from the server and transform it into useful fields
	function handleResponse($data) {
		if ($data) {
			$xml = simplexml_load_string($data );
			
			
			$results = array();
			foreach ($xml->result->doc as $story) {
				$xmlarray = array();
				try{
				        foreach ($story as $item) {
				                $name = $item->attributes()->name;
				                $value = $item;
				                $xmlarray["$name"] = "$value";
				
				        } // end foreach
				} catch (Exception $e) {
				        echo 'Problem handling XML array.'; 
				}
				
				// 
				if ($this->debug) echo "checking if ".$xmlarray['score']." > ".$this->min_score;
				if ($xmlarray['score'] > $this->min_score) $results[] = $xmlarray;
			
			} // end foreach
		
		} else {
		        $results=false;
		} // end if / else data
		
		 
		return $results;
	}
	
	// Send the query to the server
	function fetchResults($query) {
		
		if ($this->debug) echo "parsing string $query";
		
		$url = "http://".SOLR_META_QUERY."/solr/select";
		
		$querystring = "stylesheet=&q=".trim(urlencode($query))."&version=2.1&start=0&rows=".$this->limit."&fl=*+score&qt=standard";
		
		if (!$this->httppost) $selecturl = "/?$querystring";
		$url .= $selecturl;
		
		$header[] = "Accept: text/xml,application/xml,application/xhtml+xml,text/html;q=0.9,text/plain;q=0.8,image/png,*/*;q=0.5";
		$header[] = "Accept-Language: en-us,en;q=0.5";
		$header[] = "Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.7";
		
		$ch = curl_init();
		curl_setopt($ch, CURLOPT_URL, $url); // set url to post to
		curl_setopt($ch, CURLOPT_RETURNTRANSFER, 1);
		curl_setopt($ch, CURLOPT_TIMEOUT, 10);
		curl_setopt($ch, CURLOPT_FOLLOWLOCATION, 1);
		curl_setopt($ch, CURLOPT_ENCODING,"");          
		curl_setopt($ch, CURLOPT_HTTPHEADER, $header);
		curl_setopt($ch, CURLOPT_CONNECTTIMEOUT,10);
		curl_setopt($ch, CURLOPT_DNS_USE_GLOBAL_CACHE, 0);
		
		if ($this->httppost) {
			curl_setopt($ch, CURLOPT_POST, 1 );
			curl_setopt($ch, CURLOPT_POSTFIELDS,$querystring);
		}
		
		if ($this->debug) echo "\r\nRetrieving <A HREF='$url' target=_BLANK>$url</a>...\r\n";
		
		$data = curl_exec($ch);
		
		if (curl_errno($ch)) {
			logger "setting results to false, error";
			print curl_error($ch);
			$results=false;
		} else {
			curl_close($ch);
			if ( strstr ( $data, '<status>0</status>')) {
				$results = $data;
			}
		} 
		if ($this->debug) echo $data;
		return $results;
	}
} // end class

?>