Examples with TemporaryClassLoaderContext org.apache.flink.util.TemporaryClassLoaderContext used on opensource projects

Example 6 with TemporaryClassLoaderContext

use of org.apache.flink.util.TemporaryClassLoaderContext in project flink by apache.

the class SourceCoordinator method resetToCheckpoint.

@Override
public void resetToCheckpoint(final long checkpointId, @Nullable final byte[] checkpointData) throws Exception {
    checkState(!started, "The coordinator can only be reset if it was not yet started");
    assert enumerator == null;
    // when "start()" is called.
    if (checkpointData == null) {
        return;
    }
    LOG.info("Restoring SplitEnumerator of source {} from checkpoint.", operatorName);
    final ClassLoader userCodeClassLoader = context.getCoordinatorContext().getUserCodeClassloader();
    try (TemporaryClassLoaderContext ignored = TemporaryClassLoaderContext.of(userCodeClassLoader)) {
        final EnumChkT enumeratorCheckpoint = deserializeCheckpoint(checkpointData);
        enumerator = source.restoreEnumerator(context, enumeratorCheckpoint);
    }
}

Example 7 with TemporaryClassLoaderContext

use of org.apache.flink.util.TemporaryClassLoaderContext in project flink by apache.

the class CatalogITCase method testCreateLegacyCatalogFromUserClassLoader.

@Test
public void testCreateLegacyCatalogFromUserClassLoader() throws Exception {
    final String className = "UserCatalogFactory";
    URLClassLoader classLoader = ClassLoaderUtils.withRoot(temporaryFolder.newFolder()).addResource("META-INF/services/org.apache.flink.table.factories.TableFactory", "UserCatalogFactory").addClass(className, "import org.apache.flink.table.catalog.GenericInMemoryCatalog;\n" + "import org.apache.flink.table.factories.CatalogFactory;\n" + "import java.util.Collections;\n" + "import org.apache.flink.table.catalog.Catalog;\n" + "import java.util.HashMap;\n" + "import java.util.List;\n" + "import java.util.Map;\n" + "\tpublic class UserCatalogFactory implements CatalogFactory {\n" + "\t\t@Override\n" + "\t\tpublic Catalog createCatalog(\n" + "\t\t\t\tString name,\n" + "\t\t\t\tMap<String, String> properties) {\n" + "\t\t\treturn new GenericInMemoryCatalog(name);\n" + "\t\t}\n" + "\n" + "\t\t@Override\n" + "\t\tpublic Map<String, String> requiredContext() {\n" + "\t\t\tHashMap<String, String> hashMap = new HashMap<>();\n" + "\t\t\thashMap.put(\"type\", \"userCatalog\");\n" + "\t\t\treturn hashMap;\n" + "\t\t}\n" + "\n" + "\t\t@Override\n" + "\t\tpublic List<String> supportedProperties() {\n" + "\t\t\treturn Collections.emptyList();\n" + "\t\t}\n" + "\t}").build();
    try (TemporaryClassLoaderContext context = TemporaryClassLoaderContext.of(classLoader)) {
        TableEnvironment tableEnvironment = getTableEnvironment();
        tableEnvironment.executeSql("CREATE CATALOG cat WITH ('type'='userCatalog')");
        assertTrue(tableEnvironment.getCatalog("cat").isPresent());
    }
}