com.github.sdmcraft.slingdynamo.impl.DynamoDBResourceProviderFactory.java Source code

Introduction

Here is the source code for com.github.sdmcraft.slingdynamo.impl.DynamoDBResourceProviderFactory.java

Source

/*
 * Copyright (c) 2013-14, Satya Deep Maheshwari. All rights reserved.
 *
 * The contents of this file are subject to the MIT License
 * You may not use this file except in compliance with the License.
 * A copy of the License is available at
 * http://opensource.org/licenses/MIT
 *
 * Copyright (c) 2013-2014 Satya Deep Maheshwari
 */
package com.github.sdmcraft.slingdynamo.impl;

import com.amazonaws.auth.AWSCredentialsProvider;

import com.amazonaws.regions.Region;
import com.amazonaws.regions.Regions;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDBClient;
import com.amazonaws.services.dynamodbv2.document.DynamoDB;

import org.apache.felix.scr.annotations.Activate;
import org.apache.felix.scr.annotations.Component;
import org.apache.felix.scr.annotations.Deactivate;
import org.apache.felix.scr.annotations.Properties;
import org.apache.felix.scr.annotations.Property;
import org.apache.felix.scr.annotations.Reference;
import org.apache.felix.scr.annotations.Service;

import org.apache.sling.api.SlingConstants;
import org.apache.sling.api.resource.LoginException;
import org.apache.sling.api.resource.ResourceProvider;
import org.apache.sling.api.resource.ResourceProviderFactory;
import org.apache.sling.commons.osgi.PropertiesUtil;

import org.osgi.framework.BundleContext;

import java.util.Map;

// TODO: Auto-generated Javadoc
/**
 * A factory for creating DynamoDBResourceProvider objects.
 */
@Component(label = "DynamoDB Resource Provider Factory", description = "Dynamo DB Resource Provider Factory", immediate = true, metatype = true)
@Service
@Properties({ @Property(name = "service.description", value = "Dynamo DB Resource Provider Factory"),
        @Property(name = "service.vendor", value = "com.github.sdmcraft"),
        @Property(name = ResourceProvider.ROOTS, value = DynamoDBConstants.DEFAULT_ROOT) })
public class DynamoDBResourceProviderFactory implements ResourceProviderFactory {
    /** The Constant PROP_REGION. */
    @Property
    private static final String PROP_REGION = "aws.region";
    @Property
    private static final String PROP_ENDPOINT = "aws.endpoint";

    /** The Constant PROP_RESOURCE_TYPE. */
    @Property
    private static final String PROP_RESOURCE_TYPE = SlingConstants.PROPERTY_RESOURCE_TYPE;

    /** The dynamo db client. */
    private AmazonDynamoDBClient dynamoDBClient;

    /** The dynamo db. */
    private DynamoDB dynamoDB;

    /** The resource type. */
    private String resourceType;

    /** The root. */
    private String root;
    private final String DEFAULT_ROOT = DynamoDBConstants.DEFAULT_ROOT;
    private String region;
    private String endpoint;
    @Reference
    AWSCredentialsProvider awsCredentialsProvider;

    /* (non-Javadoc)
     * @see org.apache.sling.api.resource.ResourceProviderFactory#getAdministrativeResourceProvider(java.util.Map)
     */
    public ResourceProvider getAdministrativeResourceProvider(Map<String, Object> arg0) throws LoginException {
        return new DynamoDBResourceProvider(root, dynamoDBClient, dynamoDB, resourceType);
    }

    /* (non-Javadoc)
     * @see org.apache.sling.api.resource.ResourceProviderFactory#getResourceProvider(java.util.Map)
     */
    public ResourceProvider getResourceProvider(Map<String, Object> arg0) throws LoginException {
        return new DynamoDBResourceProvider(root, dynamoDBClient, dynamoDB, resourceType);
    }

    /**
     * Activate.
     *
     * @param context the context
     * @param config the config
     */
    @Activate
    protected void activate(BundleContext context, Map<String, Object> config) {
        this.root = PropertiesUtil.toString(config.get(ResourceProvider.ROOTS), DEFAULT_ROOT);

        if ((this.root == null) || this.root.isEmpty()) {
            this.root = DEFAULT_ROOT;
        }

        this.resourceType = PropertiesUtil.toString(config.get(SlingConstants.PROPERTY_RESOURCE_TYPE),
                Constants.DEFAULT_GET_SERVLET);

        if ((this.resourceType == null) || this.resourceType.isEmpty()) {
            this.resourceType = Constants.DEFAULT_GET_SERVLET;
        }

        dynamoDBClient = new AmazonDynamoDBClient(awsCredentialsProvider.getCredentials());

        this.region = PropertiesUtil.toString(config.get(PROP_REGION), Constants.DEFAULT_REGION);

        if ((this.region != null) && !this.region.isEmpty()) {
            Region awsRegion = Region.getRegion(Regions.fromName(region));
            dynamoDBClient.setRegion(awsRegion);
        }

        this.endpoint = PropertiesUtil.toString(config.get(PROP_ENDPOINT), Constants.DEFAULT_ENDPOINT);

        if ((this.endpoint != null) && !this.endpoint.isEmpty()) {
            dynamoDBClient.setEndpoint(this.endpoint);
        }

        dynamoDB = new DynamoDB(dynamoDBClient);
    }

    /**
     * Dectivate.
     *
     * @param context the context
     * @param config the config
     */
    @Deactivate
    protected void dectivate(BundleContext context, Map<String, Object> config) {
        dynamoDBClient.shutdown();
    }
}