Examples with NumericDocValuesUpdate org.apache.lucene.index.DocValuesUpdate.NumericDocValuesUpdate used on opensource projects

Example 1 with NumericDocValuesUpdate

use of org.apache.lucene.index.DocValuesUpdate.NumericDocValuesUpdate in project lucene-solr by apache.

the class IndexWriter method updateDocValues.

/**
   * Updates documents' DocValues fields to the given values. Each field update
   * is applied to the set of documents that are associated with the
   * {@link Term} to the same value. All updates are atomically applied and
   * flushed together.
   * 
   * @param updates
   *          the updates to apply
   *
   * @return The <a href="#sequence_number">sequence number</a>
   * for this operation
   *
   * @throws CorruptIndexException
   *           if the index is corrupt
   * @throws IOException
   *           if there is a low-level IO error
   */
public long updateDocValues(Term term, Field... updates) throws IOException {
    ensureOpen();
    DocValuesUpdate[] dvUpdates = new DocValuesUpdate[updates.length];
    for (int i = 0; i < updates.length; i++) {
        final Field f = updates[i];
        final DocValuesType dvType = f.fieldType().docValuesType();
        if (dvType == null) {
            throw new NullPointerException("DocValuesType must not be null (field: \"" + f.name() + "\")");
        }
        if (dvType == DocValuesType.NONE) {
            throw new IllegalArgumentException("can only update NUMERIC or BINARY fields! field=" + f.name());
        }
        if (!globalFieldNumberMap.contains(f.name(), dvType)) {
            throw new IllegalArgumentException("can only update existing docvalues fields! field=" + f.name() + ", type=" + dvType);
        }
        if (config.getIndexSortFields().contains(f.name())) {
            throw new IllegalArgumentException("cannot update docvalues field involved in the index sort, field=" + f.name() + ", sort=" + config.getIndexSort());
        }
        switch(dvType) {
            case NUMERIC:
                dvUpdates[i] = new NumericDocValuesUpdate(term, f.name(), (Long) f.numericValue());
                break;
            case BINARY:
                dvUpdates[i] = new BinaryDocValuesUpdate(term, f.name(), f.binaryValue());
                break;
            default:
                throw new IllegalArgumentException("can only update NUMERIC or BINARY fields: field=" + f.name() + ", type=" + dvType);
        }
    }
    try {
        long seqNo = docWriter.updateDocValues(dvUpdates);
        if (seqNo < 0) {
            seqNo = -seqNo;
            processEvents(true, false);
        }
        return seqNo;
    } catch (VirtualMachineError tragedy) {
        tragicEvent(tragedy, "updateDocValues");
        // dead code but javac disagrees:
        return -1;
    }
}

Example 2 with NumericDocValuesUpdate

use of org.apache.lucene.index.DocValuesUpdate.NumericDocValuesUpdate in project lucene-solr by apache.

the class IndexWriter method updateNumericDocValue.

/**
   * Updates a document's {@link NumericDocValues} for <code>field</code> to the
   * given <code>value</code>. You can only update fields that already exist in
   * the index, not add new fields through this method.
   * 
   * @param term
   *          the term to identify the document(s) to be updated
   * @param field
   *          field name of the {@link NumericDocValues} field
   * @param value
   *          new value for the field
   *
   * @return The <a href="#sequence_number">sequence number</a>
   * for this operation
   *
   * @throws CorruptIndexException
   *           if the index is corrupt
   * @throws IOException
   *           if there is a low-level IO error
   */
public long updateNumericDocValue(Term term, String field, long value) throws IOException {
    ensureOpen();
    if (!globalFieldNumberMap.contains(field, DocValuesType.NUMERIC)) {
        throw new IllegalArgumentException("can only update existing numeric-docvalues fields!");
    }
    if (config.getIndexSortFields().contains(field)) {
        throw new IllegalArgumentException("cannot update docvalues field involved in the index sort, field=" + field + ", sort=" + config.getIndexSort());
    }
    try {
        long seqNo = docWriter.updateDocValues(new NumericDocValuesUpdate(term, field, value));
        if (seqNo < 0) {
            seqNo = -seqNo;
            processEvents(true, false);
        }
        return seqNo;
    } catch (VirtualMachineError tragedy) {
        tragicEvent(tragedy, "updateNumericDocValue");
        // dead code but javac disagrees:
        return -1;
    }
}

Example 3 with NumericDocValuesUpdate

use of org.apache.lucene.index.DocValuesUpdate.NumericDocValuesUpdate in project lucene-solr by apache.

the class CoalescedUpdates method update.

void update(FrozenBufferedUpdates in) {
    totalTermCount += in.terms.size();
    terms.add(in.terms);
    for (int queryIdx = 0; queryIdx < in.queries.length; queryIdx++) {
        final Query query = in.queries[queryIdx];
        queries.put(query, BufferedUpdates.MAX_INT);
    }
    List<DocValuesUpdate> numericPacket = new ArrayList<>();
    numericDVUpdates.add(numericPacket);
    for (NumericDocValuesUpdate nu : in.numericDVUpdates) {
        NumericDocValuesUpdate clone = new NumericDocValuesUpdate(nu.term, nu.field, (Long) nu.value);
        clone.docIDUpto = Integer.MAX_VALUE;
        numericPacket.add(clone);
    }
    List<DocValuesUpdate> binaryPacket = new ArrayList<>();
    binaryDVUpdates.add(binaryPacket);
    for (BinaryDocValuesUpdate bu : in.binaryDVUpdates) {
        BinaryDocValuesUpdate clone = new BinaryDocValuesUpdate(bu.term, bu.field, (BytesRef) bu.value);
        clone.docIDUpto = Integer.MAX_VALUE;
        binaryPacket.add(clone);
    }
}

Example 4 with NumericDocValuesUpdate

use of org.apache.lucene.index.DocValuesUpdate.NumericDocValuesUpdate in project lucene-solr by apache.

the class BufferedUpdates method addNumericUpdate.

public void addNumericUpdate(NumericDocValuesUpdate update, int docIDUpto) {
    LinkedHashMap<Term, NumericDocValuesUpdate> fieldUpdates = numericUpdates.get(update.field);
    if (fieldUpdates == null) {
        fieldUpdates = new LinkedHashMap<>();
        numericUpdates.put(update.field, fieldUpdates);
        bytesUsed.addAndGet(BYTES_PER_NUMERIC_FIELD_ENTRY);
    }
    final NumericDocValuesUpdate current = fieldUpdates.get(update.term);
    if (current != null && docIDUpto < current.docIDUpto) {
        // got a higher docID is scheduled before the other threads.
        return;
    }
    update.docIDUpto = docIDUpto;
    // it's added last (we're interested in insertion-order).
    if (current != null) {
        fieldUpdates.remove(update.term);
    }
    fieldUpdates.put(update.term, update);
    numNumericUpdates.incrementAndGet();
    if (current == null) {
        bytesUsed.addAndGet(BYTES_PER_NUMERIC_UPDATE_ENTRY + update.sizeInBytes());
    }
}