volodymyr-babak
10 years ago
11 changed files with 483 additions and 37 deletions
@ -0,0 +1,42 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.model.sql; |
||||
|
|
||||
|
import lombok.AllArgsConstructor; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
|
||||
|
import java.io.Serializable; |
||||
|
import java.util.UUID; |
||||
|
|
||||
|
@AllArgsConstructor |
||||
|
public class RelationCompositeKey implements Serializable { |
||||
|
|
||||
|
private UUID fromId; |
||||
|
private String fromType; |
||||
|
private UUID toId; |
||||
|
private String toType; |
||||
|
private String relationTypeGroup; |
||||
|
private String relationType; |
||||
|
|
||||
|
public RelationCompositeKey(EntityRelation relation) { |
||||
|
this.fromId = relation.getFrom().getId(); |
||||
|
this.fromType = relation.getFrom().getEntityType().name(); |
||||
|
this.toId = relation.getTo().getId(); |
||||
|
this.toType = relation.getTo().getEntityType().name(); |
||||
|
this.relationType = relation.getType(); |
||||
|
this.relationTypeGroup = relation.getTypeGroup().name(); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,183 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.model.sql; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import lombok.Data; |
||||
|
import org.hibernate.annotations.Type; |
||||
|
import org.thingsboard.server.common.data.id.EntityIdFactory; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
import org.thingsboard.server.common.data.relation.RelationTypeGroup; |
||||
|
import org.thingsboard.server.dao.model.ToData; |
||||
|
|
||||
|
import javax.persistence.*; |
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.thingsboard.server.dao.model.ModelConstants.*; |
||||
|
|
||||
|
@Data |
||||
|
@Entity |
||||
|
@Table(name = RELATION_COLUMN_FAMILY_NAME) |
||||
|
@IdClass(RelationCompositeKey.class) |
||||
|
public final class RelationEntity implements ToData<EntityRelation> { |
||||
|
|
||||
|
@Transient |
||||
|
private static final long serialVersionUID = -4089175869616037592L; |
||||
|
|
||||
|
@Id |
||||
|
@Column(name = RELATION_FROM_ID_PROPERTY) |
||||
|
private UUID fromId; |
||||
|
|
||||
|
@Id |
||||
|
@Column(name = RELATION_FROM_TYPE_PROPERTY) |
||||
|
private String fromType; |
||||
|
|
||||
|
@Id |
||||
|
@Column(name = RELATION_TO_ID_PROPERTY) |
||||
|
private UUID toId; |
||||
|
|
||||
|
@Id |
||||
|
@Column(name = RELATION_TO_TYPE_PROPERTY) |
||||
|
private String toType; |
||||
|
|
||||
|
@Id |
||||
|
@Column(name = RELATION_TYPE_GROUP_PROPERTY) |
||||
|
private String relationTypeGroup; |
||||
|
|
||||
|
@Id |
||||
|
@Column(name = RELATION_TYPE_PROPERTY) |
||||
|
private String relationType; |
||||
|
|
||||
|
@Type(type = "jsonb") |
||||
|
@Column(name = ADDITIONAL_INFO_PROPERTY, columnDefinition = "jsonb") |
||||
|
private JsonNode additionalInfo; |
||||
|
|
||||
|
public RelationEntity() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public RelationEntity(EntityRelation relation) { |
||||
|
if (relation.getTo() != null) { |
||||
|
this.toId = relation.getTo().getId(); |
||||
|
this.toType = relation.getTo().getEntityType().name(); |
||||
|
} |
||||
|
if (relation.getFrom() != null) { |
||||
|
this.fromId = relation.getFrom().getId(); |
||||
|
this.fromType = relation.getFrom().getEntityType().name(); |
||||
|
} |
||||
|
this.relationType = relation.getType(); |
||||
|
this.relationTypeGroup = relation.getTypeGroup().name(); |
||||
|
this.additionalInfo = relation.getAdditionalInfo(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public int hashCode() { |
||||
|
final int prime = 31; |
||||
|
int result = 1; |
||||
|
result = prime * result + ((additionalInfo == null) ? 0 : additionalInfo.hashCode()); |
||||
|
result = prime * result + ((toId == null) ? 0 : toId.hashCode()); |
||||
|
result = prime * result + ((toType == null) ? 0 : toType.hashCode()); |
||||
|
result = prime * result + ((fromId == null) ? 0 : fromId.hashCode()); |
||||
|
result = prime * result + ((fromType == null) ? 0 : fromType.hashCode()); |
||||
|
result = prime * result + ((relationType == null) ? 0 : relationType.hashCode()); |
||||
|
result = prime * result + ((relationTypeGroup == null) ? 0 : relationTypeGroup.hashCode()); |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean equals(Object obj) { |
||||
|
if (this == obj) |
||||
|
return true; |
||||
|
if (obj == null) |
||||
|
return false; |
||||
|
if (getClass() != obj.getClass()) |
||||
|
return false; |
||||
|
RelationEntity other = (RelationEntity) obj; |
||||
|
if (additionalInfo == null) { |
||||
|
if (other.additionalInfo != null) |
||||
|
return false; |
||||
|
} else if (!additionalInfo.equals(other.additionalInfo)) |
||||
|
return false; |
||||
|
if (toId == null) { |
||||
|
if (other.toId != null) |
||||
|
return false; |
||||
|
} else if (!toId.equals(other.toId)) |
||||
|
return false; |
||||
|
if (fromId == null) { |
||||
|
if (other.fromId != null) |
||||
|
return false; |
||||
|
} else if (!fromId.equals(other.fromId)) |
||||
|
return false; |
||||
|
if (toType == null) { |
||||
|
if (other.toType != null) |
||||
|
return false; |
||||
|
} else if (!toType.equals(other.toType)) |
||||
|
return false; |
||||
|
if (fromType == null) { |
||||
|
if (other.fromType != null) |
||||
|
return false; |
||||
|
} else if (!fromType.equals(other.fromType)) |
||||
|
return false; |
||||
|
if (relationType == null) { |
||||
|
if (other.relationType != null) |
||||
|
return false; |
||||
|
} else if (!relationType.equals(other.relationType)) |
||||
|
return false; |
||||
|
if (relationTypeGroup == null) { |
||||
|
if (other.relationTypeGroup != null) |
||||
|
return false; |
||||
|
} else if (!relationTypeGroup.equals(other.relationTypeGroup)) |
||||
|
return false; |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public String toString() { |
||||
|
StringBuilder builder = new StringBuilder(); |
||||
|
builder.append("AssetEntity [toId="); |
||||
|
builder.append(toId); |
||||
|
builder.append(", toType="); |
||||
|
builder.append(toType); |
||||
|
builder.append(", fromId="); |
||||
|
builder.append(fromId); |
||||
|
builder.append(", fromType="); |
||||
|
builder.append(fromType); |
||||
|
builder.append(", relationType="); |
||||
|
builder.append(relationType); |
||||
|
builder.append(", relationTypeGroup="); |
||||
|
builder.append(relationTypeGroup); |
||||
|
builder.append(", additionalInfo="); |
||||
|
builder.append(additionalInfo); |
||||
|
builder.append("]"); |
||||
|
return builder.toString(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public EntityRelation toData() { |
||||
|
EntityRelation relation = new EntityRelation(); |
||||
|
if (toId != null && toType != null) { |
||||
|
relation.setTo(EntityIdFactory.getByTypeAndUuid(toType, toId)); |
||||
|
} |
||||
|
if (fromId != null && fromType != null) { |
||||
|
relation.setFrom(EntityIdFactory.getByTypeAndUuid(fromType, fromId)); |
||||
|
} |
||||
|
relation.setType(relationType); |
||||
|
relation.setTypeGroup(RelationTypeGroup.valueOf(relationTypeGroup)); |
||||
|
relation.setAdditionalInfo(additionalInfo); |
||||
|
return relation; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,23 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.model.sql; |
||||
|
|
||||
|
public interface TenantDeviceTypeProjection { |
||||
|
|
||||
|
String getTenantId(); |
||||
|
|
||||
|
String getType(); |
||||
|
} |
||||
@ -0,0 +1,107 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2017 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.sql.relation; |
||||
|
|
||||
|
import com.datastax.driver.core.querybuilder.QueryBuilder; |
||||
|
import com.datastax.driver.core.querybuilder.Select; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.data.jpa.repository.Query; |
||||
|
import org.springframework.data.repository.CrudRepository; |
||||
|
import org.springframework.data.repository.query.Param; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.page.TimePageLink; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
import org.thingsboard.server.common.data.relation.RelationTypeGroup; |
||||
|
import org.thingsboard.server.dao.CassandraAbstractSearchTimeDao; |
||||
|
import org.thingsboard.server.dao.model.ModelConstants; |
||||
|
import org.thingsboard.server.dao.model.sql.RelationCompositeKey; |
||||
|
import org.thingsboard.server.dao.model.sql.RelationEntity; |
||||
|
|
||||
|
import java.util.Arrays; |
||||
|
import java.util.List; |
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static com.datastax.driver.core.querybuilder.QueryBuilder.eq; |
||||
|
|
||||
|
@ConditionalOnProperty(prefix = "sql", value = "enabled", havingValue = "true") |
||||
|
public interface RelationRepository extends CrudRepository<RelationEntity, RelationCompositeKey> { |
||||
|
|
||||
|
List<RelationEntity> findAllByFromIdAndFromTypeAndRelationTypeGroup(UUID fromId, |
||||
|
String fromType, |
||||
|
String relationTypeGroup); |
||||
|
|
||||
|
List<RelationEntity> findAllByFromIdAndFromTypeAndRelationTypeAndRelationTypeGroup(UUID fromId, |
||||
|
String fromType, |
||||
|
String relationType, |
||||
|
String relationTypeGroup); |
||||
|
|
||||
|
List<RelationEntity> findAllByToIdAndToTypeAndRelationTypeGroup(UUID toId, |
||||
|
String toType, |
||||
|
String relationTypeGroup); |
||||
|
|
||||
|
List<RelationEntity> findAllByToIdAndToTypeAndRelationTypeAndRelationTypeGroup(UUID toId, |
||||
|
String toType, |
||||
|
String relationType, |
||||
|
String relationTypeGroup); |
||||
|
|
||||
|
List<RelationEntity> findAllByFromIdAndFromType(UUID fromId, |
||||
|
String fromType); |
||||
|
|
||||
|
@Query(nativeQuery = true, value = "SELECT * FROM DEVICE WHERE TENANT_ID = :tenantId " + |
||||
|
"AND CUSTOMER_ID = :customerId " + |
||||
|
"AND LOWER(SEARCH_TEXT) LIKE LOWER(CONCAT(:searchText, '%')) " + |
||||
|
"AND ID > :idOffset ORDER BY ID LIMIT :limit") |
||||
|
List<RelationEntity> findRelations(@Param("fromId") UUID fromId, |
||||
|
@Param("fromType") String fromType, |
||||
|
@Param("toType") String toType, |
||||
|
@Param("relationType") String relationType, |
||||
|
@Param("relationTypeGroup") String relationTypeGroup, |
||||
|
TimePageLink pageLink); |
||||
|
|
||||
|
|
||||
|
// Select.Where query = CassandraAbstractSearchTimeDao.buildQuery(ModelConstants.RELATION_BY_TYPE_AND_CHILD_TYPE_VIEW_NAME,
|
||||
|
// Arrays.asList(eq(ModelConstants.RELATION_FROM_ID_PROPERTY, from.getId()),
|
||||
|
// eq(ModelConstants.RELATION_FROM_TYPE_PROPERTY, from.getEntityType().name()),
|
||||
|
// eq(ModelConstants.RELATION_TYPE_GROUP_PROPERTY, typeGroup.name()),
|
||||
|
// eq(ModelConstants.RELATION_TYPE_PROPERTY, relationType),
|
||||
|
// eq(ModelConstants.RELATION_TO_TYPE_PROPERTY, childType.name())),
|
||||
|
// Arrays.asList(QueryBuilder.asc(ModelConstants.RELATION_TYPE_GROUP_PROPERTY),
|
||||
|
// QueryBuilder.asc(ModelConstants.RELATION_TYPE_PROPERTY),
|
||||
|
// QueryBuilder.asc(ModelConstants.RELATION_TO_TYPE_PROPERTY)),
|
||||
|
// pageLink, ModelConstants.RELATION_TO_ID_PROPERTY);
|
||||
|
} |
||||
|
|
||||
|
// private UUID fromId;
|
||||
|
// private String fromType;
|
||||
|
// private UUID toId;
|
||||
|
// private String toType;
|
||||
|
// private String relationTypeGroup;
|
||||
|
// private String relationType;
|
||||
|
//
|
||||
|
//
|
||||
|
// ListenableFuture<Boolean> checkRelation(EntityId from, EntityId to, String relationType, RelationTypeGroup typeGroup);
|
||||
|
//
|
||||
|
// ListenableFuture<Boolean> saveRelation(EntityRelation relation);
|
||||
|
//
|
||||
|
// ListenableFuture<Boolean> deleteRelation(EntityRelation relation);
|
||||
|
//
|
||||
|
// ListenableFuture<Boolean> deleteRelation(EntityId from, EntityId to, String relationType, RelationTypeGroup typeGroup);
|
||||
|
//
|
||||
|
// ListenableFuture<Boolean> deleteOutboundRelations(EntityId entity);
|
||||
|
//
|
||||
|
// ListenableFuture<List<EntityRelation>> findRelations(EntityId from, String relationType, RelationTypeGroup typeGroup, EntityType toType, TimePageLink pageLink);
|
||||
Loading…
Reference in new issue