gerrit.onap Code Review - ccsdk/features.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP : ccsdk features
   4  * ================================================================================
   5  * Copyright (C) 2021 highstreet technologies GmbH Intellectual Property.
   6  * All rights reserved.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *     http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  *
  21  */
  22 package org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.query;
  23
  24 import java.util.ArrayList;
  25 import java.util.Arrays;
  26 import java.util.List;
  27 import java.util.Map;
  28 import org.eclipse.jdt.annotation.Nullable;
  29 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.database.SqlDBMapper;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.EntityInput;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.SortOrder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.Filter;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.FilterBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.FilterKey;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.Pagination;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.Sortorder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.entity.input.SortorderKey;
  38
  39 public class SelectQuery implements SqlQuery {
  40
  41     private static final long DEFAULT_PAGESIZE = 20;
  42     private static final long DEFAULT_PAGE = 1;
  43     private final String tableName;
  44     private final List<Filter> filters;
  45     private final List<String> sortExpressions;
  46     private long page;
  47     private long pageSize;
  48     private final List<String> fields;
  49     private final List<String> groups;
  50
  51     public SelectQuery(String tableName) {
  52         this(tableName, Arrays.asList("*"), null);
  53     }
  54
  55     public SelectQuery(String tableName, List<String> fields, String controllerId) {
  56         this.tableName = tableName;
  57         this.fields = fields;
  58         this.filters = new ArrayList<>();
  59         this.sortExpressions = new ArrayList<>();
  60         this.groups = new ArrayList<>();
  61         this.page = DEFAULT_PAGE;
  62         this.pageSize = DEFAULT_PAGESIZE;
  63         if (controllerId != null) {
  64             this.addFilter(SqlDBMapper.ODLID_DBCOL, controllerId);
  65         }
  66     }
  67
  68     public SelectQuery(String tableName, String field, String controllerId) {
  69         this(tableName, Arrays.asList(field), controllerId);
  70     }
  71
  72     public SelectQuery(String tableName, EntityInput input) {
  73         this(tableName, input, null);
  74     }
  75
  76     public SelectQuery(String tableName, EntityInput input, String controllerId) {
  77         this(tableName);
  78         Map<FilterKey, Filter> filter = input != null ? input.getFilter() : null;
  79         if (filter != null && filter.size() > 0) {
  80             for (Filter f : filter.values()) {
  81                 this.addFilter(f.getProperty(), f.getFiltervalue());
  82             }
  83         }
  84         if (controllerId != null) {
  85             this.addFilter(SqlDBMapper.ODLID_DBCOL, controllerId);
  86         }
  87
  88         Map<SortorderKey, Sortorder> so = input.getSortorder();
  89         if (so != null && so.size() > 0) {
  90             for (Sortorder s : so.values()) {
  91                 this.addSortOrder(s.getProperty(),
  92                         s.getSortorder() == SortOrder.Ascending ? "ASC" : "DESC");
  93             }
  94         }
  95         Pagination pagination = input.getPagination();
  96         if (pagination != null) {
  97             this.setPagination(pagination.getPage().longValue(), pagination.getSize().longValue());
  98         } else {
  99             this.setPagination(1, 30);
 100         }
 101
 102     }
 103
 104     public void addFilter(String property, String filtervalue) {
 105         this.filters.add(new FilterBuilder().setProperty(property).setFiltervalue(filtervalue).build());
 106
 107     }
 108
 109     public void addSortOrder(String col, String order) {
 110         this.sortExpressions.add(String.format("`%s` %s", col, order));
 111     }
 112
 113     public void setPagination(long page, long pageSize) {
 114         this.page = page;
 115         this.pageSize = pageSize;
 116     }
 117
 118     public void setPagination(@Nullable Pagination pagination) {
 119         long page = DEFAULT_PAGE;
 120         long pageSize = DEFAULT_PAGESIZE;
 121         if(pagination!=null) {
 122             if(pagination.getPage()!=null) {
 123                 page = pagination.getPage().longValue();
 124             }
 125             if(pagination.getSize()!=null) {
 126                 pageSize = pagination.getSize().longValue();
 127             }
 128         }
 129         this.setPagination(page, pageSize);
 130
 131     }
 132
 133     @Override
 134     public String toSql() {
 135         StringBuilder sb = new StringBuilder();
 136         if (this.fields.size() == 1 && this.fields.contains("*")) {
 137             sb.append(String.format("SELECT * FROM `%s`", this.tableName));
 138         } else {
 139             sb.append(String.format("SELECT `%s` FROM `%s`", String.join("`,`", this.fields), this.tableName));
 140         }
 141         sb.append(SqlQuery.getWhereExpression(this.filters));
 142         if (this.groups.size() > 0) {
 143             sb.append(String.format(" GROUP BY `%s`", String.join("`,`", this.groups)));
 144         }
 145         if (this.sortExpressions.size() > 0) {
 146             sb.append(" ORDER BY " + String.join(",", this.sortExpressions));
 147         }
 148         sb.append(String.format(" LIMIT %d,%d;", (this.page - 1) * this.pageSize, this.pageSize));
 149         return sb.toString();
 150     }
 151
 152     public long getPage() {
 153         return this.page;
 154     }
 155
 156     public long getPageSize() {
 157         return this.pageSize;
 158     }
 159
 160     public SelectQuery groupBy(String group) {
 161         this.groups.add(group);
 162         return this;
 163     }
 164
 165 }