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

   1 /**
   2  * Copyright (c) 2010-2022 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.influxdb.internal.influx1;
  14
  15 import static org.influxdb.querybuilder.BuiltQuery.QueryBuilder.*;
  16 import static org.openhab.persistence.influxdb.internal.InfluxDBConstants.*;
  17 import static org.openhab.persistence.influxdb.internal.InfluxDBStateConvertUtils.stateToObject;
  18
  19 import org.eclipse.jdt.annotation.NonNullByDefault;
  20 import org.eclipse.jdt.annotation.Nullable;
  21 import org.influxdb.dto.Query;
  22 import org.influxdb.querybuilder.Appender;
  23 import org.influxdb.querybuilder.BuiltQuery;
  24 import org.influxdb.querybuilder.Select;
  25 import org.influxdb.querybuilder.Where;
  26 import org.influxdb.querybuilder.clauses.SimpleClause;
  27 import org.openhab.core.items.MetadataRegistry;
  28 import org.openhab.core.persistence.FilterCriteria;
  29 import org.openhab.persistence.influxdb.internal.FilterCriteriaQueryCreator;
  30 import org.openhab.persistence.influxdb.internal.InfluxDBConfiguration;
  31 import org.openhab.persistence.influxdb.internal.InfluxDBMetadataUtils;
  32 import org.openhab.persistence.influxdb.internal.InfluxDBVersion;
  33
  34 /**
  35  * Implementation of {@link FilterCriteriaQueryCreator} for InfluxDB 1.0
  36  *
  37  * @author Joan Pujol Espinar - Initial contribution
  38  */
  39 @NonNullByDefault
  40 public class Influx1FilterCriteriaQueryCreatorImpl implements FilterCriteriaQueryCreator {
  41
  42     private InfluxDBConfiguration configuration;
  43     private MetadataRegistry metadataRegistry;
  44
  45     public Influx1FilterCriteriaQueryCreatorImpl(InfluxDBConfiguration configuration,
  46             MetadataRegistry metadataRegistry) {
  47         this.configuration = configuration;
  48         this.metadataRegistry = metadataRegistry;
  49     }
  50
  51     @Override
  52     public String createQuery(FilterCriteria criteria, String retentionPolicy) {
  53         final String tableName;
  54         final String itemName = criteria.getItemName();
  55         boolean hasCriteriaName = itemName != null;
  56
  57         tableName = calculateTableName(itemName);
  58
  59         Select select = select().column("\"" + COLUMN_VALUE_NAME_V1 + "\"::field")
  60                 .column("\"" + TAG_ITEM_NAME + "\"::tag")
  61                 .fromRaw(null, fullQualifiedTableName(retentionPolicy, tableName, hasCriteriaName));
  62
  63         Where where = select.where();
  64
  65         if (itemName != null && !tableName.equals(itemName)) {
  66             where = where.and(BuiltQuery.QueryBuilder.eq(TAG_ITEM_NAME, itemName));
  67         }
  68
  69         if (criteria.getBeginDate() != null) {
  70             where = where.and(
  71                     BuiltQuery.QueryBuilder.gte(COLUMN_TIME_NAME_V1, criteria.getBeginDate().toInstant().toString()));
  72         }
  73         if (criteria.getEndDate() != null) {
  74             where = where.and(
  75                     BuiltQuery.QueryBuilder.lte(COLUMN_TIME_NAME_V1, criteria.getEndDate().toInstant().toString()));
  76         }
  77
  78         if (criteria.getState() != null && criteria.getOperator() != null) {
  79             where = where.and(new SimpleClause(COLUMN_VALUE_NAME_V1,
  80                     getOperationSymbol(criteria.getOperator(), InfluxDBVersion.V1),
  81                     stateToObject(criteria.getState())));
  82         }
  83
  84         if (criteria.getOrdering() == FilterCriteria.Ordering.DESCENDING) {
  85             select = select.orderBy(desc());
  86         } else if (criteria.getOrdering() == FilterCriteria.Ordering.ASCENDING) {
  87             select = select.orderBy(asc());
  88         }
  89
  90         if (criteria.getPageSize() != Integer.MAX_VALUE) {
  91             if (criteria.getPageNumber() != 0) {
  92                 select = select.limit(criteria.getPageSize(), criteria.getPageSize() * criteria.getPageNumber());
  93             } else {
  94                 select = select.limit(criteria.getPageSize());
  95             }
  96         }
  97
  98         final Query query = (Query) select;
  99         return query.getCommand();
 100     }
 101
 102     private String calculateTableName(@Nullable String itemName) {
 103         if (itemName == null) {
 104             return "/.*/";
 105         }
 106
 107         String name = itemName;
 108
 109         name = InfluxDBMetadataUtils.calculateMeasurementNameFromMetadataIfPresent(metadataRegistry, name, itemName);
 110
 111         if (configuration.isReplaceUnderscore()) {
 112             name = name.replace('_', '.');
 113         }
 114
 115         return name;
 116     }
 117
 118     private String fullQualifiedTableName(String retentionPolicy, String tableName, boolean escapeTableName) {
 119         StringBuilder sb = new StringBuilder();
 120         Appender.appendName(retentionPolicy, sb);
 121         sb.append(".");
 122         if (escapeTableName) {
 123             Appender.appendName(tableName, sb);
 124         } else {
 125             sb.append(tableName);
 126         }
 127         return sb.toString();
 128     }
 129 }