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

   1 /**
   2  * Copyright (c) 2010-2023 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.binding.dbquery.internal.dbimpl.influx2;
  14
  15 import java.util.ArrayList;
  16 import java.util.List;
  17 import java.util.concurrent.CompletableFuture;
  18 import java.util.concurrent.ExecutorService;
  19 import java.util.concurrent.Executors;
  20
  21 import org.eclipse.jdt.annotation.NonNullByDefault;
  22 import org.openhab.binding.dbquery.internal.config.InfluxDB2BridgeConfiguration;
  23 import org.openhab.binding.dbquery.internal.domain.Database;
  24 import org.openhab.binding.dbquery.internal.domain.Query;
  25 import org.openhab.binding.dbquery.internal.domain.QueryFactory;
  26 import org.openhab.binding.dbquery.internal.domain.QueryResult;
  27 import org.openhab.binding.dbquery.internal.error.DatabaseException;
  28 import org.slf4j.Logger;
  29 import org.slf4j.LoggerFactory;
  30
  31 import com.influxdb.query.FluxRecord;
  32
  33 /**
  34  * Influx2 implementation of {@link Database}
  35  *
  36  * @author Joan Pujol - Initial contribution
  37  */
  38 @NonNullByDefault
  39 public class Influx2Database implements Database {
  40     private final Logger logger = LoggerFactory.getLogger(Influx2Database.class);
  41     private final ExecutorService executors;
  42     private final InfluxDB2BridgeConfiguration config;
  43     private final InfluxDBClientFacade client;
  44     private final QueryFactory queryFactory;
  45
  46     public Influx2Database(InfluxDB2BridgeConfiguration config, InfluxDBClientFacade influxDBClientFacade) {
  47         this.config = config;
  48         this.client = influxDBClientFacade;
  49         executors = Executors.newSingleThreadScheduledExecutor();
  50         queryFactory = new Influx2QueryFactory();
  51     }
  52
  53     @Override
  54     public boolean isConnected() {
  55         return client.isConnected();
  56     }
  57
  58     @Override
  59     public CompletableFuture<Boolean> connect() {
  60         return CompletableFuture.supplyAsync(() -> {
  61             synchronized (Influx2Database.this) {
  62                 return client.connect();
  63             }
  64         }, executors);
  65     }
  66
  67     @Override
  68     public CompletableFuture<Boolean> disconnect() {
  69         return CompletableFuture.supplyAsync(() -> {
  70             synchronized (Influx2Database.this) {
  71                 return client.disconnect();
  72             }
  73         }, executors);
  74     }
  75
  76     @Override
  77     public QueryFactory queryFactory() throws DatabaseException {
  78         return queryFactory;
  79     }
  80
  81     @Override
  82     public CompletableFuture<QueryResult> executeQuery(Query query) {
  83         try {
  84             if (query instanceof Influx2QueryFactory.Influx2Query) {
  85                 Influx2QueryFactory.Influx2Query influxQuery = (Influx2QueryFactory.Influx2Query) query;
  86
  87                 CompletableFuture<QueryResult> asyncResult = new CompletableFuture<>();
  88                 List<FluxRecord> records = new ArrayList<>();
  89                 client.query(influxQuery.getQuery(), (cancellable, record) -> { // onNext
  90                     records.add(record);
  91                 }, error -> { // onError
  92                     logger.warn("Error executing query {}", query, error);
  93                     asyncResult.complete(QueryResult.ofIncorrectResult("Error executing query"));
  94                 }, () -> { // onComplete
  95                     asyncResult.complete(new Influx2QueryResultExtractor().apply(records));
  96                 });
  97                 return asyncResult;
  98             } else {
  99                 return CompletableFuture
 100                         .completedFuture(QueryResult.ofIncorrectResult("Unnexpected query type " + query));
 101             }
 102         } catch (RuntimeException e) {
 103             return CompletableFuture.failedFuture(e);
 104         }
 105     }
 106
 107     @Override
 108     public String toString() {
 109         return "Influx2Database{config=" + config + '}';
 110     }
 111 }