/**
* Licensed to Cloudera, Inc. under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. Cloudera, Inc. licenses this file
* to you 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.
*/
package com.cloudera.flume.reporter.server;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import org.apache.thrift.TException;
import org.apache.thrift.TProcessor;
import org.apache.thrift.transport.TTransportException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.cloudera.flume.conf.FlumeConfiguration;
import com.cloudera.flume.core.Attributes;
import com.cloudera.flume.core.Attributes.Type;
import com.cloudera.flume.reporter.ReportEvent;
import com.cloudera.flume.reporter.ReportManager;
import com.cloudera.flume.reporter.Reportable;
import com.cloudera.flume.reporter.server.thrift.ThriftFlumeReport;
import com.cloudera.flume.reporter.server.thrift.ThriftFlumeReportServer;
import com.cloudera.flume.util.ThriftServer;
import com.google.common.base.Preconditions;
/**
* Serves reports over Thrift
*/
public class ThriftReportServer extends ThriftServer implements
ThriftFlumeReportServer.Iface {
protected static final Logger LOG = LoggerFactory
.getLogger(ThriftReportServer.class);
/**
* Reads the port to start on from cfg, property REPORT_SERVER_PORT.
*/
public ThriftReportServer(FlumeConfiguration cfg) {
this.port = cfg.getReportServerPort();
}
/**
* Constructs a ReportServer to start on supplied port
*/
public ThriftReportServer(int port) {
this.port = port;
}
/**
* Turn a ReportEvent into a serializable object
*/
static public ThriftFlumeReport reportToThrift(ReportEvent report) {
Preconditions.checkNotNull(report, "reportToThrift: report is null");
Map<String, String> stringMap = new HashMap<String, String>(report
.getAllStringMetrics());
Map<String, Double> doubleMap = new HashMap<String, Double>(report
.getAllDoubleMetrics());
Map<String, Long> longMap = new HashMap<String, Long>(report
.getAllLongMetrics());
for (String k : report.getAttrs().keySet()) {
Type t = Attributes.getType(k);
// If there's nothing in the Attributes table, guess at String
// When the Attributes table goes away this won't be necessary.
if (t == null) {
t = Type.STRING;
}
switch (t) {
case DOUBLE:
doubleMap.put(k, Attributes.readDouble(report, k));
break;
case STRING:
stringMap.put(k, Attributes.readString(report, k));
break;
case LONG:
longMap.put(k, Attributes.readLong(report, k));
break;
default:
LOG.warn("Unknown type " + t);
}
}
return new ThriftFlumeReport(stringMap, longMap, doubleMap);
}
/**
* Thrift interface: returns a serializable report with given name or null if
* report doesn't exist
*/
@Override
public ThriftFlumeReport getReportByName(String reportName) throws TException {
ReportManager reportManager = ReportManager.get();
Map<String, Reportable> reports = reportManager.getReportables();
if (reports.containsKey(reportName)) {
return reportToThrift(reports.get(reportName).getMetrics());
}
return null;
}
/**
* Thrift interface: returns a map of reports in serializable form
*/
@Override
public Map<String, ThriftFlumeReport> getAllReports() throws TException {
Map<String, ThriftFlumeReport> retMap = new HashMap<String, ThriftFlumeReport>();
ReportManager reportManager = ReportManager.get();
Map<String, Reportable> reports = reportManager.getReportables();
for (Entry<String, Reportable> e : reports.entrySet()) {
ThriftFlumeReport report = reportToThrift(e.getValue().getMetrics());
retMap.put(e.getKey(), report);
}
return retMap;
}
/**
* Starts the Thrift server
*/
public void serve() throws TTransportException {
LOG.info("Starting ReportServer...");
TProcessor processor = new ThriftFlumeReportServer.Processor(this);
this.start(processor, port, "Flume Report Server");
LOG.info("ReportServer started on port " + port);
}
/**
* Stops the Thrift server
*/
public void stop() {
LOG.info("Stopping ReportServer...");
super.stop();
LOG.info("ReportServer stopped");
}
}