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