package com.jobcn.test;
 
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
 
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.SortField;
import org.apache.lucene.store.FSDirectory;
 
import com.browseengine.bobo.api.BoboBrowser;
import com.browseengine.bobo.api.BoboIndexReader;
import com.browseengine.bobo.api.Browsable;
import com.browseengine.bobo.api.BrowseFacet;
import com.browseengine.bobo.api.BrowseHit;
import com.browseengine.bobo.api.BrowseRequest;
import com.browseengine.bobo.api.BrowseResult;
import com.browseengine.bobo.api.FacetAccessible;
import com.browseengine.bobo.api.FacetSpec;
import com.browseengine.bobo.api.FacetSpec.FacetSortSpec;
import com.browseengine.bobo.facets.FacetHandler;
import com.browseengine.bobo.facets.impl.SimpleFacetHandler;
 
public class TestSearchPosByLucene {
 
   private static FSDirectory fsd;
 
	static {
		String indexDir = "/use/lucene/target/index/compos_0";
	    try {
			if(fsd == null){
	    	   fsd = FSDirectory.getDirectory(indexDir);
			}
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
 
	public void test(String keyWord) throws Exception{
 
        // reqworkyear facet handler
        SimpleFacetHandler reqworkyearHandler = new SimpleFacetHandler("reqworkyear");
 
        // reqdegreeid facet handler
        SimpleFacetHandler reqdegreeidHandler = new SimpleFacetHandler("reqdegreeid");
 
 
        List<FacetHandler> handlerList = Arrays.asList(new FacetHandler[]{reqworkyearHandler, reqdegreeidHandler});
 
        IndexReader reader = IndexReader.open(fsd,true);
 
          // decorate it with a bobo index reader
          BoboIndexReader boboReader = BoboIndexReader.getInstance(reader,handlerList);
 
          // creating a browse request
          BrowseRequest br=new BrowseRequest();
          br.setCount(5);
          br.setOffset(0);
 
          br.setFetchStoredFields(true);
 
          // parse a query
          QueryParser parser = new QueryParser("alltext",new StandardAnalyzer());
          Query q=parser.parse(keyWord);
          br.setQuery(q);
 
          SortField[] sortFields = new SortField[2]; //按学历+发布日期排序
          sortFields[0] = new SortField("reqdegreeid", SortField.INT, true);
          sortFields[1] = new SortField("postdate", SortField.STRING, true);
          br.setSort(sortFields);
 
 
          // add the facet output specs
          FacetSpec reqworkyearSpec = new FacetSpec();
          reqworkyearSpec.setOrderBy(FacetSortSpec.OrderHitsDesc);
 
          FacetSpec reqdegreeidSpec = new FacetSpec();
          reqdegreeidSpec.setMinHitCount(2);
          reqdegreeidSpec.setOrderBy(FacetSortSpec.OrderHitsDesc);
 
 
          br.setFacetSpec("reqworkyear",reqworkyearSpec);
          br.setFacetSpec("reqdegreeid",reqdegreeidSpec);
 
          long startTime = System.currentTimeMillis();
 
          // perform browse
          Browsable browser=new BoboBrowser(boboReader);
 
          BrowseResult result=browser.browse(br);
 
          int totalHits = result.getNumHits();
          System.out.println("公司发布的职位信息中含有‘"+keyWord+"’关键字的职位总数:" + totalHits);
           System.out.println("打印前"+ br.getCount() +"个职位Id：");
          BrowseHit[] hits = result.getHits();
          for(BrowseHit hit : hits){
            	System.out.println(hit.getStoredFields().get("$/compos/Id"));
           }
 
          Map<String,FacetAccessible> facetMap = result.getFacetMap();
 
          FacetAccessible colorFacets = facetMap.get("reqdegreeid");
          List<BrowseFacet> facetVals = colorFacets.getFacets();
          System.out.println("学历统计:");
          for(BrowseFacet facetVal : facetVals){
        	    System.out.println(facetVal.getValue() +"--->"+ facetVal.getHitCount() );  
            }
 
          FacetAccessible colorFacets2 = facetMap.get("reqworkyear");
          List<BrowseFacet> facetVals2 = colorFacets2.getFacets();
          System.out.println("工作经验统计:");
          for(BrowseFacet facetVal : facetVals2){
        	    System.out.println(facetVal.getValue() +"--->"+ facetVal.getHitCount() );  
            }
 
          long endTime = System.currentTimeMillis();
          System.out.println("总共用时："+(endTime - startTime) + " ms"); 
	}
 
	public static void main(String args[]) throws Exception{
 
		TestSearchPosByLucene t = new TestSearchPosByLucene();
 
		t.test("软件工程师");
		t.test("销售");
		t.test("文员");
		t.test("程序员");
		t.test("机械");
		t.test("品质");
		t.test("采购");
		t.test("网络");
 
	}
 
}