git.basschouten.com Git - openhab-addons.git/blob

   1 /**
   2  * Copyright (c) 2010-2021 Contributors to the openHAB project
   3  *
   4  * See the NOTICE file(s) distributed with this work for additional
   5  * information.
   6  *
   7  * This program and the accompanying materials are made available under the
   8  * terms of the Eclipse Public License 2.0 which is available at
   9  * http://www.eclipse.org/legal/epl-2.0
  10  *
  11  * SPDX-License-Identifier: EPL-2.0
  12  */
  13 package org.openhab.persistence.dynamodb.internal;
  14
  15 import org.slf4j.Logger;
  16 import org.slf4j.LoggerFactory;
  17
  18 import com.amazonaws.auth.AWSCredentials;
  19 import com.amazonaws.auth.AWSStaticCredentialsProvider;
  20 import com.amazonaws.regions.Regions;
  21 import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
  22 import com.amazonaws.services.dynamodbv2.AmazonDynamoDBClientBuilder;
  23 import com.amazonaws.services.dynamodbv2.document.DynamoDB;
  24
  25 /**
  26  * Shallow wrapper for Dynamo DB wrappers
  27  *
  28  * @author Sami Salonen - Initial contribution
  29  */
  30 public class DynamoDBClient {
  31     private final Logger logger = LoggerFactory.getLogger(DynamoDBClient.class);
  32     private DynamoDB dynamo;
  33     private AmazonDynamoDB client;
  34
  35     public DynamoDBClient(AWSCredentials credentials, Regions region) {
  36         client = AmazonDynamoDBClientBuilder.standard().withRegion(region)
  37                 .withCredentials(new AWSStaticCredentialsProvider(credentials)).build();
  38         dynamo = new DynamoDB(client);
  39     }
  40
  41     public DynamoDBClient(DynamoDBConfig clientConfig) {
  42         this(clientConfig.getCredentials(), clientConfig.getRegion());
  43     }
  44
  45     public AmazonDynamoDB getDynamoClient() {
  46         return client;
  47     }
  48
  49     public DynamoDB getDynamoDB() {
  50         return dynamo;
  51     }
  52
  53     public void shutdown() {
  54         dynamo.shutdown();
  55     }
  56
  57     public boolean checkConnection() {
  58         try {
  59             dynamo.listTables(1).firstPage();
  60         } catch (Exception e) {
  61             logger.warn("Got internal server error when trying to list tables: {}", e.getMessage());
  62             return false;
  63         }
  64         return true;
  65     }
  66 }