org.elasticsearch.index.fielddata.ScriptDocValues Java Examples

@Override
public void hitExecute(SearchContext context, HitContext hitContext) {
    for (FieldDataFieldsContext.FieldDataField field : context.getFetchSubPhaseContext(CONTEXT_FACTORY).fields()) {
        if (hitContext.hit().fieldsOrNull() == null) {
            hitContext.hit().fields(new HashMap<String, SearchHitField>(2));
        }
        SearchHitField hitField = hitContext.hit().fields().get(field.name());
        if (hitField == null) {
            hitField = new InternalSearchHitField(field.name(), new ArrayList<>(2));
            hitContext.hit().fields().put(field.name(), hitField);
        }
        MappedFieldType fieldType = context.mapperService().smartNameFieldType(field.name());
        if (fieldType != null) {
            AtomicFieldData data = context.fieldData().getForField(fieldType).load(hitContext.readerContext());
            ScriptDocValues values = data.getScriptValues();
            values.setNextDocId(hitContext.docId());
            hitField.values().addAll(values.getValues());
        }
    }
}
 

@Override
public Object get(Object key) {
    // assume its a string...
    String fieldName = key.toString();
    ScriptDocValues scriptValues = localCacheFieldData.get(fieldName);
    if (scriptValues == null) {
        final MappedFieldType fieldType = mapperService.smartNameFieldType(fieldName, types);
        if (fieldType == null) {
            throw new IllegalArgumentException("No field found for [" + fieldName + "] in mapping with types " + Arrays.toString(types) + "");
        }
        // load fielddata on behalf of the script: otherwise it would need additional permissions
        // to deal with pagedbytes/ramusagestimator/etc
        scriptValues = AccessController.doPrivileged(new PrivilegedAction<ScriptDocValues>() {
            @Override
            public ScriptDocValues run() {
                return fieldDataService.getForField(fieldType).load(reader).getScriptValues();
            }
        });
        localCacheFieldData.put(fieldName, scriptValues);
    }
    scriptValues.setNextDocId(docId);
    return scriptValues;
}
 

@Override
@SuppressWarnings("unchecked")
public Object run() {
    // First we get field using doc lookup
    ScriptDocValues<Long> docValue = (ScriptDocValues<Long>) doc().get(fieldName);
    // Check if field exists
    if (docValue != null && !docValue.isEmpty()) {
        try {
            // Try to parse it as an integer
            BigInteger bigInteger = BigInteger.valueOf(((Longs) docValue).getValue());
            // Check if it's prime
            return bigInteger.isProbablePrime(certainty);
        } catch (NumberFormatException ex) {
            return false;
        }
    }
    return false;
}
 

@Override
public Object run() {
    if (this.field == null || this.inputFeatureVector == null || this.inputFeatureVectorNorm == 0) {
        return this.baseConstant;
    }

    if (!doc().containsKey(this.field) || doc().get(this.field) == null) {
        return this.baseConstant;
    }

    String docFeatureVectorStr = ((ScriptDocValues.Strings) doc().get(this.field)).getValue();
    return calculateScore(docFeatureVectorStr);
}
 

@Override
public boolean containsKey(Object key) {
    // assume its a string...
    String fieldName = key.toString();
    ScriptDocValues scriptValues = localCacheFieldData.get(fieldName);
    if (scriptValues == null) {
        MappedFieldType fieldType = mapperService.smartNameFieldType(fieldName, types);
        if (fieldType == null) {
            return false;
        }
    }
    return true;
}
 

@Override
@SuppressWarnings("unchecked")
public Object run() {
    ArrayList<Long> transactions = (ArrayList<Long>) agg.get(InitScriptFactory.TRANSACTIONS_FIELD);
    ScriptDocValues.Longs amount = (ScriptDocValues.Longs) doc().get("amount");
    ScriptDocValues.Strings type = (ScriptDocValues.Strings) doc().get("type");
    if ("sale".equals(type.getValue())) {
        transactions.add(amount.getValue());
    } else {
        transactions.add(-amount.getValue());
    }
    return null;
}
 

/**
 * Returns field data strings access for the provided field.
 */
protected ScriptDocValues.Strings docFieldStrings(String field) {
    return (ScriptDocValues.Strings) doc().get(field);
}
 

/**
 * Returns field data double (floating point) access for the provided field.
 */
protected ScriptDocValues.Doubles docFieldDoubles(String field) {
    return (ScriptDocValues.Doubles) doc().get(field);
}
 

/**
 * Returns field data long (integers) access for the provided field.
 */
protected ScriptDocValues.Longs docFieldLongs(String field) {
    return (ScriptDocValues.Longs) doc().get(field);
}
 

@Override
public final ScriptDocValues getScriptValues() {
    return new ScriptDocValues.Doubles(getDoubleValues());
}
 

@Override
public final ScriptDocValues getScriptValues() {
    return new ScriptDocValues.Longs(getLongValues());
}
 

@Override
public final ScriptDocValues getScriptValues() {
    return new ScriptDocValues.Strings(getBytesValues());
}
 

@Override
public final ScriptDocValues.GeoPoints getScriptValues() {
    return new ScriptDocValues.GeoPoints(getGeoPointValues());
}
 

@Override
public final ScriptDocValues getScriptValues() {
    return new ScriptDocValues.Strings(getBytesValues());
}
 

@Override
public Strings getScriptValues() {
    return new ScriptDocValues.Strings(getBytesValues());
}
 

@Override
public ScriptDocValues getScriptValues() {
    throw new UnsupportedOperationException();
}