org.apache.flink.api.common.state.State<TXN> to org.apache.flink.api.common.state.State<TXN, CONTEXT>