com.edmunds.tools.databricks.maven.model.LibraryClustersModel Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of databricks-maven-plugin Show documentation

A databricks maven plugin to automate databricks deployments as part of a build

There is a newer version: 1.13.1

Show newest version

package com.edmunds.tools.databricks.maven.model;

import com.edmunds.tools.databricks.maven.util.ObjectMapperUtils;
import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Collection;
import org.apache.commons.io.FileUtils;
import org.apache.maven.plugin.MojoExecutionException;

/**
 * Used to serialize artifact and cluster data, for library attachment.
 * //TODO There is no reason to have a separate LibraryClustersModel and JobTemplatesModel.
 * //TODO We should have file where we serialize all information pertaining to a project for None-Project invocation.
 */
public class LibraryClustersModel {

    private String artifactPath;
    private Collection clusterNames;

    /**
     * Don't use this - it's for jackson deserialization only.
     */
    public LibraryClustersModel() {
    }

    public LibraryClustersModel(String artifactPath, Collection clusterNames) {
        this.artifactPath = artifactPath;
        this.clusterNames = clusterNames;
    }

    /**
     * Load LibraryClustersModel from file.
     *
     * @param libraryClusterModelFile libraryClusterModel file
     * @return LibraryClustersModel
     * @throws MojoExecutionException exception
     */
    public static LibraryClustersModel loadFromFile(File libraryClusterModelFile) throws MojoExecutionException {
        if (libraryClusterModelFile == null || !libraryClusterModelFile.exists()) {
            return null;
        }
        try {
            String libraryMappingModelJson = FileUtils
                .readFileToString(libraryClusterModelFile, StandardCharsets.UTF_8);
            return ObjectMapperUtils.deserialize(libraryMappingModelJson, LibraryClustersModel.class);
        } catch (IOException e) {
            throw new MojoExecutionException(e.getMessage(), e);
        }
    }

    public String getArtifactPath() {
        return artifactPath;
    }

    public Collection getClusterNames() {
        return clusterNames;
    }
}