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

org.appenders.log4j2.elasticsearch.ahc.BatchResultDeserializer Maven / Gradle / Ivy

Go to download

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

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

/*-
 * #%L
 * log4j2-elasticsearch
 * %%
 * Copyright (C) 2022 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 BatchResultDeserializer extends JsonDeserializer {

    @Override
    public Object deserialize(final JsonParser p, final DeserializationContext ctxt) throws IOException {
        int took = 0;
        boolean errors = false;
        Error 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
                    final JsonDeserializer typeDeserializer = ctxt.findNonContextualValueDeserializer(ctxt.constructType(Error.class));
                    error = (Error) typeDeserializer.deserialize(p, ctxt);
                    break;
                }
                case "items": {
                    if (errors) {
                        items = new ArrayList<>();
                        p.nextValue(); // skip to START_ARRAY
                        p.nextValue(); // skip to START_OBJECT
                        final ObjectMapper mapper = (ObjectMapper) p.getCodec();
                        final MappingIterator batchResultItemMappingIterator = mapper.readValues(p, BatchItemResult.class);

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

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

}