Checkpoint.java

// Copyright 2017 Twitter. All rights reserved.
//
// 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.twitter.heron.spi.statefulstorage;

import com.twitter.heron.proto.ckptmgr.CheckpointManager;
import com.twitter.heron.proto.system.PhysicalPlans;

public class Checkpoint {
  private final String topologyName;
  private final String checkpointId;
  private final PhysicalPlans.Instance instanceInfo;

  private CheckpointManager.InstanceStateCheckpoint checkpoint;
  private int nBytes;

  public Checkpoint(String topologyName, PhysicalPlans.Instance instanceInfo,
                    CheckpointManager.InstanceStateCheckpoint checkpoint) {
    this.topologyName = topologyName;
    this.checkpointId = checkpoint.getCheckpointId();
    this.instanceInfo = instanceInfo;
    this.checkpoint = checkpoint;
    this.nBytes = checkpoint.getSerializedSize();
  }

  public String getTopologyName() {
    return topologyName;
  }

  public String getCheckpointId() {
    return checkpointId;
  }

  public String getComponent() {
    return instanceInfo.getInfo().getComponentName();
  }

  public String getInstance() {
    return instanceInfo.getInstanceId();
  }

  public int getTaskId() {
    return instanceInfo.getInfo().getTaskId();
  }

  public CheckpointManager.InstanceStateCheckpoint getCheckpoint() {
    return this.checkpoint;
  }

  @Override
  public String toString() {
    return String.format("%s %s %s %s", topologyName, checkpointId, getComponent(), getInstance());
  }
}