com.cloudera.nav.plugin.examples.stetson.StetsonScript.java Source code

Introduction

Here is the source code for com.cloudera.nav.plugin.examples.stetson.StetsonScript.java

Source

/*
 * Copyright (c) 2015 Cloudera, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.cloudera.nav.plugin.examples.stetson;

import com.cloudera.nav.plugin.model.MD5IdGenerator;
import com.cloudera.nav.plugin.model.SourceType;
import com.cloudera.nav.plugin.model.annotations.MClass;
import com.cloudera.nav.plugin.model.annotations.MProperty;
import com.cloudera.nav.plugin.model.annotations.MRelation;
import com.cloudera.nav.plugin.model.entities.CustomEntity;
import com.cloudera.nav.plugin.model.entities.EndPointProxy;
import com.cloudera.nav.plugin.model.entities.Entity;
import com.cloudera.nav.plugin.model.entities.EntityType;
import com.cloudera.nav.plugin.model.relations.RelationRole;
import com.google.common.base.Preconditions;

import org.apache.commons.lang.StringUtils;

/**
 * Represents a template defined by a script in a hypothetical custom DSL
 */
@MClass
public class StetsonScript extends CustomEntity {

    private EndPointProxy pigOperation;
    private String script;

    public StetsonScript(String namespace) {
        // Because the namespace is given to input/output we ensure it
        // exists when it is used by adding it as a c'tor parameter
        Preconditions.checkArgument(StringUtils.isNotEmpty(namespace));
        setNamespace(namespace);
    }

    /**
     * The StetsonScript represents a template and is therefore always an
     * OPERATION entity
     */
    @Override
    @MProperty
    public EntityType getType() {
        return EntityType.OPERATION;
    }

    /**
     * The script template is uniquely defined by the name and the owner
     */
    @Override
    public String generateId() {
        return MD5IdGenerator.generateIdentity(getNamespace(), getPigOperation().getIdentity());
    }

    /**
     * The script contents in the custom DSL
     */
    @MProperty
    public String getScript() {
        return script;
    }

    /**
     * The StetsonScript is linked to a PIG operation via a Logical-Physical
     * relationship where the Pig operation is the PHYSICAL node
     */
    @MRelation(role = RelationRole.PHYSICAL)
    public Entity getPigOperation() {
        return pigOperation;
    }

    public void setPigOperation(String pigOperationId) {
        this.pigOperation = new EndPointProxy(pigOperationId, SourceType.PIG, EntityType.OPERATION);
    }

    public void setScript(String script) {
        this.script = script;
    }
}