public static boolean isSystemDatasetInUserNamespace(DatasetId datasetInstanceId) { return !NamespaceId.SYSTEM.equals(datasetInstanceId.getParent()) && ("system.queue.config".equals(datasetInstanceId.getEntityName()) || datasetInstanceId.getEntityName().startsWith("system.sharded.queue") || datasetInstanceId.getEntityName().startsWith("system.queue") || datasetInstanceId.getEntityName().startsWith("system.stream")); }
@Override public void addInstance(String datasetType, DatasetId datasetInstanceId, DatasetProperties props, @Nullable KerberosPrincipalId ownerPrincipal) throws DatasetManagementException { clientCache.getUnchecked(datasetInstanceId.getParent()) .addInstance(datasetInstanceId.getEntityName(), datasetType, props, ownerPrincipal); }
@Override public void updateInstance(DatasetId datasetInstanceId, DatasetProperties props) throws DatasetManagementException { clientCache.getUnchecked(datasetInstanceId.getParent()) .updateInstance(datasetInstanceId.getEntityName(), props); }
@Override public boolean hasInstance(DatasetId datasetInstanceId) throws DatasetManagementException { return clientCache.getUnchecked(datasetInstanceId.getParent()) .getInstance(datasetInstanceId.getEntityName()) != null; }
@Nullable public DatasetSpecification get(DatasetId datasetInstanceId) { return get(getInstanceKey(datasetInstanceId.getParent(), datasetInstanceId.getEntityName()), DatasetSpecification.class); }
@Nullable public DatasetSpecification get(DatasetId datasetInstanceId) { return get(getInstanceKey(datasetInstanceId.getParent(), datasetInstanceId.getEntityName()), DatasetSpecification.class); }
@Override @Nullable public DatasetId fromNamespaced(DatasetId datasetInstanceId) { String namespacedDatasetName = datasetInstanceId.getEntityName(); if (!contains(namespacedDatasetName, datasetInstanceId.getNamespace())) { return null; } String prefix = rootPrefix + datasetInstanceId.getNamespace() + "."; String nonNamespaced = namespacedDatasetName.substring(prefix.length()); return datasetInstanceId.getParent().dataset(nonNamespaced); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetId dataset = cliConfig.getCurrentNamespace().dataset(arguments.get(ArgumentName.DATASET.toString())); datasetClient.delete(dataset); output.printf("Successfully deleted dataset instance '%s'\n", dataset.getEntityName()); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetId instance = cliConfig.getCurrentNamespace().dataset(arguments.get(ArgumentName.DATASET.toString())); datasetClient.truncate(instance); output.printf("Successfully truncated dataset '%s'\n", instance.getEntityName()); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetId instance = cliConfig.getCurrentNamespace().dataset(arguments.get(ArgumentName.DATASET.toString())); datasetClient.truncate(instance); output.printf("Successfully truncated dataset '%s'\n", instance.getEntityName()); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetId dataset = cliConfig.getCurrentNamespace().dataset(arguments.get(ArgumentName.DATASET.toString())); datasetClient.delete(dataset); output.printf("Successfully deleted dataset instance '%s'\n", dataset.getEntityName()); }
private void addDataset(MDSKey.Builder keyBuilder, DatasetId datasetInstance) { keyBuilder.add(DATASET_MARKER) .add(datasetInstance.getNamespace()) .add(datasetInstance.getEntityName()); }
@Override public DatasetId namespace(DatasetId datasetInstanceId) { String namespaced = namespace(datasetInstanceId.getParent(), datasetInstanceId.getEntityName()); return datasetInstanceId.getParent().dataset(namespaced); }
public boolean delete(DatasetId datasetInstanceId) { if (get(datasetInstanceId) == null) { return false; } deleteAll(getInstanceKey(datasetInstanceId.getParent(), datasetInstanceId.getEntityName())); return true; }
@Override public MDSKey getKey(DatasetId datasetInstance) { return new MDSKey.Builder() .add(datasetInstance.getNamespace()) .add(datasetInstance.getEntityName()) .build(); }
@Override public MDSKey getKey(DatasetId datasetInstance) { return new MDSKey.Builder() .add(datasetInstance.getNamespace()) .add(datasetInstance.getEntityName()) .build(); }
@Override public void perform(Arguments arguments, PrintStream output) throws Exception { DatasetId instance = cliConfig.getCurrentNamespace().dataset(arguments.get(ArgumentName.DATASET.toString())); Map<String, String> properties = ArgumentParser.parseMap(arguments.get(ArgumentName.DATASET_PROPERTIES.toString()), ArgumentName.DATASET_PROPERTIES.toString()); datasetClient.updateExisting(instance, properties); output.printf("Successfully updated properties for dataset instance '%s' to %s", instance.getEntityName(), GSON.toJson(properties)); }
private HttpResponse getInstance(DatasetId instance) throws IOException { URL instanceUrl = getUrl(instance.getNamespace(), "/data/datasets/" + instance.getEntityName()); HttpRequest request = HttpRequest.get(instanceUrl).build(); return HttpRequests.execute(request); }
private HttpResponse deleteInstance(DatasetId instance) throws IOException { HttpRequest request = HttpRequest.delete(getUrl(instance.getNamespace(), "/data/datasets/" + instance.getEntityName())).build(); return HttpRequests.execute(request); }
private void testAdminOp(DatasetId datasetInstanceId, String opName, int expectedStatus, Object expectedResult) throws IOException { String path = String.format("/namespaces/%s/data/datasets/%s/admin/%s", datasetInstanceId.getNamespace(), datasetInstanceId.getEntityName(), opName); URL targetUrl = resolve(path); HttpResponse response = HttpRequests.execute(HttpRequest.post(targetUrl).build()); DatasetAdminOpResponse body = getResponse(response.getResponseBody()); Assert.assertEquals(expectedStatus, response.getResponseCode()); Assert.assertEquals(expectedResult, body.getResult()); }