All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.appenders.log4j2.elasticsearch.jest.BulkResultDeserializer Maven / Gradle / Ivy

Go to download

Log4j2 Appender plugin pushing logs in batches to Elasticsearch (2.x/5.x/6.x) clusters

The newest version!
package org.appenders.log4j2.elasticsearch.jest;

/*-
 * #%L
 * log4j2-elasticsearch
 * %%
 * Copyright (C) 2019 Rafal Foltynski
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.JsonDeserializer;
import com.fasterxml.jackson.databind.MappingIterator;
import com.fasterxml.jackson.databind.ObjectMapper;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class BulkResultDeserializer extends JsonDeserializer {

    @Override
    public Object deserialize(JsonParser p, DeserializationContext ctxt) throws IOException {
        int took = 0;
        boolean errors = false;
        BulkError error = null;
        int status = -1;
        List items = null;

        String fieldName;
        while ((fieldName = p.nextFieldName()) != null)  {
            switch (fieldName) {
                case "took": {
                    took = p.nextIntValue(-1);
                    break;
                }
                case "errors": {
                    errors = p.nextBooleanValue();
                    break;
                }
                case "status": {
                    status = p.nextIntValue(-1);
                    break;
                }
                case "error": {
                    p.nextValue(); // skip to START_OBJECT or VALUE_NULL
                    JsonDeserializer typeDeserializer = ctxt.findNonContextualValueDeserializer(ctxt.constructType(BulkError.class));
                    error = (BulkError) typeDeserializer.deserialize(p, ctxt);
                    break;
                }
                case "items": {
                    if (errors) {
                        items = new ArrayList<>();
                        p.nextValue(); // skip to START_ARRAY
                        p.nextValue(); // skip to START_OBJECT
                        ObjectMapper mapper = (ObjectMapper) p.getCodec();
                        MappingIterator bulkResultItemMappingIterator = mapper.readValues(p, BulkResultItem.class);

                        while (bulkResultItemMappingIterator.hasNext()) {
                            items.add(bulkResultItemMappingIterator.next());
                        }
                    }
                    break;
                }
            }
        }

        return new BufferedBulkResult(took, errors, error, status, items);
    }

}