io.zulia.server.analysis.analyzer.BooleanAnalyzer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zulia-server Show documentation

$project.description

There is a newer version: 1.6.4

package io.zulia.server.analysis.analyzer;

import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.Tokenizer;
import org.apache.lucene.analysis.tokenattributes.CharTermAttribute;

import java.io.IOException;

/**
 * Created by Matt Davis on 2/10/16.
 * Based on org.apache.solr.schema.BoolField from Solr
 */
public class BooleanAnalyzer extends Analyzer {

	protected final static char[] TRUE_TOKEN = { 'T' };
	protected final static char[] FALSE_TOKEN = { 'F' };

	@Override
	protected TokenStreamComponents createComponents(String fieldName) {
		Tokenizer tokenizer = new Tokenizer() {
			final CharTermAttribute termAtt = addAttribute(CharTermAttribute.class);
			boolean done = false;

			@Override
			public void reset() throws IOException {
				super.reset();
				done = false;
			}

			@Override
			public boolean incrementToken() throws IOException {
				clearAttributes();
				if (done)
					return false;
				done = true;
				int ch = input.read();
				if (ch == -1)
					return false;
				termAtt.copyBuffer(((ch == 't' || ch == 'T' || ch == 'y' || ch == 'Y' || ch == '1') ? TRUE_TOKEN : FALSE_TOKEN), 0, 1);
				return true;
			}
		};

		return new TokenStreamComponents(tokenizer);
	}
}