io.cdap.cdap.internal.app.runtime.schedule.trigger.DefaultProgramStatusTriggerInfo Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of cdap-app-fabric Show documentation
The newest version!
/*
 * Copyright © 2017 Cask Data, 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 io.cdap.cdap.internal.app.runtime.schedule.trigger;

import io.cdap.cdap.api.ProgramStatus;
import io.cdap.cdap.api.app.ProgramType;
import io.cdap.cdap.api.schedule.ProgramStatusTriggerInfo;
import io.cdap.cdap.api.workflow.WorkflowToken;
import io.cdap.cdap.common.app.RunIds;
import java.io.Externalizable;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import javax.annotation.Nullable;
import org.apache.twill.api.RunId;

/**
 * The program status trigger information to be passed to the triggered program.
 */
public class DefaultProgramStatusTriggerInfo extends AbstractTriggerInfo
    implements ProgramStatusTriggerInfo, Externalizable {

  private static final long serialVersionUID = 1L;

  private String namespace;
  private String applicationName;
  private ProgramType programType;
  private String program;
  private RunId runId;
  private ProgramStatus programStatus;
  @Nullable
  private WorkflowToken workflowToken;
  private Map runtimeArguments;

  /**
   * Constructor for {@link Externalizable} to use only. Don't call it directly.
   */
  public DefaultProgramStatusTriggerInfo() {
    super(Type.PROGRAM_STATUS);
  }

  public DefaultProgramStatusTriggerInfo(String namespace, String applicationName,
      ProgramType programType, String program,
      RunId runId, ProgramStatus programStatus,
      @Nullable WorkflowToken workflowToken,
      Map runtimeArguments) {
    super(Type.PROGRAM_STATUS);
    this.namespace = namespace;
    this.applicationName = applicationName;
    this.programType = programType;
    this.program = program;
    this.runId = runId;
    this.programStatus = programStatus;
    this.workflowToken = workflowToken;
    this.runtimeArguments = Collections.unmodifiableMap(new HashMap<>(runtimeArguments));
  }

  @Override
  public String getNamespace() {
    return namespace;
  }

  @Override
  public String getApplicationName() {
    return applicationName;
  }

  @Override
  public ProgramType getProgramType() {
    return programType;
  }

  @Override
  public String getProgram() {
    return program;
  }

  @Override
  public RunId getRunId() {
    return runId;
  }

  @Override
  public ProgramStatus getProgramStatus() {
    return programStatus;
  }

  @Nullable
  @Override
  public WorkflowToken getWorkflowToken() {
    return workflowToken;
  }

  @Override
  public Map getRuntimeArguments() {
    return runtimeArguments;
  }

  @Override
  public void writeExternal(ObjectOutput out) throws IOException {
    out.writeUTF(namespace);
    out.writeUTF(applicationName);
    out.writeObject(programType);
    out.writeUTF(program);
    out.writeUTF(runId.getId());
    out.writeObject(programStatus);
    out.writeObject(workflowToken);

    // Write out the map size and the entries
    out.writeInt(runtimeArguments.size());
    for (Map.Entry entry : runtimeArguments.entrySet()) {
      out.writeObject(entry.getKey());
      out.writeObject(entry.getValue());
    }
  }

  @Override
  public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
    namespace = in.readUTF();
    applicationName = in.readUTF();
    programType = (ProgramType) in.readObject();
    program = in.readUTF();
    runId = RunIds.fromString(in.readUTF());
    programStatus = (ProgramStatus) in.readObject();
    workflowToken = (WorkflowToken) in.readObject();

    Map args = new HashMap<>();
    int argsSize = in.readInt();
    for (int i = 0; i < argsSize; i++) {
      args.put((String) in.readObject(), (String) in.readObject());
    }

    runtimeArguments = Collections.unmodifiableMap(args);
  }
}