2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Modifications Copyright (C) 2018 IBM.
8 * ================================================================================
9 * Licensed under the Apache License, Version 2.0 (the "License");
10 * you may not use this file except in compliance with the License.
11 * You may obtain a copy of the License at
13 * http://www.apache.org/licenses/LICENSE-2.0
15 * Unless required by applicable law or agreed to in writing, software
16 * distributed under the License is distributed on an "AS IS" BASIS,
17 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18 * See the License for the specific language governing permissions and
19 * limitations under the License.
20 * ============LICENSE_END=========================================================
23 package org.onap.dmaap.dbcapi.resources;
25 import java.util.List;
27 import javax.ws.rs.Consumes;
28 import javax.ws.rs.GET;
29 import javax.ws.rs.PUT;
30 import javax.ws.rs.Path;
31 import javax.ws.rs.Produces;
32 import javax.ws.rs.QueryParam;
33 import javax.ws.rs.core.MediaType;
34 import javax.ws.rs.core.Response;
35 import javax.ws.rs.core.Response.Status;
37 import org.onap.dmaap.dbcapi.logging.BaseLoggingClass;
38 import org.onap.dmaap.dbcapi.model.ApiError;
39 import org.onap.dmaap.dbcapi.model.BrTopic;
40 import org.onap.dmaap.dbcapi.model.MirrorMaker;
41 import org.onap.dmaap.dbcapi.service.ApiService;
42 import org.onap.dmaap.dbcapi.service.MirrorMakerService;
44 import io.swagger.annotations.Api;
45 import io.swagger.annotations.ApiOperation;
46 import io.swagger.annotations.ApiResponse;
47 import io.swagger.annotations.ApiResponses;
50 @Api( value= "bridge", description = "Endpoint for retreiving MR Bridge metrics" )
51 @Consumes(MediaType.APPLICATION_JSON)
52 @Produces(MediaType.APPLICATION_JSON)
54 public class BridgeResource extends BaseLoggingClass {
56 private MirrorMakerService mmService = new MirrorMakerService();
59 @ApiOperation( value = "return BrTopic details",
60 notes = "Returns array of `BrTopic` objects. If source and target query params are specified, only report on that bridge. "
61 + "If detail param is true, list topics names, else just a count is returned.",
62 response = BrTopic.class)
63 @ApiResponses( value = {
64 @ApiResponse( code = 200, message = "Success", response = BrTopic.class),
65 @ApiResponse( code = 400, message = "Error", response = ApiError.class )
67 public Response getBridgedTopics(@QueryParam("mmagent") String mmagent,
68 @QueryParam("detail") Boolean detailFlag ){
69 ApiService check = new ApiService();
71 if ( mmagent == null ) {
72 return check.success(getMMcounts(Boolean.TRUE.equals(detailFlag)));
75 logger.info( "getBridgeTopics():" + " mmagent=" + mmagent);
77 if ( ! Boolean.TRUE.equals(detailFlag)) {
78 BrTopic brTopic = new BrTopic();
80 // get topics between 2 bridged locations
82 MirrorMaker mm = mmService.getMirrorMaker(mmagent);
84 return check.notFound();
87 brTopic.setTopicCount( mm.getTopicCount() );
88 brTopic.setBrSource( mm.getSourceCluster());
89 brTopic.setBrTarget( mm.getTargetCluster());
90 brTopic.setMmAgentName(mm.getMmName());
92 logger.info( "topicCount [2 locations]: " + brTopic.getTopicCount() );
94 return check.success(brTopic);
96 logger.info( "getBridgeTopics() detail:" + " mmagent=" + mmagent);
97 // get topics between 2 bridged locations
98 MirrorMaker mm = mmService.getMirrorMaker(mmagent);
100 return check.notFound();
103 return check.success(mm);
107 private BrTopic[] getMMcounts( Boolean showDetail ) {
109 List<String> mmList = mmService.getAllMirrorMakers();
112 s = mmList.size() + 1;
114 BrTopic[] brTopic = new BrTopic[s];
118 for( String key: mmList ) {
120 MirrorMaker mm = mmService.getMirrorMaker(key);
122 mCnt = mm.getTopicCount();
124 logger.info( "Count for "+ key + ": " + mCnt);
126 if (showDetail && mm!=null) {
127 brTopic[s] = new BrTopic();
128 brTopic[s].setBrSource( mm.getSourceCluster());
129 brTopic[s].setBrTarget(mm.getTargetCluster());
130 brTopic[s].setMmAgentName(mm.getMmName());
131 brTopic[s].setTopicCount(mm.getTopicCount());
136 logger.info( "topicCount [all locations]: " + totCnt );
137 brTopic[s] = new BrTopic();
138 brTopic[s].setBrSource("all");
139 brTopic[s].setBrTarget("all");
140 brTopic[s].setMmAgentName("n/a");
141 brTopic[s].setTopicCount(totCnt);
146 @ApiOperation( value = "update MirrorMaker details",
147 notes = "replace the topic list for a specific Bridge. Use JSON Body for value to replace whitelist, "
148 + "but if refreshFlag param is true, simply refresh using existing whitelist."
149 + "If split param is true, spread whitelist over smaller mmagents.",
150 response = MirrorMaker.class)
151 @ApiResponses( value = {
152 @ApiResponse( code = 200, message = "Success", response = BrTopic.class),
153 @ApiResponse( code = 400, message = "Error", response = ApiError.class )
155 public Response putBridgedTopics(@QueryParam("mmagent") String mmagent,
156 @QueryParam("refresh") Boolean refreshFlag,
157 @QueryParam("split") Boolean splitFlag,
158 MirrorMaker newBridge ){
159 ApiService check = new ApiService();
161 logger.info( "putBridgeTopics() mmagent:" + mmagent );
163 if ( mmagent != null ) { // put topics between 2 bridged locations
165 MirrorMaker mm = mmService.getMirrorMaker(mmagent);
167 return check.notFound();
170 if ( splitFlag != null && splitFlag == true ) {
171 mm = mmService.splitMM( mm );
172 } else if ( refreshFlag == null || refreshFlag == false ) {
173 logger.info( "setting whitelist from message body containing mmName=" + newBridge.getMmName());
174 if ( ! mmagent.equals(newBridge.getMmName()) ){
175 logger.error( "mmagent query param does not match mmName in body");
176 check.setCode(Status.BAD_REQUEST.getStatusCode());
177 check.setMessage("mmagent query param does not match mmName in body");
178 return check.error();
180 mm.setTopics( newBridge.getTopics() );
182 logger.info( "refreshing whitelist from memory");
184 mmService.updateMirrorMaker(mm);
185 return check.success(mm);
190 logger.error( "mmagent is required for PUT");
191 check.setCode(Status.BAD_REQUEST.getStatusCode());
192 check.setMessage("mmagent is required for PUT");
193 return check.error();