Github user milleruntime commented on a diff in the pull request: https://github.com/apache/accumulo/pull/279#discussion_r127239847 --- Diff: core/src/main/java/org/apache/accumulo/core/client/impl/Namespaces.java --- @@ -100,50 +106,150 @@ private static ZooCache getZooCache(Instance instance) { return namespaceMap; } - public static boolean exists(Instance instance, String namespaceId) { + public static boolean exists(Instance instance, Namespace.ID namespaceId) { ZooCache zc = getZooCache(instance); List<String> namespaceIds = zc.getChildren(ZooUtil.getRoot(instance) + Constants.ZNAMESPACES); - return namespaceIds.contains(namespaceId); - } - - public static String getNamespaceId(Instance instance, String namespace) throws NamespaceNotFoundException { - String id = getNameToIdMap(instance).get(namespace); - if (id == null) - throw new NamespaceNotFoundException(null, namespace, "getNamespaceId() failed to find namespace"); - return id; + return namespaceIds.contains(namespaceId.canonicalID()); } + /** + * @deprecated Do not use - String for namespace ID is not type safe. Use {@link #getNamespaceName(Instance, Namespace.ID)} + */ + @Deprecated public static String getNamespaceName(Instance instance, String namespaceId) throws NamespaceNotFoundException { String namespaceName = getIdToNameMap(instance).get(namespaceId); if (namespaceName == null) throw new NamespaceNotFoundException(namespaceId, null, "getNamespaceName() failed to find namespace"); return namespaceName; } + /** + * @deprecated Do not use - Not type safe, ambiguous String-String map. Use {@link #getNameMap(Instance)} + */ + @Deprecated public static SortedMap<String,String> getNameToIdMap(Instance instance) { return getMap(instance, true); } + /** + * @deprecated Do not use - Not type safe, ambiguous String-String map. Use {@link #getIdMap(Instance)} + */ + @Deprecated public static SortedMap<String,String> getIdToNameMap(Instance instance) { return getMap(instance, false); } - public static List<String> getTableIds(Instance instance, String namespaceId) throws NamespaceNotFoundException { + public static List<Table.ID> getTableIds(Instance instance, Namespace.ID namespaceId) throws NamespaceNotFoundException { String namespace = getNamespaceName(instance, namespaceId); - List<String> names = new LinkedList<>(); - for (Entry<String,String> nameToId : Tables.getNameToIdMap(instance).entrySet()) + List<Table.ID> tableIds = new LinkedList<>(); + for (Entry<String,Table.ID> nameToId : Tables.getNameMap(instance).entrySet()) if (namespace.equals(Tables.qualify(nameToId.getKey()).getFirst())) - names.add(nameToId.getValue()); - return names; + tableIds.add(nameToId.getValue()); + return tableIds; } - public static List<String> getTableNames(Instance instance, String namespaceId) throws NamespaceNotFoundException { + public static List<String> getTableNames(Instance instance, Namespace.ID namespaceId) throws NamespaceNotFoundException { String namespace = getNamespaceName(instance, namespaceId); List<String> names = new LinkedList<>(); - for (String name : Tables.getNameToIdMap(instance).keySet()) + for (String name : Tables.getNameMap(instance).keySet()) if (namespace.equals(Tables.qualify(name).getFirst())) names.add(name); return names; } + /** + * Populate map passed in as the BiConsumer. key = ID, value = namespaceName + */ + private static void populateMap(Instance instance, BiConsumer<String,String> biConsumer) { + final ZooCache zc = getZooCache(instance); + List<String> namespaceIds = zc.getChildren(ZooUtil.getRoot(instance) + Constants.ZNAMESPACES); + for (String id : namespaceIds) { + byte[] path = zc.get(ZooUtil.getRoot(instance) + Constants.ZNAMESPACES + "/" + id + Constants.ZNAMESPACE_NAME); + if (path != null) { + biConsumer.accept(id, new String(path, UTF_8)); + } + } + } + + /** + * Return sorted map with key = ID, value = namespaceName + */ + public static SortedMap<Namespace.ID,String> getIdMap(Instance instance) { + SortedMap<Namespace.ID,String> idMap = new TreeMap<>(); + populateMap(instance, (id, name) -> idMap.put(new Namespace.ID(id), name)); + return idMap; + } + + /** + * Return sorted map with key = namespaceName, value = ID + */ + public static SortedMap<String,Namespace.ID> getNameMap(Instance instance) { + SortedMap<String,Namespace.ID> nameMap = new TreeMap<>(); + populateMap(instance, (id, name) -> nameMap.put(name, new Namespace.ID(id))); + return nameMap; + } + + /** + * Look for namespace ID in ZK. Throw NamespaceNotFoundException if not found. + */ + public static Namespace.ID getNamespaceId(Instance instance, String namespaceName) throws NamespaceNotFoundException { + final ArrayList<Namespace.ID> singleId = new ArrayList<>(1); + populateMap(instance, (id, name) -> { + if (name.equals(namespaceName)) + singleId.add(new Namespace.ID(id)); + }); + if (singleId.isEmpty()) + throw new NamespaceNotFoundException(null, namespaceName, "getNamespaceId() failed to find namespace"); + return singleId.get(0); + } + + /** + * Look for namespace ID in ZK. Fail quietly by logging and returning null. + */ + public static Namespace.ID lookupNamespaceId(Instance instance, String namespaceName) { + Namespace.ID id = null; + try { + id = getNamespaceId(instance, namespaceName); + } catch (NamespaceNotFoundException e) { + if (log.isDebugEnabled()) + log.debug("Failed to find namespace ID from name: " + namespaceName, e); + } + return id; + } + + /** + * Return true if namespace name exists + */ + public static boolean namespaceNameExists(Instance instance, String namespaceName) { + return lookupNamespaceId(instance, namespaceName) != null; + } + + /** + * Look for namespace name in ZK. Throw NamespaceNotFoundException if not found. + */ + public static String getNamespaceName(Instance instance, Namespace.ID namespaceId) throws NamespaceNotFoundException { + final ArrayList<String> singleName = new ArrayList<>(1); + populateMap(instance, (id, name) -> { --- End diff -- This is similar to what we had: `String namespaceName = getIdToNameMap(instance).get(namespaceId);` But uses the new populateMap method I created. I created the list so we could pass a mutable object to the BiConsumer. I didn't think a String would work since they are immutable. I could change this method back to operate the old way, if you think that is cleaner. It seems like some situations Java 8 code looks messier.
--- If your project is set up for it, you can reply to this email and have your reply appear on GitHub as well. If your project does not have this feature enabled and wishes so, or if the feature is enabled but not working, please contact infrastructure at infrastruct...@apache.org or file a JIRA ticket with INFRA. ---