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.Dmaap;
41 import org.onap.dmaap.dbcapi.model.MirrorMaker;
42 import org.onap.dmaap.dbcapi.service.ApiService;
43 import org.onap.dmaap.dbcapi.service.MirrorMakerService;
45 import io.swagger.annotations.Api;
46 import io.swagger.annotations.ApiOperation;
47 import io.swagger.annotations.ApiResponse;
48 import io.swagger.annotations.ApiResponses;
51 @Api( value= "bridge", description = "Endpoint for retreiving MR Bridge metrics" )
52 @Consumes(MediaType.APPLICATION_JSON)
53 @Produces(MediaType.APPLICATION_JSON)
55 public class BridgeResource extends BaseLoggingClass {
57 private MirrorMakerService mmService = new MirrorMakerService();
60 @ApiOperation( value = "return BrTopic details",
61 notes = "Returns array of `BrTopic` objects. If source and target query params are specified, only report on that bridge. "
62 + "If detail param is true, list topics names, else just a count is returned.",
63 response = BrTopic.class)
64 @ApiResponses( value = {
65 @ApiResponse( code = 200, message = "Success", response = Dmaap.class),
66 @ApiResponse( code = 400, message = "Error", response = ApiError.class )
68 public Response getBridgedTopics(@QueryParam("mmagent") String mmagent,
69 @QueryParam("detail") Boolean detailFlag ){
70 ApiService check = new ApiService();
72 if ( mmagent == null ) {
73 return check.success(getMMcounts(Boolean.TRUE.equals(detailFlag)));
76 logger.info( "getBridgeTopics():" + " mmagent=" + mmagent);
78 if ( ! Boolean.TRUE.equals(detailFlag)) {
79 BrTopic brTopic = new BrTopic();
81 // get topics between 2 bridged locations
83 MirrorMaker mm = mmService.getMirrorMaker(mmagent);
85 return check.notFound();
88 brTopic.setTopicCount( mm.getTopicCount() );
89 brTopic.setBrSource( mm.getSourceCluster());
90 brTopic.setBrTarget( mm.getTargetCluster());
91 brTopic.setMmAgentName(mm.getMmName());
93 logger.info( "topicCount [2 locations]: " + brTopic.getTopicCount() );
95 return check.success(brTopic);
97 logger.info( "getBridgeTopics() detail:" + " mmagent=" + mmagent);
98 // get topics between 2 bridged locations
99 MirrorMaker mm = mmService.getMirrorMaker(mmagent);
101 return check.notFound();
104 return check.success(mm);
108 private BrTopic[] getMMcounts( Boolean showDetail ) {
110 List<String> mmList = mmService.getAllMirrorMakers();
113 s = mmList.size() + 1;
115 BrTopic[] brTopic = new BrTopic[s];
119 for( String key: mmList ) {
121 MirrorMaker mm = mmService.getMirrorMaker(key);
123 mCnt = mm.getTopicCount();
125 logger.info( "Count for "+ key + ": " + mCnt);
127 if (showDetail && mm!=null) {
128 brTopic[s] = new BrTopic();
129 brTopic[s].setBrSource( mm.getSourceCluster());
130 brTopic[s].setBrTarget(mm.getTargetCluster());
131 brTopic[s].setMmAgentName(mm.getMmName());
132 brTopic[s].setTopicCount(mm.getTopicCount());
137 logger.info( "topicCount [all locations]: " + totCnt );
138 brTopic[s] = new BrTopic();
139 brTopic[s].setBrSource("all");
140 brTopic[s].setBrTarget("all");
141 brTopic[s].setMmAgentName("n/a");
142 brTopic[s].setTopicCount(totCnt);
147 @ApiOperation( value = "update MirrorMaker details",
148 notes = "replace the topic list for a specific Bridge. Use JSON Body for value to replace whitelist, "
149 + "but if refreshFlag param is true, simply refresh using existing whitelist."
150 + "If split param is true, spread whitelist over smaller mmagents.",
151 response = MirrorMaker.class)
152 @ApiResponses( value = {
153 @ApiResponse( code = 200, message = "Success", response = Dmaap.class),
154 @ApiResponse( code = 400, message = "Error", response = ApiError.class )
156 public Response putBridgedTopics(@QueryParam("mmagent") String mmagent,
157 @QueryParam("refresh") Boolean refreshFlag,
158 @QueryParam("split") Boolean splitFlag,
159 MirrorMaker newBridge ){
160 ApiService check = new ApiService();
162 logger.info( "putBridgeTopics() mmagent:" + mmagent );
164 if ( mmagent != null ) { // put topics between 2 bridged locations
166 MirrorMaker mm = mmService.getMirrorMaker(mmagent);
168 return check.notFound();
171 if ( splitFlag != null && splitFlag == true ) {
172 mm = mmService.splitMM( mm );
173 } else if ( refreshFlag == null || refreshFlag == false ) {
174 logger.info( "setting whitelist from message body containing mmName=" + newBridge.getMmName());
175 if ( ! mmagent.equals(newBridge.getMmName()) ){
176 logger.error( "mmagent query param does not match mmName in body");
177 check.setCode(Status.BAD_REQUEST.getStatusCode());
178 check.setMessage("mmagent query param does not match mmName in body");
179 return check.error();
181 mm.setTopics( newBridge.getTopics() );
183 logger.info( "refreshing whitelist from memory");
185 mmService.updateMirrorMaker(mm);
186 return check.success(mm);
191 logger.error( "mmagent is required for PUT");
192 check.setCode(Status.BAD_REQUEST.getStatusCode());
193 check.setMessage("mmagent is required for PUT");
194 return check.error();