com.symphony.oss.fugue.aws.metrics.AwsMetricManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fugue-aws Show documentation
There is a newer version: 0.3.0
/*
 *
 *
 * Copyright 2018 Symphony Communication Services, LLC.
 *
 * Licensed to The Symphony Software Foundation (SSF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF 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.symphony.oss.fugue.aws.metrics;

import java.util.Date;
import java.util.LinkedList;
import java.util.List;

import com.amazonaws.metrics.AwsSdkMetrics;
import com.amazonaws.services.cloudwatch.AmazonCloudWatch;
import com.amazonaws.services.cloudwatch.AmazonCloudWatchClientBuilder;
import com.amazonaws.services.cloudwatch.model.Dimension;
import com.amazonaws.services.cloudwatch.model.MetricDatum;
import com.amazonaws.services.cloudwatch.model.PutMetricDataRequest;
import com.amazonaws.services.cloudwatch.model.StandardUnit;
import com.symphony.oss.fugue.config.IConfiguration;
import com.symphony.oss.fugue.metrics.IMetricManager;
import com.symphony.oss.fugue.naming.INameFactory;
import com.symphony.oss.fugue.naming.NameFactory;
import com.symphony.oss.fugue.server.IFugueServer;

public class AwsMetricManager implements IMetricManager
{
  private final AmazonCloudWatch cwClient_;
  private String nameSpace_;
  private String tenantId_;
  
  public AwsMetricManager(IConfiguration config, INameFactory nameFactory)
  {
    cwClient_ =
        AmazonCloudWatchClientBuilder.defaultClient();
    
    nameSpace_ = nameFactory.getMultiTenantServiceName().toString();
    tenantId_ = config.getString("id/tenantId", null);
    
    AwsSdkMetrics.enableDefaultMetrics();

//  AwsSdkMetrics.setCredentialProvider(credentialsProvider);

    AwsSdkMetrics.setMetricNameSpace(nameSpace_);
  }

  @Override
  public void putMetric(long timestamp, int count)
  {
    List dimensions = new LinkedList<>();
    
    dimensions.add(new Dimension()
      .withName("MessageType")
      .withValue("PubSub"));
    
    if(tenantId_ != null)
      dimensions.add(new Dimension()
          .withName("Tenant")
          .withValue(tenantId_));
    
    dimensions.add(new Dimension()
      .withName("Instance")
      .withValue(IFugueServer.getInstanceId()));
    
    MetricDatum datum = new MetricDatum()
        .withMetricName("MessageCount")
        .withUnit(StandardUnit.Count)
        .withValue((double)count)
        .withDimensions(
            dimensions)
        .withTimestamp(new Date(timestamp))
        ;

    PutMetricDataRequest request = new PutMetricDataRequest()
        .withNamespace(nameSpace_)
        .withMetricData(datum);

    cwClient_.putMetricData(request);
  }
}