1
package org.apache.lucene.analysis.payloads;
3
* Licensed to the Apache Software Foundation (ASF) under one or more
4
* contributor license agreements. See the NOTICE file distributed with
5
* this work for additional information regarding copyright ownership.
6
* The ASF licenses this file to You under the Apache License, Version 2.0
7
* (the "License"); you may not use this file except in compliance with
8
* the License. You may obtain a copy of the License at
10
* http://www.apache.org/licenses/LICENSE-2.0
12
* Unless required by applicable law or agreed to in writing, software
13
* distributed under the License is distributed on an "AS IS" BASIS,
14
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15
* See the License for the specific language governing permissions and
16
* limitations under the License.
20
import org.apache.lucene.analysis.TokenFilter;
21
import org.apache.lucene.analysis.TokenStream;
22
import org.apache.lucene.analysis.tokenattributes.PayloadAttribute;
23
import org.apache.lucene.analysis.tokenattributes.TypeAttribute;
24
import org.apache.lucene.index.Payload;
26
import java.io.IOException;
30
* Assigns a payload to a token based on the {@link org.apache.lucene.analysis.Token#type()}
33
public class NumericPayloadTokenFilter extends TokenFilter {
35
private String typeMatch;
36
private Payload thePayload;
38
private final PayloadAttribute payloadAtt = addAttribute(PayloadAttribute.class);
39
private final TypeAttribute typeAtt = addAttribute(TypeAttribute.class);
41
public NumericPayloadTokenFilter(TokenStream input, float payload, String typeMatch) {
43
//Need to encode the payload
44
thePayload = new Payload(PayloadHelper.encodeFloat(payload));
45
this.typeMatch = typeMatch;
49
public final boolean incrementToken() throws IOException {
50
if (input.incrementToken()) {
51
if (typeAtt.type().equals(typeMatch))
52
payloadAtt.setPayload(thePayload);