private NameSpaceContainer getNamespaceEntity(NamespaceKey namespaceKey) throws NamespaceException { return namespaceService.getEntities(Collections.singletonList(namespaceKey)).get(0); }
protected NameSpaceContainer getRootContainer(List<String> path) throws NamespaceException { NamespaceKey parentKey = new NamespaceKey(path.get(0)); List<NameSpaceContainer> entities = namespaceService.getEntities(Collections.singletonList(parentKey)); return entities.get(0); }
private static DatasetType getType(NamespaceService service, List<String> namespacePath){ try{ List<NameSpaceContainer> containers = service.getEntities(Collections.singletonList(new NamespaceKey(namespacePath))); if(containers != null && !containers.isEmpty()){ NameSpaceContainer container = containers.get(0); if(container.getType() == Type.DATASET){ return container.getDataset().getType(); } } }catch(Exception ex){ } return null; }
@GET @Path("parents") @Produces(APPLICATION_JSON) public List<ParentDatasetUI> getParents() throws DatasetNotFoundException, NamespaceException { final VirtualDatasetUI virtualDatasetUI = datasetService.get(datasetPath, version); final List<ParentDatasetUI> parentDatasetUIs = Lists.newArrayList(); final List<NamespaceKey> parentDatasetPaths = Lists.newArrayList(); for (ParentDataset parentDataset : virtualDatasetUI.getParentsList()) { parentDatasetPaths.add(new NamespaceKey(parentDataset.getDatasetPathList())); } for (NameSpaceContainer nameSpaceContainer : datasetService.getNamespaceService().getEntities(parentDatasetPaths)) { if (nameSpaceContainer != null && nameSpaceContainer.getType() == Type.DATASET) { parentDatasetUIs.add(new ParentDatasetUI(nameSpaceContainer.getFullPathList(), nameSpaceContainer.getDataset().getType())); } } return parentDatasetUIs; }
protected Folder createFolder(Folder folder, NamespaceAttribute... attributes) throws NamespaceException { NamespaceKey parentKey = new NamespaceKey(folder.getPath().subList(0, folder.getPath().size() - 1)); List<NameSpaceContainer> entities = namespaceService.getEntities(Collections.singletonList(parentKey)); NameSpaceContainer container = entities.get(0); if (container == null) { // if we can't find it by id, maybe its not in the namespace throw new IllegalArgumentException(String.format("Could not find entity with path [%s].", folder.getPath())); } NamespaceKey key = new NamespaceKey(folder.getPath()); switch (container.getType()) { case SPACE: case HOME: case FOLDER: { namespaceService.addOrUpdateFolder(key, getFolderConfig(folder), attributes); break; } default: { throw new UnsupportedOperationException(String.format("Can not create a folder inside a [%s].", container.getType())); } } return getFolderFromConfig(namespaceService.getFolder(key), null); }
List<NameSpaceContainer> containers = namespace.getEntities(Collections.singletonList(new NamespaceKey(paths))); if (!containers.isEmpty()) { final NameSpaceContainer container = containers.get(0);
for (NameSpaceContainer container : namespace.getEntities(parentKeys)) { if (container != null && container.getType() == Type.DATASET) { // missing parent if (container.getDataset() != null) {
public static String getVersion(NamespaceKey namespaceKey, NamespaceService namespaceService) throws NamespaceException { NameSpaceContainer container = namespaceService.getEntities(Arrays.asList(namespaceKey)).get(0); switch (container.getType()) { case SOURCE: return container.getSource().getTag(); case SPACE: return container.getSpace().getTag(); case HOME: return container.getHome().getTag(); case FOLDER: return container.getFolder().getTag(); case DATASET: return container.getDataset().getTag(); default: throw new RuntimeException("Invalid container type"); } } }
NameSpaceContainer container = systemNamespaceService.getEntities(ImmutableList.of(new NamespaceKey(key.getRoot()))).get(0); if(container == null) { return SchemaType.UNKNOWN;
@Test public void testCreateVDSInSourceShouldFail() throws Exception { when(namespaceService.exists(any(NamespaceKey.class))).thenReturn(true); // provide a source container NameSpaceContainer namespaceContainer = new NameSpaceContainer(); namespaceContainer.setType(NameSpaceContainer.Type.SOURCE); when(namespaceService.getEntities(Matchers.anyList())).thenReturn(Collections.singletonList(namespaceContainer)); // VDS needs sql Dataset dataset = new Dataset( null, Dataset.DatasetType.VIRTUAL_DATASET, Arrays.asList("source", "path"), null, null, null, null, "sql", null, null, null ); thrown.expect(IllegalArgumentException.class); catalogServiceHelper.createCatalogItem(dataset); }
@Test public void testGetSpaceCatalogEntityById() throws Exception { SpaceConfig spaceConfig = new SpaceConfig(); spaceConfig.setId(new EntityId("space-id")); spaceConfig.setName("mySpace"); NameSpaceContainer namespaceContainer = new NameSpaceContainer(); namespaceContainer.setType(NameSpaceContainer.Type.SPACE); namespaceContainer.setSpace(spaceConfig); when(namespaceService.getEntityById(spaceConfig.getId().getId())).thenReturn(namespaceContainer); when(namespaceService.getEntities(Collections.singletonList(new NamespaceKey(spaceConfig.getName())))).thenReturn(Collections.singletonList(namespaceContainer)); // for children listing, we just send the space back to keep it simple when(namespaceService.list(new NamespaceKey(spaceConfig.getName()))).thenReturn(Collections.singletonList(namespaceContainer)); Optional<CatalogEntity> catalogEntityById = catalogServiceHelper.getCatalogEntityById(spaceConfig.getId().getId()); assertTrue(catalogEntityById.isPresent()); CatalogEntity catalogEntity = catalogEntityById.get(); assertTrue(catalogEntity instanceof Space); Space space = (Space) catalogEntity; assertEquals(space.getId(), spaceConfig.getId().getId()); assertEquals(space.getChildren().size(), 1); }
List<NameSpaceContainer> containers = userNamespaceService.getEntities(ImmutableList.of(path)); NameSpaceContainer c = containers.get(0); if(c != null &&
namespaceContainer.setSource(sourceConfig); namespaceContainer.setType(NameSpaceContainer.Type.SOURCE); when(namespaceService.getEntities(Collections.singletonList(new NamespaceKey(sourceConfig.getName())))).thenReturn(Collections.singletonList(namespaceContainer));
namespaceContainer.setSource(sourceConfig); namespaceContainer.setType(NameSpaceContainer.Type.SOURCE); when(namespaceService.getEntities(Collections.singletonList(namespaceKey))).thenReturn(Collections.singletonList(namespaceContainer));