Igor Kulikov
7 years ago
132 changed files with 1546 additions and 2063 deletions
@ -0,0 +1,60 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2019 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.page; |
||||
|
|
||||
|
import com.fasterxml.jackson.annotation.JsonIgnore; |
||||
|
import lombok.Data; |
||||
|
|
||||
|
@Data |
||||
|
public class PageLink { |
||||
|
|
||||
|
private final String textSearch; |
||||
|
private final int pageSize; |
||||
|
private final int page; |
||||
|
private final SortOrder sortOrder; |
||||
|
|
||||
|
public PageLink(PageLink pageLink) { |
||||
|
this.pageSize = pageLink.getPageSize(); |
||||
|
this.page = pageLink.getPage(); |
||||
|
this.textSearch = pageLink.getTextSearch(); |
||||
|
this.sortOrder = pageLink.getSortOrder(); |
||||
|
} |
||||
|
|
||||
|
public PageLink(int pageSize) { |
||||
|
this(pageSize, 0); |
||||
|
} |
||||
|
|
||||
|
public PageLink(int pageSize, int page) { |
||||
|
this(pageSize, page, null, null); |
||||
|
} |
||||
|
|
||||
|
public PageLink(int pageSize, int page, String textSearch) { |
||||
|
this(pageSize, page, textSearch, null); |
||||
|
} |
||||
|
|
||||
|
public PageLink(int pageSize, int page, String textSearch, SortOrder sortOrder) { |
||||
|
this.pageSize = pageSize; |
||||
|
this.page = page; |
||||
|
this.textSearch = textSearch; |
||||
|
this.sortOrder = sortOrder; |
||||
|
} |
||||
|
|
||||
|
@JsonIgnore |
||||
|
public PageLink nextPageLink() { |
||||
|
return new PageLink(this.pageSize, this.page+1, this.textSearch, this.sortOrder); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -1,71 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2019 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.server.common.data.page; |
|
||||
|
|
||||
import java.util.List; |
|
||||
import java.util.UUID; |
|
||||
|
|
||||
import org.thingsboard.server.common.data.SearchTextBased; |
|
||||
import org.thingsboard.server.common.data.id.UUIDBased; |
|
||||
|
|
||||
import com.fasterxml.jackson.annotation.JsonCreator; |
|
||||
import com.fasterxml.jackson.annotation.JsonProperty; |
|
||||
|
|
||||
public class TextPageData<T extends SearchTextBased<? extends UUIDBased>> { |
|
||||
|
|
||||
private final List<T> data; |
|
||||
private final TextPageLink nextPageLink; |
|
||||
private final boolean hasNext; |
|
||||
|
|
||||
public TextPageData(List<T> data, TextPageLink pageLink) { |
|
||||
super(); |
|
||||
this.data = data; |
|
||||
int limit = pageLink.getLimit(); |
|
||||
if (data != null && data.size() == limit) { |
|
||||
int index = data.size()-1; |
|
||||
UUID idOffset = data.get(index).getId().getId(); |
|
||||
String textOffset = data.get(index).getSearchText(); |
|
||||
nextPageLink = new TextPageLink(limit, pageLink.getTextSearch(), idOffset, textOffset); |
|
||||
hasNext = true; |
|
||||
} else { |
|
||||
nextPageLink = null; |
|
||||
hasNext = false; |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@JsonCreator |
|
||||
public TextPageData(@JsonProperty("data") List<T> data, |
|
||||
@JsonProperty("nextPageLink") TextPageLink nextPageLink, |
|
||||
@JsonProperty("hasNext") boolean hasNext) { |
|
||||
this.data = data; |
|
||||
this.nextPageLink = nextPageLink; |
|
||||
this.hasNext = hasNext; |
|
||||
} |
|
||||
|
|
||||
public List<T> getData() { |
|
||||
return data; |
|
||||
} |
|
||||
|
|
||||
@JsonProperty("hasNext") |
|
||||
public boolean hasNext() { |
|
||||
return hasNext; |
|
||||
} |
|
||||
|
|
||||
public TextPageLink getNextPageLink() { |
|
||||
return nextPageLink; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,80 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2019 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.server.common.data.page; |
|
||||
|
|
||||
import com.fasterxml.jackson.annotation.JsonCreator; |
|
||||
import com.fasterxml.jackson.annotation.JsonProperty; |
|
||||
import lombok.Getter; |
|
||||
import lombok.ToString; |
|
||||
|
|
||||
import java.io.Serializable; |
|
||||
import java.util.Arrays; |
|
||||
import java.util.UUID; |
|
||||
|
|
||||
@ToString |
|
||||
public class TextPageLink extends BasePageLink implements Serializable { |
|
||||
|
|
||||
private static final long serialVersionUID = -4189954843653250480L; |
|
||||
|
|
||||
@Getter private final String textSearch; |
|
||||
@Getter private final String textSearchBound; |
|
||||
@Getter private final String textOffset; |
|
||||
|
|
||||
public TextPageLink(int limit) { |
|
||||
this(limit, null, null, null); |
|
||||
} |
|
||||
|
|
||||
public TextPageLink(int limit, String textSearch) { |
|
||||
this(limit, textSearch, null, null); |
|
||||
} |
|
||||
|
|
||||
public TextPageLink(int limit, String textSearch, UUID idOffset, String textOffset) { |
|
||||
super(limit, idOffset); |
|
||||
this.textSearch = textSearch != null ? textSearch.toLowerCase() : null; |
|
||||
this.textSearchBound = nextSequence(this.textSearch); |
|
||||
this.textOffset = textOffset != null ? textOffset.toLowerCase() : null; |
|
||||
} |
|
||||
|
|
||||
@JsonCreator |
|
||||
public TextPageLink(@JsonProperty("limit") int limit, |
|
||||
@JsonProperty("textSearch") String textSearch, |
|
||||
@JsonProperty("textSearchBound") String textSearchBound, |
|
||||
@JsonProperty("textOffset") String textOffset, |
|
||||
@JsonProperty("idOffset") UUID idOffset) { |
|
||||
super(limit, idOffset); |
|
||||
this.textSearch = textSearch; |
|
||||
this.textSearchBound = textSearchBound; |
|
||||
this.textOffset = textOffset; |
|
||||
this.idOffset = idOffset; |
|
||||
} |
|
||||
|
|
||||
private static String nextSequence(String input) { |
|
||||
if (input != null && input.length() > 0) { |
|
||||
char[] chars = input.toCharArray(); |
|
||||
int i = chars.length - 1; |
|
||||
while (i >= 0 && ++chars[i--] == Character.MIN_VALUE) ; |
|
||||
if (i == -1 && (chars.length == 0 || chars[0] == Character.MIN_VALUE)) { |
|
||||
char buf[] = Arrays.copyOf(input.toCharArray(), input.length() + 1); |
|
||||
buf[buf.length - 1] = Character.MIN_VALUE; |
|
||||
return new String(buf); |
|
||||
} |
|
||||
return new String(chars); |
|
||||
} else { |
|
||||
return null; |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,33 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2019 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.server.dao; |
|
||||
|
|
||||
import org.springframework.boot.autoconfigure.EnableAutoConfiguration; |
|
||||
import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration; |
|
||||
import org.springframework.boot.autoconfigure.jdbc.DataSourceTransactionManagerAutoConfiguration; |
|
||||
import org.springframework.boot.autoconfigure.orm.jpa.HibernateJpaAutoConfiguration; |
|
||||
import org.springframework.context.annotation.Configuration; |
|
||||
import org.thingsboard.server.dao.util.NoSqlDao; |
|
||||
|
|
||||
@Configuration |
|
||||
@EnableAutoConfiguration( |
|
||||
exclude = { |
|
||||
DataSourceAutoConfiguration.class, |
|
||||
DataSourceTransactionManagerAutoConfiguration.class, |
|
||||
HibernateJpaAutoConfiguration.class}) |
|
||||
@NoSqlDao |
|
||||
public class NoSqlDaoConfig { |
|
||||
} |
|
||||
@ -1,195 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2019 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.server.dao.nosql; |
|
||||
|
|
||||
import com.datastax.driver.core.ResultSet; |
|
||||
import com.datastax.driver.core.ResultSetFuture; |
|
||||
import com.datastax.driver.core.Statement; |
|
||||
import com.datastax.driver.core.querybuilder.QueryBuilder; |
|
||||
import com.datastax.driver.core.querybuilder.Select; |
|
||||
import com.datastax.driver.core.utils.UUIDs; |
|
||||
import com.datastax.driver.mapping.Mapper; |
|
||||
import com.datastax.driver.mapping.Result; |
|
||||
import com.google.common.base.Function; |
|
||||
import com.google.common.util.concurrent.Futures; |
|
||||
import com.google.common.util.concurrent.ListenableFuture; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.thingsboard.server.common.data.id.TenantId; |
|
||||
import org.thingsboard.server.dao.Dao; |
|
||||
import org.thingsboard.server.dao.DaoUtil; |
|
||||
import org.thingsboard.server.dao.model.BaseEntity; |
|
||||
import org.thingsboard.server.dao.model.ModelConstants; |
|
||||
import org.thingsboard.server.dao.model.wrapper.EntityResultSet; |
|
||||
|
|
||||
import javax.annotation.Nullable; |
|
||||
import java.util.Collections; |
|
||||
import java.util.List; |
|
||||
import java.util.UUID; |
|
||||
|
|
||||
import static com.datastax.driver.core.querybuilder.QueryBuilder.eq; |
|
||||
import static com.datastax.driver.core.querybuilder.QueryBuilder.select; |
|
||||
|
|
||||
@Slf4j |
|
||||
public abstract class CassandraAbstractModelDao<E extends BaseEntity<D>, D> extends CassandraAbstractDao implements Dao<D> { |
|
||||
|
|
||||
protected abstract Class<E> getColumnFamilyClass(); |
|
||||
|
|
||||
protected abstract String getColumnFamilyName(); |
|
||||
|
|
||||
protected E updateSearchTextIfPresent(E entity) { |
|
||||
return entity; |
|
||||
} |
|
||||
|
|
||||
protected Mapper<E> getMapper() { |
|
||||
return cluster.getMapper(getColumnFamilyClass()); |
|
||||
} |
|
||||
|
|
||||
protected List<E> findListByStatement(TenantId tenantId, Statement statement) { |
|
||||
List<E> list = Collections.emptyList(); |
|
||||
if (statement != null) { |
|
||||
statement.setConsistencyLevel(cluster.getDefaultReadConsistencyLevel()); |
|
||||
ResultSet resultSet = executeRead(tenantId, statement); |
|
||||
Result<E> result = getMapper().map(resultSet); |
|
||||
if (result != null) { |
|
||||
list = result.all(); |
|
||||
} |
|
||||
} |
|
||||
return list; |
|
||||
} |
|
||||
|
|
||||
protected ListenableFuture<List<D>> findListByStatementAsync(TenantId tenantId, Statement statement) { |
|
||||
if (statement != null) { |
|
||||
statement.setConsistencyLevel(cluster.getDefaultReadConsistencyLevel()); |
|
||||
ResultSetFuture resultSetFuture = executeAsyncRead(tenantId, statement); |
|
||||
return Futures.transform(resultSetFuture, new Function<ResultSet, List<D>>() { |
|
||||
@Nullable |
|
||||
@Override |
|
||||
public List<D> apply(@Nullable ResultSet resultSet) { |
|
||||
Result<E> result = getMapper().map(resultSet); |
|
||||
if (result != null) { |
|
||||
List<E> entities = result.all(); |
|
||||
return DaoUtil.convertDataList(entities); |
|
||||
} else { |
|
||||
return Collections.emptyList(); |
|
||||
} |
|
||||
} |
|
||||
}); |
|
||||
} |
|
||||
return Futures.immediateFuture(Collections.emptyList()); |
|
||||
} |
|
||||
|
|
||||
protected E findOneByStatement(TenantId tenantId, Statement statement) { |
|
||||
E object = null; |
|
||||
if (statement != null) { |
|
||||
statement.setConsistencyLevel(cluster.getDefaultReadConsistencyLevel()); |
|
||||
ResultSet resultSet = executeRead(tenantId, statement); |
|
||||
Result<E> result = getMapper().map(resultSet); |
|
||||
if (result != null) { |
|
||||
object = result.one(); |
|
||||
} |
|
||||
} |
|
||||
return object; |
|
||||
} |
|
||||
|
|
||||
protected ListenableFuture<D> findOneByStatementAsync(TenantId tenantId, Statement statement) { |
|
||||
if (statement != null) { |
|
||||
statement.setConsistencyLevel(cluster.getDefaultReadConsistencyLevel()); |
|
||||
ResultSetFuture resultSetFuture = executeAsyncRead(tenantId, statement); |
|
||||
return Futures.transform(resultSetFuture, new Function<ResultSet, D>() { |
|
||||
@Nullable |
|
||||
@Override |
|
||||
public D apply(@Nullable ResultSet resultSet) { |
|
||||
Result<E> result = getMapper().map(resultSet); |
|
||||
if (result != null) { |
|
||||
E entity = result.one(); |
|
||||
return DaoUtil.getData(entity); |
|
||||
} else { |
|
||||
return null; |
|
||||
} |
|
||||
} |
|
||||
}); |
|
||||
} |
|
||||
return Futures.immediateFuture(null); |
|
||||
} |
|
||||
|
|
||||
protected Statement getSaveQuery(E dto) { |
|
||||
return getMapper().saveQuery(dto); |
|
||||
} |
|
||||
|
|
||||
protected EntityResultSet<E> saveWithResult(TenantId tenantId, E entity) { |
|
||||
log.debug("Save entity {}", entity); |
|
||||
if (entity.getId() == null) { |
|
||||
entity.setId(UUIDs.timeBased()); |
|
||||
} else if (isDeleteOnSave()) { |
|
||||
removeById(tenantId, entity.getId()); |
|
||||
} |
|
||||
Statement saveStatement = getSaveQuery(entity); |
|
||||
saveStatement.setConsistencyLevel(cluster.getDefaultWriteConsistencyLevel()); |
|
||||
ResultSet resultSet = executeWrite(tenantId, saveStatement); |
|
||||
return new EntityResultSet<>(resultSet, entity); |
|
||||
} |
|
||||
|
|
||||
protected boolean isDeleteOnSave() { |
|
||||
return true; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public D save(TenantId tenantId, D domain) { |
|
||||
E entity; |
|
||||
try { |
|
||||
entity = getColumnFamilyClass().getConstructor(domain.getClass()).newInstance(domain); |
|
||||
} catch (Exception e) { |
|
||||
log.error("Can't create entity for domain object {}", domain, e); |
|
||||
throw new IllegalArgumentException("Can't create entity for domain object {" + domain + "}", e); |
|
||||
} |
|
||||
entity = updateSearchTextIfPresent(entity); |
|
||||
log.debug("Saving entity {}", entity); |
|
||||
entity = saveWithResult(tenantId, entity).getEntity(); |
|
||||
return DaoUtil.getData(entity); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public D findById(TenantId tenantId, UUID key) { |
|
||||
log.debug("Get entity by key {}", key); |
|
||||
Select.Where query = select().from(getColumnFamilyName()).where(eq(ModelConstants.ID_PROPERTY, key)); |
|
||||
log.trace("Execute query {}", query); |
|
||||
E entity = findOneByStatement(tenantId, query); |
|
||||
return DaoUtil.getData(entity); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public ListenableFuture<D> findByIdAsync(TenantId tenantId, UUID key) { |
|
||||
log.debug("Get entity by key {}", key); |
|
||||
Select.Where query = select().from(getColumnFamilyName()).where(eq(ModelConstants.ID_PROPERTY, key)); |
|
||||
log.trace("Execute query {}", query); |
|
||||
return findOneByStatementAsync(tenantId, query); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public boolean removeById(TenantId tenantId, UUID key) { |
|
||||
Statement delete = QueryBuilder.delete().all().from(getColumnFamilyName()).where(eq(ModelConstants.ID_PROPERTY, key)); |
|
||||
log.debug("Remove request: {}", delete.toString()); |
|
||||
return executeWrite(tenantId, delete).wasApplied(); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public List<D> find(TenantId tenantId) { |
|
||||
log.debug("Get all entities from column family {}", getColumnFamilyName()); |
|
||||
List<E> entities = findListByStatement(tenantId, QueryBuilder.select().all().from(getColumnFamilyName()).setConsistencyLevel(cluster.getDefaultReadConsistencyLevel())); |
|
||||
return DaoUtil.convertDataList(entities); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,83 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2019 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.server.dao.nosql; |
|
||||
|
|
||||
import com.datastax.driver.core.querybuilder.Clause; |
|
||||
import com.datastax.driver.core.querybuilder.QueryBuilder; |
|
||||
import com.datastax.driver.core.querybuilder.Select; |
|
||||
import com.datastax.driver.core.querybuilder.Select.Where; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.apache.commons.lang3.StringUtils; |
|
||||
import org.thingsboard.server.common.data.id.TenantId; |
|
||||
import org.thingsboard.server.common.data.page.TextPageLink; |
|
||||
import org.thingsboard.server.dao.model.ModelConstants; |
|
||||
import org.thingsboard.server.dao.model.SearchTextEntity; |
|
||||
|
|
||||
import java.util.List; |
|
||||
|
|
||||
import static com.datastax.driver.core.querybuilder.QueryBuilder.eq; |
|
||||
import static com.datastax.driver.core.querybuilder.QueryBuilder.select; |
|
||||
|
|
||||
@Slf4j |
|
||||
public abstract class CassandraAbstractSearchTextDao<E extends SearchTextEntity<D>, D> extends CassandraAbstractModelDao<E, D> { |
|
||||
|
|
||||
@Override |
|
||||
protected E updateSearchTextIfPresent(E entity) { |
|
||||
if (entity.getSearchTextSource() != null) { |
|
||||
entity.setSearchText(entity.getSearchTextSource().toLowerCase()); |
|
||||
} else { |
|
||||
log.trace("Entity [{}] has null SearchTextSource", entity); |
|
||||
} |
|
||||
return entity; |
|
||||
} |
|
||||
|
|
||||
protected List<E> findPageWithTextSearch(TenantId tenantId, String searchView, List<Clause> clauses, TextPageLink pageLink) { |
|
||||
Select select = select().from(searchView); |
|
||||
Where query = select.where(); |
|
||||
for (Clause clause : clauses) { |
|
||||
query.and(clause); |
|
||||
} |
|
||||
query.limit(pageLink.getLimit()); |
|
||||
if (!StringUtils.isEmpty(pageLink.getTextOffset())) { |
|
||||
query.and(eq(ModelConstants.SEARCH_TEXT_PROPERTY, pageLink.getTextOffset())); |
|
||||
query.and(QueryBuilder.lt(ModelConstants.ID_PROPERTY, pageLink.getIdOffset())); |
|
||||
List<E> result = findListByStatement(tenantId, query); |
|
||||
if (result.size() < pageLink.getLimit()) { |
|
||||
select = select().from(searchView); |
|
||||
query = select.where(); |
|
||||
for (Clause clause : clauses) { |
|
||||
query.and(clause); |
|
||||
} |
|
||||
query.and(QueryBuilder.gt(ModelConstants.SEARCH_TEXT_PROPERTY, pageLink.getTextOffset())); |
|
||||
if (!StringUtils.isEmpty(pageLink.getTextSearch())) { |
|
||||
query.and(QueryBuilder.lt(ModelConstants.SEARCH_TEXT_PROPERTY, pageLink.getTextSearchBound())); |
|
||||
} |
|
||||
int limit = pageLink.getLimit() - result.size(); |
|
||||
query.limit(limit); |
|
||||
result.addAll(findListByStatement(tenantId, query)); |
|
||||
} |
|
||||
return result; |
|
||||
} else if (!StringUtils.isEmpty(pageLink.getTextSearch())) { |
|
||||
query.and(QueryBuilder.gte(ModelConstants.SEARCH_TEXT_PROPERTY, pageLink.getTextSearch())); |
|
||||
query.and(QueryBuilder.lt(ModelConstants.SEARCH_TEXT_PROPERTY, pageLink.getTextSearchBound())); |
|
||||
return findListByStatement(tenantId, query); |
|
||||
} else { |
|
||||
return findListByStatement(tenantId, query); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
|
|
||||
} |
|
||||
@ -1,107 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2019 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.server.dao.nosql; |
|
||||
|
|
||||
import com.datastax.driver.core.querybuilder.Clause; |
|
||||
import com.datastax.driver.core.querybuilder.Ordering; |
|
||||
import com.datastax.driver.core.querybuilder.QueryBuilder; |
|
||||
import com.datastax.driver.core.querybuilder.Select; |
|
||||
import com.datastax.driver.core.querybuilder.Select.Where; |
|
||||
import com.datastax.driver.core.utils.UUIDs; |
|
||||
import org.thingsboard.server.common.data.id.TenantId; |
|
||||
import org.thingsboard.server.common.data.page.TimePageLink; |
|
||||
import org.thingsboard.server.dao.model.BaseEntity; |
|
||||
import org.thingsboard.server.dao.model.ModelConstants; |
|
||||
|
|
||||
import java.util.ArrayList; |
|
||||
import java.util.Collections; |
|
||||
import java.util.List; |
|
||||
import java.util.UUID; |
|
||||
|
|
||||
import static com.datastax.driver.core.querybuilder.QueryBuilder.select; |
|
||||
|
|
||||
public abstract class CassandraAbstractSearchTimeDao<E extends BaseEntity<D>, D> extends CassandraAbstractModelDao<E, D> { |
|
||||
|
|
||||
|
|
||||
protected List<E> findPageWithTimeSearch(TenantId tenantId, String searchView, List<Clause> clauses, TimePageLink pageLink) { |
|
||||
return findPageWithTimeSearch(tenantId, searchView, clauses, Collections.emptyList(), pageLink); |
|
||||
} |
|
||||
|
|
||||
protected List<E> findPageWithTimeSearch(TenantId tenantId, String searchView, List<Clause> clauses, Ordering ordering, TimePageLink pageLink) { |
|
||||
return findPageWithTimeSearch(tenantId, searchView, clauses, Collections.singletonList(ordering), pageLink); |
|
||||
} |
|
||||
|
|
||||
protected List<E> findPageWithTimeSearch(TenantId tenantId, String searchView, List<Clause> clauses, List<Ordering> topLevelOrderings, TimePageLink pageLink) { |
|
||||
return findPageWithTimeSearch(tenantId, searchView, clauses, topLevelOrderings, pageLink, ModelConstants.ID_PROPERTY); |
|
||||
} |
|
||||
|
|
||||
protected List<E> findPageWithTimeSearch(TenantId tenantId, String searchView, List<Clause> clauses, TimePageLink pageLink, String idColumn) { |
|
||||
return findPageWithTimeSearch(tenantId, searchView, clauses, Collections.emptyList(), pageLink, idColumn); |
|
||||
} |
|
||||
|
|
||||
protected List<E> findPageWithTimeSearch(TenantId tenantId, String searchView, List<Clause> clauses, List<Ordering> topLevelOrderings, TimePageLink pageLink, String idColumn) { |
|
||||
return findListByStatement(tenantId, buildQuery(searchView, clauses, topLevelOrderings, pageLink, idColumn)); |
|
||||
} |
|
||||
|
|
||||
public static Where buildQuery(String searchView, List<Clause> clauses, TimePageLink pageLink, String idColumn) { |
|
||||
return buildQuery(searchView, clauses, Collections.emptyList(), pageLink, idColumn); |
|
||||
} |
|
||||
|
|
||||
public static Where buildQuery(String searchView, List<Clause> clauses, Ordering order, TimePageLink pageLink, String idColumn) { |
|
||||
return buildQuery(searchView, clauses, Collections.singletonList(order), pageLink, idColumn); |
|
||||
} |
|
||||
|
|
||||
public static Where buildQuery(String searchView, List<Clause> clauses, List<Ordering> topLevelOrderings, TimePageLink pageLink, String idColumn) { |
|
||||
Select select = select().from(searchView); |
|
||||
Where query = select.where(); |
|
||||
for (Clause clause : clauses) { |
|
||||
query.and(clause); |
|
||||
} |
|
||||
query.limit(pageLink.getLimit()); |
|
||||
if (pageLink.isAscOrder()) { |
|
||||
if (pageLink.getIdOffset() != null) { |
|
||||
query.and(QueryBuilder.gt(idColumn, pageLink.getIdOffset())); |
|
||||
} else if (pageLink.getStartTime() != null) { |
|
||||
final UUID startOf = UUIDs.startOf(pageLink.getStartTime()); |
|
||||
query.and(QueryBuilder.gte(idColumn, startOf)); |
|
||||
} |
|
||||
if (pageLink.getEndTime() != null) { |
|
||||
final UUID endOf = UUIDs.endOf(pageLink.getEndTime()); |
|
||||
query.and(QueryBuilder.lte(idColumn, endOf)); |
|
||||
} |
|
||||
} else { |
|
||||
if (pageLink.getIdOffset() != null) { |
|
||||
query.and(QueryBuilder.lt(idColumn, pageLink.getIdOffset())); |
|
||||
} else if (pageLink.getEndTime() != null) { |
|
||||
final UUID endOf = UUIDs.endOf(pageLink.getEndTime()); |
|
||||
query.and(QueryBuilder.lte(idColumn, endOf)); |
|
||||
} |
|
||||
if (pageLink.getStartTime() != null) { |
|
||||
final UUID startOf = UUIDs.startOf(pageLink.getStartTime()); |
|
||||
query.and(QueryBuilder.gte(idColumn, startOf)); |
|
||||
} |
|
||||
} |
|
||||
List<Ordering> orderings = new ArrayList<>(topLevelOrderings); |
|
||||
if (pageLink.isAscOrder()) { |
|
||||
orderings.add(QueryBuilder.asc(idColumn)); |
|
||||
} else { |
|
||||
orderings.add(QueryBuilder.desc(idColumn)); |
|
||||
} |
|
||||
query.orderBy(orderings.toArray(new Ordering[orderings.size()])); |
|
||||
return query; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue