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

   1 /**
   2  * Copyright (c) 2010-2020 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.influxdb.dto.Query;
  21 import org.influxdb.querybuilder.Appender;
  22 import org.influxdb.querybuilder.BuiltQuery;
  23 import org.influxdb.querybuilder.Select;
  24 import org.influxdb.querybuilder.Where;
  25 import org.influxdb.querybuilder.clauses.SimpleClause;
  26 import org.openhab.core.persistence.FilterCriteria;
  27 import org.openhab.persistence.influxdb.internal.FilterCriteriaQueryCreator;
  28 import org.openhab.persistence.influxdb.internal.InfluxDBVersion;
  29
  30 /**
  31  * Implementation of {@link FilterCriteriaQueryCreator} for InfluxDB 1.0
  32  *
  33  * @author Joan Pujol Espinar - Initial contribution
  34  */
  35 @NonNullByDefault
  36 public class Influx1FilterCriteriaQueryCreatorImpl implements FilterCriteriaQueryCreator {
  37
  38     @Override
  39     public String createQuery(FilterCriteria criteria, String retentionPolicy) {
  40         final String tableName;
  41         boolean hasCriteriaName = criteria.getItemName() != null;
  42         if (hasCriteriaName) {
  43             tableName = criteria.getItemName();
  44         } else {
  45             tableName = "/.*/";
  46         }
  47
  48         Select select = select(COLUMN_VALUE_NAME_V1).fromRaw(null,
  49                 fullQualifiedTableName(retentionPolicy, tableName, hasCriteriaName));
  50
  51         Where where = select.where();
  52         if (criteria.getBeginDate() != null) {
  53             where = where.and(
  54                     BuiltQuery.QueryBuilder.gte(COLUMN_TIME_NAME_V1, criteria.getBeginDate().toInstant().toString()));
  55         }
  56         if (criteria.getEndDate() != null) {
  57             where = where.and(
  58                     BuiltQuery.QueryBuilder.lte(COLUMN_TIME_NAME_V1, criteria.getEndDate().toInstant().toString()));
  59         }
  60
  61         if (criteria.getState() != null && criteria.getOperator() != null) {
  62             where = where.and(new SimpleClause(COLUMN_VALUE_NAME_V1,
  63                     getOperationSymbol(criteria.getOperator(), InfluxDBVersion.V1),
  64                     stateToObject(criteria.getState())));
  65         }
  66
  67         if (criteria.getOrdering() == FilterCriteria.Ordering.DESCENDING) {
  68             select = select.orderBy(desc());
  69         } else if (criteria.getOrdering() == FilterCriteria.Ordering.ASCENDING) {
  70             select = select.orderBy(asc());
  71         }
  72
  73         if (criteria.getPageSize() != Integer.MAX_VALUE) {
  74             if (criteria.getPageNumber() != 0) {
  75                 select = select.limit(criteria.getPageSize(), criteria.getPageSize() * criteria.getPageNumber());
  76             } else {
  77                 select = select.limit(criteria.getPageSize());
  78             }
  79         }
  80
  81         final Query query = (Query) select;
  82         return query.getCommand();
  83     }
  84
  85     private String fullQualifiedTableName(String retentionPolicy, String tableName, boolean escapeTableName) {
  86         StringBuilder sb = new StringBuilder();
  87         Appender.appendName(retentionPolicy, sb);
  88         sb.append(".");
  89         if (escapeTableName) {
  90             Appender.appendName(tableName, sb);
  91         } else {
  92             sb.append(tableName);
  93         }
  94         return sb.toString();
  95     }
  96 }