Fix: Make the mapping storage type identical in CLI and REST API
Change-Id: If9cd9bea747c165204428775473e75f05ac27917
diff --git a/apps/mappingmanagement/cli/src/main/java/org/onosproject/mapping/cli/MappingStoreTypeCompleter.java b/apps/mappingmanagement/cli/src/main/java/org/onosproject/mapping/cli/MappingStoreTypeCompleter.java
index 2a7c04b..768029e 100755
--- a/apps/mappingmanagement/cli/src/main/java/org/onosproject/mapping/cli/MappingStoreTypeCompleter.java
+++ b/apps/mappingmanagement/cli/src/main/java/org/onosproject/mapping/cli/MappingStoreTypeCompleter.java
@@ -16,6 +16,7 @@
package org.onosproject.mapping.cli;
import com.google.common.collect.ImmutableList;
+import org.apache.commons.lang3.StringUtils;
import org.onosproject.cli.AbstractChoicesCompleter;
import org.onosproject.mapping.MappingStore.Type;
@@ -29,10 +30,16 @@
private static final List<Type> STORE_TYPES =
ImmutableList.of(Type.MAP_CACHE, Type.MAP_DATABASE);
+ private static final String MAP_PREFIX = "map_";
@Override
protected List<String> choices() {
return STORE_TYPES.stream().map(type ->
- type.toString().toLowerCase()).collect(Collectors.toList());
+ removeMapPrefix(type.toString().toLowerCase()))
+ .collect(Collectors.toList());
+ }
+
+ private String removeMapPrefix(String type) {
+ return StringUtils.replaceAll(type, MAP_PREFIX, "");
}
}
\ No newline at end of file