@Override public void snapshotState(StateSnapshotContext context) throws Exception { OperatorStateCheckpointOutputStream outStream = context.getRawOperatorStateOutput(); IN_CHECKPOINT_LATCH.trigger(); // this should lock outStream.write(1); } }
@Override public void snapshotState(StateSnapshotContext context) throws Exception { KeyedStateCheckpointOutputStream out = context.getRawKeyedOperatorStateOutput(); DataOutputView dov = new DataOutputViewStreamWrapper(out); // write raw keyed state that goes into snapshot int count = 0; for (int kg : out.getKeyGroupList()) { out.startNewKeyGroup(kg); dov.writeInt(kg + 2); ++count; } Assert.assertEquals(MAX_PARALLELISM, count); // write raw operator state that goes into snapshot OperatorStateCheckpointOutputStream outOp = context.getRawOperatorStateOutput(); dov = new DataOutputViewStreamWrapper(outOp); for (int i = 0; i < 13; ++i) { outOp.startNewPartition(); dov.writeInt(42 + i); } }