org.apache.flink.runtime.state.IncrementalLocalKeyedStateSnapshot Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.flink.runtime.state;
import org.apache.flink.api.java.tuple.Tuple2;
import org.apache.flink.util.ExceptionUtils;
import javax.annotation.Nonnegative;
import javax.annotation.Nonnull;
import java.util.Map;
/**
* State handle for local copies of {@link IncrementalKeyedStateSnapshot}. Consists of a {@link DirectoryStateHandle} that
* represents the directory of the native RocksDB snapshot, the key keyGroupRange, and a stream state handle for Flink's state
* meta data file.
*/
public class IncrementalLocalKeyedStateSnapshot extends DirectoryKeyedStateHandle {
private static final long serialVersionUID = 1L;
/** Id of the checkpoint that created this state handle. */
@Nonnegative
private final long checkpointId;
/** Handle to Flink's state meta data. */
@Nonnull
private final StreamStateHandle metaStateHandle;
/** Map with the local state handle ID and unique gobal id of all shared state handles created by the checkpoint. */
@Nonnull
private final Map> sharedStateHandles;
public IncrementalLocalKeyedStateSnapshot(
@Nonnull KeyGroupRange keyGroupRange,
@Nonnegative long checkpointId,
@Nonnull StreamStateHandle metaStateHandle,
@Nonnull DirectoryStateHandle directoryStateHandle,
@Nonnull Map> sharedStateHandles
) {
super(directoryStateHandle, keyGroupRange);
this.checkpointId = checkpointId;
this.metaStateHandle = metaStateHandle;
this.sharedStateHandles = sharedStateHandles;
}
public long getCheckpointId() {
return checkpointId;
}
@Nonnull
public StreamStateHandle getMetaStateHandle() {
return metaStateHandle;
}
@Nonnull
public Map> getSharedStateHandles() {
return sharedStateHandles;
}
@Override
public void registerSharedStates(SharedStateRegistry stateRegistry) {
// Nothing to do, this is for local use only.
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
if (!super.equals(o)) {
return false;
}
IncrementalLocalKeyedStateHandle that = (IncrementalLocalKeyedStateHandle) o;
if (!getSharedStateHandles().equals(that.getSharedStateHandleIDs())) {
return false;
}
return getMetaStateHandle().equals(that.getMetaDataState());
}
@Override
public void discardState() throws Exception {
Exception collectedEx = null;
try {
super.discardState();
} catch (Exception e) {
collectedEx = e;
}
try {
metaStateHandle.discardState();
} catch (Exception e) {
collectedEx = ExceptionUtils.firstOrSuppressed(e, collectedEx);
}
if (collectedEx != null) {
throw collectedEx;
}
}
@Override
public long getStateSize() {
return super.getStateSize() + metaStateHandle.getStateSize();
}
@Override
public int hashCode() {
int result = super.hashCode();
result = 31 * result + getMetaStateHandle().hashCode();
result = 31 * result + getSharedStateHandles().hashCode();
return result;
}
@Override
public String toString() {
return "IncrementalLocalKeyedStateSnapshot{" +
"metaStateHandle=" + metaStateHandle +
"keyGroupRange=" + getKeyGroupRange() +
"} " + super.toString();
}
}