com.powsybl.metrix.mapping.MappingKey Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of powsybl-metrix-mapping Show documentation
There is a newer version: 2.6.0
Show newest version
/*
 * Copyright (c) 2020, RTE (http://www.rte-france.com)
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 * SPDX-License-Identifier: MPL-2.0
 */
package com.powsybl.metrix.mapping;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonToken;
import com.powsybl.commons.json.JsonUtil;
import com.powsybl.timeseries.TimeSeriesException;

import java.io.IOException;
import java.io.UncheckedIOException;
import java.util.Objects;

/**
 * @author Paul Bui-Quang {@literal }
 */
public class MappingKey {

    private final MappingVariable mappingVariable;

    private final String id;

    public static String toJson(MappingKey key) {
        Objects.requireNonNull(key);
        return JsonUtil.toJson(key::toJson);
    }

    public void toJson(JsonGenerator generator) {
        writeJson(generator, this);
    }

    static void writeJson(JsonGenerator generator, MappingKey key) {
        Objects.requireNonNull(generator);
        try {
            generator.writeStartObject();
            generator.writeFieldName("mappingVariable");
            MappingVariable.writeJson(key.getMappingVariable(), generator);
            generator.writeStringField("id", key.getId());
            generator.writeEndObject();
        } catch (IOException e) {
            throw new UncheckedIOException(e);
        }
    }

    public static MappingKey parseJson(JsonParser parser) {
        Objects.requireNonNull(parser);
        MappingVariable mappingVariable = null;
        String id = null;
        try {
            JsonToken token;
            while ((token = parser.nextToken()) != null) {
                if (token.equals(JsonToken.FIELD_NAME)) {
                    String fieldName = parser.currentName();
                    switch (fieldName) {
                        case "mappingVariable" -> mappingVariable = MappingVariable.parseJson(parser);
                        case "id" -> {
                            parser.nextToken();
                            id = parser.getValueAsString();
                        }
                        default -> throw new IllegalStateException("Unexpected field name " + fieldName);
                    }
                } else if (token.equals(JsonToken.END_OBJECT)) {
                    if (mappingVariable != null && id != null) {
                        return new MappingKey(mappingVariable, id);
                    } else {
                        throw new IllegalStateException("Incomplete mapping key json");
                    }
                }
            }
        } catch (IOException e) {
            throw new UncheckedIOException(e);
        }
        throw new TimeSeriesException("Invalid mapping key JSON");
    }

    public MappingKey(MappingVariable mappingVariable, String id) {
        Objects.requireNonNull(mappingVariable);
        Objects.requireNonNull(id);
        this.mappingVariable = Objects.requireNonNull(mappingVariable);
        this.id = Objects.requireNonNull(id);
    }

    public MappingVariable getMappingVariable() {
        return mappingVariable;
    }

    public String getId() {
        return id;
    }

    @Override
    public int hashCode() {
        return id.hashCode() + mappingVariable.hashCode();
    }

    @Override
    public boolean equals(Object obj) {
        if (obj instanceof MappingKey other) {
            return id.equals(other.id) && mappingVariable.equals(other.mappingVariable);
        }
        return false;
    }

    @Override
    public String toString() {
        return "MappingKey(mappingVariable=" + mappingVariable + ", id=" + id + ")";
    }
}