Lucene4.0过滤器的实现（含3.6版本）

一：Lucene3.6过滤器的实现

以空间搜索为例，以下是代码：

 public static final class DistanceFilter extends Filter    {                private static final long serialVersionUID = 1L;        private float Radius;        private double x;        private double y;        public DistanceFilter(String location, float radius)        {            this.Radius = radius;            String[] parts = location.split(",");            this.x = Double.valueOf(parts[0]);            this.y = Double.valueOf(parts[1]);        }        private static double rad(double d)        {            return d * Math.PI / 180.0;        }        public DocIdSet getDocIdSet(IndexReader reader) throws IOException        {            OpenBitSet result = new OpenBitSet(reader.maxDoc());            TermDocs td = reader.termDocs(new Term("type", "restaurant"));            double destination_x;            double destination_y;            while (td.next())            {                Document doc = reader.document(td.doc());                String field = doc.get("location");                String[] loca = field.split(",");                destination_x = Double.valueOf(loca[0]);                destination_y = Double.valueOf(loca[1]);                if (getDistance(x, y, destination_x, destination_y, Radius))                {                    result.set((long) td.doc());                }            }            return result;        } public boolean getDistance(double x, double y, double dx, double dy, float radius)        {            double radLat1 = rad(x);            double radLat2 = rad(y);            double wadLat1 = rad(dx);            double wadLat2 = rad(dy);            double a = radLat1 - wadLat1;            double b = radLat2 - wadLat2;            double s = 2 * Math.asin(Math.sqrt(Math.pow(Math.sin(a / 2), 2) + Math.cos(radLat1) * Math.cos(wadLat1)                    * Math.pow(Math.sin(b / 2), 2)));            s = s * EARTH_RADIUS;            if (s > radius)                return false;            return true;        }

 

在Lucene3.6中TermDoc这个类非常有用，可以直接从IndexReader中读取有关term的所有信息，过滤就是从TermDoc中展开。

方法getDocIdSet(IndexReader reader)是从filter中继承的必须实现的方法，返回值是DocIdSet类型的，就是说只要条件符合，result的set方法就将这篇文档标记为需要返回的类型，未被标记的就不返回，从而实现过滤功能。

二：Luene4.0过滤器的实现

在Lucene4.0中，TermDoc这个类被删除了，但是提供了其他的类来代替，具体来讲是用TermsEnum/FieldsEnum/DocsEnum这几个类来代替TermDoc/TermEnum/TermPosition，所以，为了实现过滤要稍微麻烦一点，以下是代码：

 public DocIdSet getDocIdSet(AtomicReaderContext context, Bits livedocs) throws IOException        {            OpenBitSet result = new OpenBitSet(context.reader().maxDoc());            double destination_x;            double destination_y;            int document;            BytesRef term=new BytesRef("restaurant");            DocsEnum docEnum = context.reader().termDocsEnum(livedocs,"type",term,false);            while((document=docEnum.nextDoc())!=DocsEnum.NO_MORE_DOCS){                Document doc = context.reader().document(document);                String xfield = doc.get("X");                String yfield=doc.get("Y");                destination_x = Double.valueOf(xfield);                destination_y = Double.valueOf(yfield);                if (getDistance(x, y, destination_x, destination_y, Radius))                {                    result.set(docEnum.docID());                }            }            return result;        }

该处省去了跟Lucene3.6中相同的代码，看起来4.0中实现过滤要比3.6中复杂一些。

首先是IndexReader被替换成了AtomicReaderContext，TermDoc被替换成了DocsEnum，具体实现详见代码。