com.optimalbi.Services.LocalRDSService.java Source code

Introduction

Here is the source code for com.optimalbi.Services.LocalRDSService.java

Source

/*
   Copyright 2015 OptimalBI
    
   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.
 */

package com.optimalbi.Services;

import com.amazonaws.regions.Region;
import com.amazonaws.services.ec2.model.DescribeTagsResult;
import com.amazonaws.services.ec2.model.TagDescription;
import com.amazonaws.services.rds.AmazonRDS;
import com.amazonaws.services.rds.AmazonRDSClient;
import com.amazonaws.services.rds.model.*;
import com.optimalbi.Controller.Containers.AmazonCredentials;
import com.optimalbi.SimpleLog.*;
import javafx.scene.control.Label;
import javafx.scene.layout.VBox;

import java.util.*;
import java.util.stream.Collectors;

/**
 * Created by Timothy Gray(timg) on 3/12/2014.
 * Version: 0.0.2
 */
public class LocalRDSService extends AmazonService {
    private final Region region;
    private DBInstance thisService;
    private Map<String, Double> pricing = null;
    private String tagString;

    public LocalRDSService(String id, AmazonCredentials credentials, Region region, DBInstance instance,
            Logger logger) {
        super(id, credentials, logger);
        this.thisService = instance;
        this.region = region;
        tagString = Service.getTagsString(this);
    }

    public void refreshInstance() {
        AmazonRDSClient rds = new AmazonRDSClient(getCredentials().getCredentials());
        rds.setRegion(region);

        DescribeDBInstancesResult result = rds.describeDBInstances();
        List<DBInstance> instances = result.getDBInstances();
        DBInstance tempInstance = null;
        for (DBInstance d : instances) {
            if (d.getDBInstanceIdentifier().equalsIgnoreCase(thisService.getDBInstanceIdentifier())) {
                tempInstance = d;
            }
        }
        if (tempInstance != null) {
            thisService = tempInstance;
        } else {
            getLogger().error("Failed to refresh " + this.serviceName());
        }
    }

    public String serviceState() {
        return thisService.getDBInstanceStatus();
    }

    public String serviceName() {
        return thisService.getDBInstanceIdentifier();
    }

    public String serviceSize() {
        return thisService.getDBInstanceClass();
    }

    public double servicePrice() {
        if (pricing != null) {
            if (pricing.containsKey(this.serviceSize())) {
                return pricing.get(this.serviceSize());
            }
        }
        return 0;
    }

    public Map<String, String> getTags() {
        AmazonRDS rds = new AmazonRDSClient(this.getCredentials().getCredentials());
        String arn = thisService.getTdeCredentialArn();
        ListTagsForResourceResult tagsList = rds
                .listTagsForResource(new ListTagsForResourceRequest().withResourceName(arn));

        List<Tag> tagList = tagsList.getTagList();
        Map<String, String> tagMap = new HashMap<>();
        for (Tag t : tagList) {
            tagMap.put(t.getKey(), t.getValue());
        }
        return tagMap;
    }

    @Override
    public String getTagsString() {
        return tagString;
    }

    public Region serviceRegion() {
        return region;
    }

    public void attachPricing(Map<String, Double> pricing) {
        this.pricing = pricing;
    }

    public Map<String, Double> getPricing() {
        return pricing;
    }

    public String serviceType() {
        return "RDS";
    }

}