def get_state_backend(self) -> StateBackend: """ Gets the state backend that defines how to store and checkpoint state. .. seealso:: :func:`set_state_backend` :return: The :class:`StateBackend`. """ j_state_backend = self._j_stream_execution_environment.getStateBackend() return _from_j_state_backend(j_state_backend)
def test_create_custom_state_backend(self): gateway = get_gateway() JConfiguration = gateway.jvm.org.apache.flink.configuration.Configuration j_config = JConfiguration() j_factory = load_java_class("org.apache.flink.streaming.runtime.tasks." "StreamTaskTest$TestMemoryStateBackendFactory").newInstance() context_classloader = gateway.jvm.Thread.currentThread().getContextClassLoader() state_backend = _from_j_state_backend(j_factory.createFromConfig(j_config, context_classloader)) self.assertIsInstance(state_backend, CustomStateBackend)