X-Git-Url: https://gerrit.onap.org/r/gitweb?p=dmaap%2Fdbcapi.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fonap%2Fdmaap%2Fdbcapi%2Fservice%2FDcaeLocationService.java;h=ad6c9932c291f5be66af42cca8217c9147141c1f;hp=de72ade62fe090d915161e212f08e3753eba6ba5;hb=4b870c6d9272e93c09952cca81f4d11e0653df9b;hpb=4444a934c6ad97d0222abc351af4c392d42f654e

diff --git a/src/main/java/org/onap/dmaap/dbcapi/service/DcaeLocationService.java b/src/main/java/org/onap/dmaap/dbcapi/service/DcaeLocationService.java
index de72ade..ad6c993 100644
--- a/src/main/java/org/onap/dmaap/dbcapi/service/DcaeLocationService.java
+++ b/src/main/java/org/onap/dmaap/dbcapi/service/DcaeLocationService.java
@@ -7,9 +7,9 @@
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
- * 
+ *
  *      http://www.apache.org/licenses/LICENSE-2.0
- * 
+ *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -20,71 +20,66 @@
 
 package org.onap.dmaap.dbcapi.service;
 
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Map;
-
 import org.onap.dmaap.dbcapi.database.DatabaseClass;
 import org.onap.dmaap.dbcapi.model.DcaeLocation;
 import org.onap.dmaap.dbcapi.model.DmaapObject.DmaapObject_Status;
 
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+
 public class DcaeLocationService {
-	
-	private Map<String, DcaeLocation> dcaeLocations = DatabaseClass.getDcaeLocations();
-	
-	public Map<String, DcaeLocation> getDcaeLocations() {
-		return dcaeLocations;
-	}
-	
-	public List<DcaeLocation> getAllDcaeLocations() {
-		return new ArrayList<DcaeLocation>(dcaeLocations.values());
-	}
-	
-	public DcaeLocation getDcaeLocation( String name ) {
-		return dcaeLocations.get(name);
-	}
-
-	public DcaeLocation addDcaeLocation( DcaeLocation location ) {
-		location.setLastMod();
-		location.setStatus(DmaapObject_Status.VALID);
-		dcaeLocations.put( location.getDcaeLocationName(), location );
-		return location;
-	}
-	
-	public DcaeLocation updateDcaeLocation( DcaeLocation location ) {
-		if ( location.getDcaeLocationName().isEmpty()) {
-			return null;
-		}
-		location.setLastMod();
-		dcaeLocations.put( location.getDcaeLocationName(), location );
-		return location;
-	}
-	
-	public DcaeLocation removeDcaeLocation( String locationName ) {
-		return dcaeLocations.remove(locationName);
-	}
-
-	public String getCentralLocation() {
-		for( Map.Entry<String, DcaeLocation> entry: dcaeLocations.entrySet() ) {
-			DcaeLocation loc = entry.getValue();
-			if ( loc.isCentral() ) {
-				// use the name of the first central location we hit
-				return loc.getDcaeLocationName();
-			}
-			
-		}
-		return "aCentralLocation";  // default value that is obvious to see is wrong
-	}	
-	
-	public boolean isEdgeLocation(String aName) {
-		DcaeLocation loc = dcaeLocations.get(aName);
-		if ( loc == null ) {
-			return false;
-		}
-		if ( ! loc.isCentral() ) {
-				return true;
-		}
-		return false;
-	}	
+
+    private static final String DEFAULT_CENTRAL_LOCATION = "aCentralLocation"; // default value that is obvious to see is wrong
+    private final Map<String, DcaeLocation> dcaeLocations;
+
+    public DcaeLocationService() {
+        this(DatabaseClass.getDcaeLocations());
+    }
+
+    DcaeLocationService(Map<String, DcaeLocation> dcaeLocations) {
+        this.dcaeLocations = dcaeLocations;
+    }
+
+    public List<DcaeLocation> getAllDcaeLocations() {
+        return new ArrayList<>(dcaeLocations.values());
+    }
+
+    public DcaeLocation getDcaeLocation(String name) {
+        return dcaeLocations.get(name);
+    }
+
+    public DcaeLocation addDcaeLocation(DcaeLocation location) {
+        location.setLastMod();
+        location.setStatus(DmaapObject_Status.VALID);
+        dcaeLocations.put(location.getDcaeLocationName(), location);
+        return location;
+    }
+
+    public DcaeLocation updateDcaeLocation(DcaeLocation location) {
+        if (location.getDcaeLocationName().isEmpty()) {
+            return null;
+        }
+        location.setLastMod();
+        dcaeLocations.put(location.getDcaeLocationName(), location);
+        return location;
+    }
+
+    public DcaeLocation removeDcaeLocation(String locationName) {
+        return dcaeLocations.remove(locationName);
+    }
+
+    String getCentralLocation() {
+
+        Optional<DcaeLocation> firstCentralLocation =
+                dcaeLocations.values().stream().filter(DcaeLocation::isCentral).findFirst();
+
+        return firstCentralLocation.isPresent() ? firstCentralLocation.get().getDcaeLocationName() : DEFAULT_CENTRAL_LOCATION;
+    }
+
+    boolean isEdgeLocation(String aName) {
+        return dcaeLocations.get(aName) != null && !dcaeLocations.get(aName).isCentral();
+    }
 
 }