ml.comet.experiment.impl.BaseArtifactImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of comet-java-client Show documentation

There is a newer version: 1.1.14

package ml.comet.experiment.impl;

import com.vdurmont.semver4j.Semver;
import lombok.Getter;
import lombok.Setter;
import lombok.ToString;
import ml.comet.experiment.impl.utils.RestApiUtils;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * The base superclass for all artifact implementations.
 */
abstract class BaseArtifactImpl {
    @Getter
    @ToString.Include
    private final String name;
    @Getter
    @ToString.Include
    private final String type;
    @Setter
    Set aliases;
    @Getter
    @Setter
    @ToString.Include
    Semver semanticVersion;
    @Setter
    Set versionTags;
    @Setter
    String metadataJson;
    @Setter
    private Map metadata;

    BaseArtifactImpl(String name, String type) {
        this.name = name;
        this.type = type;
    }

    public Map getMetadata() {
        if (this.metadata != null) {
            return this.metadata;
        }
        if (StringUtils.isNotBlank(this.metadataJson)) {
            try {
                this.metadata = RestApiUtils.metadataFromJson(this.metadataJson);
                return this.metadata;
            } catch (Throwable e) {
                this.getLogger().error("Failed to parse artifact metadata from JSON {}", this.metadataJson, e);
            }
        }
        return new HashMap<>();
    }

    public Set getAliases() {
        if (this.aliases == null) {
            return new HashSet<>();
        }
        return this.aliases;
    }

    public Set getVersionTags() {
        if (this.versionTags == null) {
            return new HashSet<>();
        }
        return this.versionTags;
    }

    abstract Logger getLogger();
}