org.opensearch.migrations.dashboards.savedobjects.SavedObjectParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dashboardsSanitizer Show documentation

Everything opensearch migrations

There is a newer version: 0.2.0.4

package org.opensearch.migrations.dashboards.savedobjects;

import java.util.HashMap;
import java.util.Map;
import java.util.Optional;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;

import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@NoArgsConstructor
public class SavedObjectParser {

    private static final ObjectMapper objectMapper = new ObjectMapper();
    
    private Map> specialSavedObjectTypes = new HashMap<>() {{
        put("dashboard", Dashboard.class);
    }};

    public SavedObject load(String jsonString) throws JsonProcessingException {
        if (jsonString == null) {
            throw new IllegalArgumentException("Input string is null!");
        }

        final JsonNode json = objectMapper.readTree(jsonString);

        if (json.has("exportedCount")) {
            log.debug("Skipping the exported summary line.");
            return null;
        }

        final String dataType = Optional.ofNullable(json.get("type"))
            .orElseThrow(() -> new IllegalArgumentException("Input string doesn't contain 'type' attribute!"))
            .asText();

        if (dataType.isEmpty()) {
            throw new IllegalArgumentException("The 'type' attribute is empty!");
        }

        final Class savedObjectClass = specialSavedObjectTypes.getOrDefault(dataType, SavedObject.class);
        try {
            return savedObjectClass.getConstructor(ObjectNode.class).newInstance(json);
        } catch (Exception e) {
            throw new IllegalArgumentException("Error creating instance of " + savedObjectClass.getSimpleName(), e);
        }
    }
}