1 files changed, 191 insertions, 0 deletions
diff --git a/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/resources/BridgeResource.java b/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/resources/BridgeResource.java
new file mode 100644
index 0000000..299c48f
--- /dev/null
+++ b/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/resources/BridgeResource.java
@@ -0,0 +1,191 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * org.onap.dmaap
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Modifications Copyright (C) 2018 IBM.
+ * ================================================================================
+ * 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.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.dmaap.dbcapi.resources;
+
+import java.util.List;
+
+import javax.ws.rs.Consumes;
+import javax.ws.rs.GET;
+import javax.ws.rs.PUT;
+import javax.ws.rs.Path;
+import javax.ws.rs.Produces;
+import javax.ws.rs.QueryParam;
+import javax.ws.rs.core.MediaType;
+import javax.ws.rs.core.Response;
+
+import org.onap.dmaap.dbcapi.logging.BaseLoggingClass;
+import org.onap.dmaap.dbcapi.model.ApiError;
+import org.onap.dmaap.dbcapi.model.BrTopic;
+import org.onap.dmaap.dbcapi.model.MirrorMaker;
+import org.onap.dmaap.dbcapi.service.MirrorMakerService;
+
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import io.swagger.annotations.ApiResponse;
+import io.swagger.annotations.ApiResponses;
+
+import static javax.ws.rs.core.Response.Status.BAD_REQUEST;
+
+@Path("/bridge")
+@Api( value= "bridge", description = "Endpoint for retreiving MR Bridge metrics" )
+@Consumes(MediaType.APPLICATION_JSON)
+@Produces(MediaType.APPLICATION_JSON)
+@Authorization
+public class BridgeResource extends BaseLoggingClass {
+	
+	private MirrorMakerService mmService = new MirrorMakerService();
+	private ResponseBuilder responseBuilder = new ResponseBuilder();
+
+	@GET
+	@ApiOperation( value = "return BrTopic details", 
+	notes = "Returns array of  `BrTopic` objects. If source and target query params are specified, only report on that bridge.  "
+			+ "If detail param is true, list topics names, else just a count is returned.", 
+	response = BrTopic.class)
+@ApiResponses( value = {
+    @ApiResponse( code = 200, message = "Success", response = BrTopic.class),
+    @ApiResponse( code = 400, message = "Error", response = ApiError.class )
+})
+	public Response	getBridgedTopics(@QueryParam("mmagent") String mmagent,
+						   			@QueryParam("detail") Boolean detailFlag ){
+
+		if ( mmagent == null ) {
+			return responseBuilder.success(getMMcounts(Boolean.TRUE.equals(detailFlag)));
+
+		}
+		logger.info( "getBridgeTopics():" + " mmagent=" + mmagent);
+
+		if ( ! Boolean.TRUE.equals(detailFlag)) {
+			BrTopic brTopic = new BrTopic();
+			
+			// get topics between 2 bridged locations
+
+			MirrorMaker mm = mmService.getMirrorMaker(mmagent);
+			if ( mm == null ) {		
+				return responseBuilder.notFound();
+			} 
+					
+			brTopic.setTopicCount( mm.getTopicCount() );
+			brTopic.setBrSource( mm.getSourceCluster());
+			brTopic.setBrTarget( mm.getTargetCluster());
+			brTopic.setMmAgentName(mm.getMmName());
+			
+			logger.info( "topicCount [2 locations]: " + brTopic.getTopicCount() );
+		
+			return responseBuilder.success(brTopic);
+		} else {	
+			logger.info( "getBridgeTopics() detail:" + " mmagent=" + mmagent);
+			// get topics between 2 bridged locations	
+			MirrorMaker mm = mmService.getMirrorMaker(mmagent);
+			if ( mm == null ) {		
+				return responseBuilder.notFound();
+			} 
+
+			return responseBuilder.success(mm);
+		}
+	}
+	
+	private BrTopic[] getMMcounts( Boolean showDetail ) {
+		
+		List<String> mmList = mmService.getAllMirrorMakers();
+		int s = 1;
+		if ( showDetail ) {
+			s = mmList.size() + 1;
+		}
+		BrTopic[] brTopic = new BrTopic[s];
+		
+		int totCnt = 0;
+		s = 0;
+		for( String key: mmList ) {
+			int mCnt = 0;
+			MirrorMaker mm = mmService.getMirrorMaker(key);
+			if ( mm != null ) {
+				mCnt = mm.getTopicCount();
+			}
+			logger.info( "Count for "+ key + ": " + mCnt);
+			totCnt += mCnt;
+			if (showDetail && mm!=null) {
+				brTopic[s] =  new BrTopic();
+				brTopic[s].setBrSource( mm.getSourceCluster());
+				brTopic[s].setBrTarget(mm.getTargetCluster());
+				brTopic[s].setMmAgentName(mm.getMmName());
+				brTopic[s].setTopicCount(mm.getTopicCount());
+				s++;
+			}
+		}
+		
+		logger.info( "topicCount [all locations]: " + totCnt );
+		brTopic[s] =  new BrTopic();
+		brTopic[s].setBrSource("all");
+		brTopic[s].setBrTarget("all");
+		brTopic[s].setMmAgentName("n/a");
+		brTopic[s].setTopicCount(totCnt);
+		return brTopic;
+	}
+	
+	@PUT
+	@ApiOperation( value = "update MirrorMaker details", 
+		notes = "replace the topic list for a specific Bridge.  Use JSON Body for value to replace whitelist, "
+				+ "but if refreshFlag param is true, simply refresh using existing whitelist."
+				+ "If split param is true, spread whitelist over smaller mmagents.", 
+		response = MirrorMaker.class)
+	@ApiResponses( value = {
+	    @ApiResponse( code = 200, message = "Success", response = BrTopic.class),
+	    @ApiResponse( code = 400, message = "Error", response = ApiError.class )
+	})
+	public Response	putBridgedTopics(@QueryParam("mmagent") String mmagent,
+						   			@QueryParam("refresh") Boolean refreshFlag,
+						   			@QueryParam("split") Boolean splitFlag,
+						   			MirrorMaker newBridge ){
+		logger.info( "putBridgeTopics() mmagent:" +  mmagent );
+
+		if ( mmagent != null ) {		// put topics between 2 bridged locations
+			
+			MirrorMaker mm = mmService.getMirrorMaker(mmagent);
+			if ( mm == null ) {		
+				return responseBuilder.notFound();
+			} 
+			
+			if ( splitFlag != null && splitFlag == true ) {
+				mm = mmService.splitMM( mm );
+			} else if ( refreshFlag == null  ||  refreshFlag == false ) {
+				logger.info( "setting whitelist from message body containing mmName=" + newBridge.getMmName());
+				if ( ! mmagent.equals(newBridge.getMmName()) ){
+					logger.error( "mmagent query param does not match mmName in body");
+					return responseBuilder.error(new ApiError(BAD_REQUEST.getStatusCode(),
+							"mmagent query param does not match mmName in body"));
+				}
+				mm.setTopics( newBridge.getTopics() );
+			} else {
+				logger.info( "refreshing whitelist from memory");
+			}
+			mmService.updateMirrorMaker(mm);
+			return responseBuilder.success(mm);
+		}
+
+		else {
+			logger.error( "mmagent is required for PUT");
+			return responseBuilder.error(new ApiError(BAD_REQUEST.getStatusCode(), "mmagent is required for PUT"));
+		}
+
+	}
+}