ai.databand.spark.SparkIOSource Maven / Gradle / Ivy

Go to download

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

DBND Tracking API Client.

There is a newer version: 1.0.26.1

/*
 * © Copyright Databand.ai, an IBM Company 2022
 */

package ai.databand.spark;

import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Collections;
import java.util.Map;
import java.util.UUID;

public class SparkIOSource {

    private static final Logger LOG = LoggerFactory.getLogger(SparkIOSource.class);
    private static final ObjectMapper MAPPER = new ObjectMapper();

    private final String path;

    private final Map properties;

    private final String metricKey;

    private final String trackingSource;

    public SparkIOSource(String path, String trackingSource, Map properties) {
        this.path = path;
        this.trackingSource = trackingSource;
        this.properties = properties;
        this.metricKey = "spark-io-" + UUID.randomUUID();
    }

    public SparkIOSource(String path, String trackingSource) {
        this(path, trackingSource, Collections.emptyMap());
    }

    public String getPath() {
        return path;
    }

    public String getTrackingSource() {
        return trackingSource;
    }

    @JsonIgnore
    public String metricKey() {
        return metricKey;
    }

    public Map getProperties() {
        return properties;
    }

    public String toString() {
        try {
            return MAPPER.writeValueAsString(this);
        } catch (JsonProcessingException e) {
            LOG.error("Unable to serialize Spark IO source", e);
            return "{}";
        }
    }
}