public void datasetModuleClient() throws Exception { // Construct the client used to interact with CDAP DatasetModuleClient datasetModuleClient = new DatasetModuleClient(clientConfig); // Add a dataset module File moduleJarFile = createAppJarFile(SomeDatasetModule.class); DatasetModuleId datasetModuleId = NamespaceId.DEFAULT.datasetModule("someDatasetModule"); datasetModuleClient.add(datasetModuleId, SomeDatasetModule.class.getName(), moduleJarFile); // Fetch the dataset module information DatasetModuleMeta datasetModuleMeta = datasetModuleClient.get(datasetModuleId); // Delete all dataset modules datasetModuleClient.deleteAll(NamespaceId.DEFAULT); }
@Override public Collection<String> get() { try { List<DatasetModuleMeta> list = datasetModuleClient.list(cliConfig.getCurrentNamespace()); return Lists.newArrayList( Iterables.transform(list, new Function<DatasetModuleMeta, String>() { @Override public String apply(DatasetModuleMeta input) { return input.getName(); } }) ); } catch (IOException | UnauthenticatedException | UnauthorizedException e) { return new ArrayList<>(); } } });
@Override public void deployDatasetModule(DatasetModuleId datasetModuleId, Class<? extends DatasetModule> datasetModule) throws Exception { datasetModuleClient.add(datasetModuleId, datasetModule.getName(), createModuleJarFile(datasetModule)); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetModuleId moduleId = cliConfig.getCurrentNamespace().datasetModule( arguments.get(ArgumentName.DATASET_MODULE.toString())); DatasetModuleMeta datasetModuleMeta = datasetModuleClient.get(moduleId); Table table = Table.builder() .setHeader("name", "className", "jarLocationPath", "types", "usesModules", "usedByModules") .setRows(ImmutableList.of(datasetModuleMeta), new RowMaker<DatasetModuleMeta>() { @Override public List<?> makeRow(DatasetModuleMeta object) { return Lists.newArrayList(object.getName(), object.getClassName(), object.getJarLocationPath(), Joiner.on(", ").join(object.getTypes()), Joiner.on(", ").join(object.getUsesModules()), Joiner.on(", ").join(object.getUsedByModules())); } }).build(); cliConfig.getTableRenderer().render(cliConfig, output, table); }
public IntegrationTestManager(ClientConfig clientConfig, RESTClient restClient, File tmpFolder) { this.clientConfig = clientConfig; this.restClient = restClient; this.tmpFolder = tmpFolder; this.locationFactory = new LocalLocationFactory(tmpFolder); this.applicationClient = new ApplicationClient(clientConfig, restClient); this.artifactClient = new ArtifactClient(clientConfig, restClient); this.datasetClient = new DatasetClient(clientConfig, restClient); this.datasetModuleClient = new DatasetModuleClient(clientConfig, restClient); this.namespaceClient = new NamespaceClient(clientConfig, restClient); this.programClient = new ProgramClient(clientConfig, restClient); this.scheduleClient = new ScheduleClient(clientConfig, restClient); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetModuleId module = cliConfig.getCurrentNamespace().datasetModule( arguments.get(ArgumentName.DATASET_MODULE.toString())); datasetClient.delete(module); output.printf("Successfully deleted dataset module '%s'\n", module.getEntityName()); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { File moduleJarFile = resolver.resolvePathToFile(arguments.get(ArgumentName.DATASET_MODULE_JAR_FILE.toString())); Preconditions.checkArgument(moduleJarFile.exists(), "Module jar file '" + moduleJarFile.getAbsolutePath() + "' does not exist"); Preconditions.checkArgument(moduleJarFile.canRead()); String moduleName = arguments.get(ArgumentName.NEW_DATASET_MODULE.toString()); String moduleJarClassname = arguments.get(ArgumentName.DATASET_MODULE_JAR_CLASSNAME.toString()); datasetModuleClient.add(cliConfig.getCurrentNamespace().datasetModule(moduleName), moduleJarClassname, moduleJarFile); output.printf("Successfully deployed dataset module '%s'\n", moduleName); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetModuleId moduleId = cliConfig.getCurrentNamespace().datasetModule( arguments.get(ArgumentName.DATASET_MODULE.toString())); DatasetModuleMeta datasetModuleMeta = datasetModuleClient.get(moduleId); Table table = Table.builder() .setHeader("name", "className", "jarLocationPath", "types", "usesModules", "usedByModules") .setRows(ImmutableList.of(datasetModuleMeta), new RowMaker<DatasetModuleMeta>() { @Override public List<?> makeRow(DatasetModuleMeta object) { return Lists.newArrayList(object.getName(), object.getClassName(), object.getJarLocationPath(), Joiner.on(", ").join(object.getTypes()), Joiner.on(", ").join(object.getUsesModules()), Joiner.on(", ").join(object.getUsedByModules())); } }).build(); cliConfig.getTableRenderer().render(cliConfig, output, table); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetModuleId module = cliConfig.getCurrentNamespace().datasetModule( arguments.get(ArgumentName.DATASET_MODULE.toString())); datasetClient.delete(module); output.printf("Successfully deleted dataset module '%s'\n", module.getEntityName()); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { File moduleJarFile = resolver.resolvePathToFile(arguments.get(ArgumentName.DATASET_MODULE_JAR_FILE.toString())); Preconditions.checkArgument(moduleJarFile.exists(), "Module jar file '" + moduleJarFile.getAbsolutePath() + "' does not exist"); Preconditions.checkArgument(moduleJarFile.canRead()); String moduleName = arguments.get(ArgumentName.NEW_DATASET_MODULE.toString()); String moduleJarClassname = arguments.get(ArgumentName.DATASET_MODULE_JAR_CLASSNAME.toString()); datasetModuleClient.add(cliConfig.getCurrentNamespace().datasetModule(moduleName), moduleJarClassname, moduleJarFile); output.printf("Successfully deployed dataset module '%s'\n", moduleName); }
@Override public Collection<String> get() { try { List<DatasetModuleMeta> list = datasetModuleClient.list(cliConfig.getCurrentNamespace()); return Lists.newArrayList( Iterables.transform(list, new Function<DatasetModuleMeta, String>() { @Override public String apply(DatasetModuleMeta input) { return input.getName(); } }) ); } catch (IOException | UnauthenticatedException | UnauthorizedException e) { return new ArrayList<>(); } } });
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { List<DatasetModuleMeta> modules = client.list(cliConfig.getCurrentNamespace()); Table table = Table.builder() .setHeader("name", "className") .setRows(modules, new RowMaker<DatasetModuleMeta>() { @Override public List<?> makeRow(DatasetModuleMeta object) { return Lists.newArrayList(object.getName(), object.getClassName()); } }).build(); cliConfig.getTableRenderer().render(cliConfig, output, table); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { List<DatasetModuleMeta> modules = client.list(cliConfig.getCurrentNamespace()); Table table = Table.builder() .setHeader("name", "className") .setRows(modules, new RowMaker<DatasetModuleMeta>() { @Override public List<?> makeRow(DatasetModuleMeta object) { return Lists.newArrayList(object.getName(), object.getClassName()); } }).build(); cliConfig.getTableRenderer().render(cliConfig, output, table); }