Andrii Shvaika
6 years ago
38 changed files with 1262 additions and 122 deletions
@ -0,0 +1,59 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.service.subscription; |
|||
|
|||
import lombok.Data; |
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.thingsboard.server.common.data.id.CustomerId; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.common.data.query.AbstractDataQuery; |
|||
import org.thingsboard.server.service.telemetry.TelemetryWebSocketService; |
|||
import org.thingsboard.server.service.telemetry.TelemetryWebSocketSessionRef; |
|||
|
|||
@Slf4j |
|||
@Data |
|||
public abstract class TbAbstractDataSubCtx<T extends AbstractDataQuery> { |
|||
|
|||
protected final String serviceId; |
|||
protected final TelemetryWebSocketService wsService; |
|||
protected final TelemetryWebSocketSessionRef sessionRef; |
|||
protected final int cmdId; |
|||
@Getter |
|||
@Setter |
|||
protected T query; |
|||
|
|||
public TbAbstractDataSubCtx(String serviceId, TelemetryWebSocketService wsService, TelemetryWebSocketSessionRef sessionRef, int cmdId) { |
|||
this.serviceId = serviceId; |
|||
this.wsService = wsService; |
|||
this.sessionRef = sessionRef; |
|||
this.cmdId = cmdId; |
|||
} |
|||
|
|||
public String getSessionId() { |
|||
return sessionRef.getSessionId(); |
|||
} |
|||
|
|||
public TenantId getTenantId() { |
|||
return sessionRef.getSecurityCtx().getTenantId(); |
|||
} |
|||
|
|||
public CustomerId getCustomerId() { |
|||
return sessionRef.getSecurityCtx().getCustomerId(); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,63 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.service.subscription; |
|||
|
|||
import lombok.Getter; |
|||
import lombok.Setter; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.thingsboard.server.common.data.id.EntityId; |
|||
import org.thingsboard.server.common.data.page.PageData; |
|||
import org.thingsboard.server.common.data.query.AlarmData; |
|||
import org.thingsboard.server.common.data.query.AlarmDataQuery; |
|||
import org.thingsboard.server.common.data.query.EntityData; |
|||
import org.thingsboard.server.service.telemetry.TelemetryWebSocketService; |
|||
import org.thingsboard.server.service.telemetry.TelemetryWebSocketSessionRef; |
|||
|
|||
import java.util.Collection; |
|||
import java.util.LinkedHashMap; |
|||
import java.util.List; |
|||
|
|||
@Slf4j |
|||
public class TbAlarmDataSubCtx extends TbAbstractDataSubCtx<AlarmDataQuery> { |
|||
|
|||
@Getter |
|||
@Setter |
|||
private final LinkedHashMap<EntityId, EntityData> entitiesMap; |
|||
@Getter |
|||
@Setter |
|||
private boolean tooManyEntities; |
|||
|
|||
public TbAlarmDataSubCtx(String serviceId, TelemetryWebSocketService wsService, TelemetryWebSocketSessionRef sessionRef, int cmdId) { |
|||
super(serviceId, wsService, sessionRef, cmdId); |
|||
this.entitiesMap = new LinkedHashMap<>(); |
|||
} |
|||
|
|||
public void setEntitiesData(PageData<EntityData> entitiesData) { |
|||
entitiesMap.clear(); |
|||
tooManyEntities = entitiesData.hasNext(); |
|||
for (EntityData entityData : entitiesData.getData()) { |
|||
entitiesMap.put(entityData.getEntityId(), entityData); |
|||
} |
|||
} |
|||
|
|||
public Collection<EntityId> getOrderedEntityIds() { |
|||
return entitiesMap.keySet(); |
|||
} |
|||
|
|||
public void setAlarmsData(PageData<AlarmData> alarms) { |
|||
// TODO: implement
|
|||
} |
|||
} |
|||
@ -0,0 +1,33 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.service.telemetry.cmd.v2; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonCreator; |
|||
import com.fasterxml.jackson.annotation.JsonProperty; |
|||
import lombok.Getter; |
|||
import org.thingsboard.server.common.data.query.AlarmDataQuery; |
|||
|
|||
public class AlarmDataCmd extends DataCmd { |
|||
|
|||
@Getter |
|||
private final AlarmDataQuery query; |
|||
|
|||
@JsonCreator |
|||
public AlarmDataCmd(@JsonProperty("cmdId") int cmdId, @JsonProperty("query") AlarmDataQuery query) { |
|||
super(cmdId); |
|||
this.query = query; |
|||
} |
|||
} |
|||
@ -0,0 +1,25 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.service.telemetry.cmd.v2; |
|||
|
|||
import lombok.Data; |
|||
|
|||
@Data |
|||
public class AlarmDataUnsubscribeCmd { |
|||
|
|||
private final int cmdId; |
|||
|
|||
} |
|||
@ -0,0 +1,46 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.service.telemetry.cmd.v2; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonCreator; |
|||
import com.fasterxml.jackson.annotation.JsonProperty; |
|||
import lombok.NoArgsConstructor; |
|||
import org.thingsboard.server.common.data.page.PageData; |
|||
import org.thingsboard.server.common.data.query.AlarmData; |
|||
import org.thingsboard.server.common.data.query.EntityData; |
|||
import org.thingsboard.server.service.telemetry.sub.SubscriptionErrorCode; |
|||
|
|||
import java.util.List; |
|||
|
|||
public class AlarmDataUpdate extends DataUpdate<AlarmData> { |
|||
|
|||
public AlarmDataUpdate(int cmdId, PageData<AlarmData> data, List<AlarmData> update) { |
|||
super(cmdId, data, update, SubscriptionErrorCode.NO_ERROR.getCode(), null); |
|||
} |
|||
|
|||
public AlarmDataUpdate(int cmdId, int errorCode, String errorMsg) { |
|||
super(cmdId, null, null, errorCode, errorMsg); |
|||
} |
|||
|
|||
@JsonCreator |
|||
public AlarmDataUpdate(@JsonProperty("cmdId") int cmdId, |
|||
@JsonProperty("data") PageData<AlarmData> data, |
|||
@JsonProperty("update") List<AlarmData> update, |
|||
@JsonProperty("errorCode") int errorCode, |
|||
@JsonProperty("errorMsg") String errorMsg) { |
|||
super(cmdId, data, update, errorCode, errorMsg); |
|||
} |
|||
} |
|||
@ -0,0 +1,32 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.service.telemetry.cmd.v2; |
|||
|
|||
import lombok.Data; |
|||
import lombok.Getter; |
|||
import lombok.NoArgsConstructor; |
|||
|
|||
@Data |
|||
public class DataCmd { |
|||
|
|||
@Getter |
|||
private final int cmdId; |
|||
|
|||
public DataCmd(int cmdId) { |
|||
this.cmdId = cmdId; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,43 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.service.telemetry.cmd.v2; |
|||
|
|||
import lombok.AllArgsConstructor; |
|||
import lombok.Data; |
|||
import org.thingsboard.server.common.data.page.PageData; |
|||
import org.thingsboard.server.service.telemetry.sub.SubscriptionErrorCode; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Data |
|||
@AllArgsConstructor |
|||
public abstract class DataUpdate<T> { |
|||
|
|||
private final int cmdId; |
|||
private final PageData<T> data; |
|||
private final List<T> update; |
|||
private final int errorCode; |
|||
private final String errorMsg; |
|||
|
|||
public DataUpdate(int cmdId, PageData<T> data, List<T> update) { |
|||
this(cmdId, data, update, SubscriptionErrorCode.NO_ERROR.getCode(), null); |
|||
} |
|||
|
|||
public DataUpdate(int cmdId, int errorCode, String errorMsg) { |
|||
this(cmdId, null, null, errorCode, errorMsg); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,56 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.query; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonIgnore; |
|||
import lombok.Getter; |
|||
import lombok.ToString; |
|||
|
|||
import java.util.List; |
|||
|
|||
@ToString |
|||
public abstract class AbstractDataQuery<T extends EntityDataPageLink> extends EntityCountQuery { |
|||
|
|||
@Getter |
|||
protected T pageLink; |
|||
@Getter |
|||
protected List<EntityKey> entityFields; |
|||
@Getter |
|||
protected List<EntityKey> latestValues; |
|||
@Getter |
|||
protected List<KeyFilter> keyFilters; |
|||
|
|||
public AbstractDataQuery() { |
|||
super(); |
|||
} |
|||
|
|||
public AbstractDataQuery(EntityFilter entityFilter) { |
|||
super(entityFilter); |
|||
} |
|||
|
|||
public AbstractDataQuery(EntityFilter entityFilter, |
|||
T pageLink, |
|||
List<EntityKey> entityFields, |
|||
List<EntityKey> latestValues, |
|||
List<KeyFilter> keyFilters) { |
|||
super(entityFilter); |
|||
this.pageLink = pageLink; |
|||
this.entityFields = entityFields; |
|||
this.latestValues = latestValues; |
|||
this.keyFilters = keyFilters; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,37 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.query; |
|||
|
|||
import lombok.Data; |
|||
import org.thingsboard.server.common.data.alarm.Alarm; |
|||
import org.thingsboard.server.common.data.alarm.AlarmInfo; |
|||
import org.thingsboard.server.common.data.id.EntityId; |
|||
|
|||
import java.util.HashMap; |
|||
import java.util.Map; |
|||
import java.util.UUID; |
|||
|
|||
public class AlarmData extends AlarmInfo { |
|||
|
|||
private final UUID entityId; |
|||
private final Map<EntityKeyType, Map<String, TsValue>> latest; |
|||
|
|||
public AlarmData(Alarm alarm, String originatorName, UUID entityId) { |
|||
super(alarm, originatorName); |
|||
this.entityId = entityId; |
|||
this.latest = new HashMap<>(); |
|||
} |
|||
} |
|||
@ -0,0 +1,67 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.query; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonIgnore; |
|||
import lombok.AllArgsConstructor; |
|||
import lombok.Data; |
|||
import lombok.Getter; |
|||
import org.thingsboard.server.common.data.alarm.AlarmSearchStatus; |
|||
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
|||
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
|||
|
|||
import java.util.List; |
|||
|
|||
@Data |
|||
@AllArgsConstructor |
|||
public class AlarmDataPageLink extends EntityDataPageLink { |
|||
|
|||
private long startTs; |
|||
private long endTs; |
|||
//TODO: handle this;
|
|||
private long timeWindow; |
|||
private List<String> typeList; |
|||
private List<AlarmSearchStatus> statusList; |
|||
private List<AlarmSeverity> severityList; |
|||
private boolean searchPropagatedAlarms; |
|||
|
|||
public AlarmDataPageLink() { |
|||
super(); |
|||
} |
|||
|
|||
public AlarmDataPageLink(int pageSize, int page, String textSearch, EntityDataSortOrder sortOrder, boolean dynamic, |
|||
boolean searchPropagatedAlarms, |
|||
long startTs, long endTs, long timeWindow, |
|||
List<String> typeList, List<AlarmSearchStatus> statusList, List<AlarmSeverity> severityList) { |
|||
super(pageSize, page, textSearch, sortOrder, dynamic); |
|||
this.searchPropagatedAlarms = searchPropagatedAlarms; |
|||
this.startTs = startTs; |
|||
this.endTs = endTs; |
|||
this.timeWindow = timeWindow; |
|||
this.typeList = typeList; |
|||
this.statusList = statusList; |
|||
this.severityList = severityList; |
|||
} |
|||
|
|||
@JsonIgnore |
|||
public AlarmDataPageLink nextPageLink() { |
|||
return new AlarmDataPageLink(this.getPageSize(), this.getPage() + 1, this.getTextSearch(), this.getSortOrder(), this.isDynamic(), |
|||
this.searchPropagatedAlarms, |
|||
this.startTs, this.endTs, this.timeWindow, |
|||
this.typeList, this.statusList, this.severityList |
|||
); |
|||
} |
|||
} |
|||
@ -0,0 +1,42 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.query; |
|||
|
|||
import com.fasterxml.jackson.annotation.JsonIgnore; |
|||
import lombok.Getter; |
|||
import lombok.ToString; |
|||
|
|||
import java.util.List; |
|||
|
|||
@ToString |
|||
public class AlarmDataQuery extends AbstractDataQuery<AlarmDataPageLink> { |
|||
|
|||
public AlarmDataQuery() { |
|||
} |
|||
|
|||
public AlarmDataQuery(EntityFilter entityFilter) { |
|||
super(entityFilter); |
|||
} |
|||
|
|||
public AlarmDataQuery(EntityFilter entityFilter, AlarmDataPageLink pageLink, List<EntityKey> entityFields, List<EntityKey> latestValues, List<KeyFilter> keyFilters) { |
|||
super(entityFilter, pageLink, entityFields, latestValues, keyFilters); |
|||
} |
|||
|
|||
@JsonIgnore |
|||
public AlarmDataQuery next() { |
|||
return new AlarmDataQuery(getEntityFilter(), getPageLink().nextPageLink(), entityFields, latestValues, keyFilters); |
|||
} |
|||
} |
|||
@ -0,0 +1,100 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.query; |
|||
|
|||
import com.fasterxml.jackson.core.JsonProcessingException; |
|||
import com.fasterxml.jackson.databind.ObjectMapper; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.util.StringUtils; |
|||
import org.thingsboard.server.common.data.EntityType; |
|||
import org.thingsboard.server.common.data.alarm.Alarm; |
|||
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
|||
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
|||
import org.thingsboard.server.common.data.id.AlarmId; |
|||
import org.thingsboard.server.common.data.id.EntityIdFactory; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.common.data.page.PageData; |
|||
import org.thingsboard.server.common.data.query.AlarmData; |
|||
import org.thingsboard.server.common.data.query.EntityDataPageLink; |
|||
import org.thingsboard.server.dao.model.ModelConstants; |
|||
|
|||
import java.util.Arrays; |
|||
import java.util.Collections; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
import java.util.UUID; |
|||
import java.util.stream.Collectors; |
|||
|
|||
@Slf4j |
|||
public class AlarmDataAdapter { |
|||
|
|||
private final static ObjectMapper mapper = new ObjectMapper(); |
|||
|
|||
public static PageData<AlarmData> createAlarmData(EntityDataPageLink pageLink, |
|||
List<Map<String, Object>> rows, |
|||
int totalElements) { |
|||
int totalPages = pageLink.getPageSize() > 0 ? (int) Math.ceil((float) totalElements / pageLink.getPageSize()) : 1; |
|||
int startIndex = pageLink.getPageSize() * pageLink.getPage(); |
|||
boolean hasNext = pageLink.getPageSize() > 0 && totalElements > startIndex + rows.size(); |
|||
List<AlarmData> entitiesData = convertListToAlarmData(rows); |
|||
return new PageData<>(entitiesData, totalPages, totalElements, hasNext); |
|||
} |
|||
|
|||
private static List<AlarmData> convertListToAlarmData(List<Map<String, Object>> result) { |
|||
return result.stream().map(AlarmDataAdapter::toEntityData).collect(Collectors.toList()); |
|||
} |
|||
|
|||
private static AlarmData toEntityData(Map<String, Object> row) { |
|||
Alarm alarm = new Alarm(); |
|||
alarm.setId(new AlarmId((UUID) row.get(ModelConstants.ID_PROPERTY))); |
|||
alarm.setCreatedTime((long) row.get(ModelConstants.CREATED_TIME_PROPERTY)); |
|||
alarm.setAckTs((long) row.get(ModelConstants.ALARM_ACK_TS_PROPERTY)); |
|||
alarm.setClearTs((long) row.get(ModelConstants.ALARM_CLEAR_TS_PROPERTY)); |
|||
alarm.setStartTs((long) row.get(ModelConstants.ALARM_START_TS_PROPERTY)); |
|||
alarm.setEndTs((long) row.get(ModelConstants.ALARM_END_TS_PROPERTY)); |
|||
Object additionalInfo = row.get(ModelConstants.ADDITIONAL_INFO_PROPERTY); |
|||
if (additionalInfo != null) { |
|||
try { |
|||
alarm.setDetails(mapper.readTree(additionalInfo.toString())); |
|||
} catch (JsonProcessingException e) { |
|||
log.warn("Failed to parse json: {}", row.get(ModelConstants.ADDITIONAL_INFO_PROPERTY), e); |
|||
} |
|||
} |
|||
EntityType originatorType = EntityType.values()[(int) row.get(ModelConstants.ALARM_ORIGINATOR_TYPE_PROPERTY)]; |
|||
UUID originatorId = (UUID) row.get(ModelConstants.ALARM_ORIGINATOR_ID_PROPERTY); |
|||
alarm.setOriginator(EntityIdFactory.getByTypeAndUuid(originatorType, originatorId)); |
|||
alarm.setPropagate((boolean) row.get(ModelConstants.ALARM_PROPAGATE_PROPERTY)); |
|||
alarm.setType(row.get(ModelConstants.ALARM_TYPE_PROPERTY).toString()); |
|||
alarm.setSeverity(AlarmSeverity.valueOf(row.get(ModelConstants.ALARM_SEVERITY_PROPERTY).toString())); |
|||
alarm.setStatus(AlarmStatus.valueOf(row.get(ModelConstants.ALARM_STATUS_PROPERTY).toString())); |
|||
alarm.setTenantId(new TenantId((UUID) row.get(ModelConstants.TENANT_ID_PROPERTY))); |
|||
if (row.get(ModelConstants.ALARM_PROPAGATE_RELATION_TYPES) != null) { |
|||
String propagateRelationTypes = row.get(ModelConstants.ALARM_PROPAGATE_RELATION_TYPES).toString(); |
|||
if (!StringUtils.isEmpty(propagateRelationTypes)) { |
|||
alarm.setPropagateRelationTypes(Arrays.asList(propagateRelationTypes.split(","))); |
|||
} else { |
|||
alarm.setPropagateRelationTypes(Collections.emptyList()); |
|||
} |
|||
} else { |
|||
alarm.setPropagateRelationTypes(Collections.emptyList()); |
|||
} |
|||
UUID entityId = (UUID) row.get(ModelConstants.ENTITY_ID_COLUMN); |
|||
Object originatorNameObj = row.get(ModelConstants.ALARM_ORIGINATOR_NAME_PROPERTY); |
|||
String originatorName = originatorNameObj != null ? originatorNameObj.toString() : null; |
|||
return new AlarmData(alarm, originatorName, entityId); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,32 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.query; |
|||
|
|||
import org.thingsboard.server.common.data.id.CustomerId; |
|||
import org.thingsboard.server.common.data.id.EntityId; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.common.data.page.PageData; |
|||
import org.thingsboard.server.common.data.query.AlarmData; |
|||
import org.thingsboard.server.common.data.query.AlarmDataPageLink; |
|||
|
|||
import java.util.Collection; |
|||
|
|||
public interface AlarmQueryRepository { |
|||
|
|||
PageData<AlarmData> findAlarmDataByQueryForEntities(TenantId tenantId, CustomerId customerId, |
|||
AlarmDataPageLink pageLink, Collection<EntityId> orderedEntityIds); |
|||
|
|||
} |
|||
@ -0,0 +1,241 @@ |
|||
/** |
|||
* Copyright © 2016-2020 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.query; |
|||
|
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate; |
|||
import org.springframework.stereotype.Repository; |
|||
import org.thingsboard.server.common.data.alarm.AlarmSearchStatus; |
|||
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
|||
import org.thingsboard.server.common.data.alarm.AlarmStatus; |
|||
import org.thingsboard.server.common.data.id.CustomerId; |
|||
import org.thingsboard.server.common.data.id.EntityId; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.common.data.page.PageData; |
|||
import org.thingsboard.server.common.data.query.AlarmData; |
|||
import org.thingsboard.server.common.data.query.AlarmDataPageLink; |
|||
import org.thingsboard.server.common.data.query.EntityDataSortOrder; |
|||
import org.thingsboard.server.common.data.query.EntityKeyType; |
|||
import org.thingsboard.server.dao.model.ModelConstants; |
|||
import org.thingsboard.server.dao.util.SqlDao; |
|||
|
|||
import java.util.Collection; |
|||
import java.util.HashMap; |
|||
import java.util.HashSet; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
import java.util.Set; |
|||
import java.util.stream.Collectors; |
|||
|
|||
@SqlDao |
|||
@Repository |
|||
@Slf4j |
|||
public class DefaultAlarmQueryRepository implements AlarmQueryRepository { |
|||
|
|||
private static final Map<String, String> alarmFieldColumnMap = new HashMap<>(); |
|||
|
|||
static { |
|||
alarmFieldColumnMap.put("createdTime", ModelConstants.CREATED_TIME_PROPERTY); |
|||
alarmFieldColumnMap.put("ackTs", ModelConstants.ALARM_ACK_TS_PROPERTY); |
|||
alarmFieldColumnMap.put("clearTs", ModelConstants.ALARM_CLEAR_TS_PROPERTY); |
|||
alarmFieldColumnMap.put("details", ModelConstants.ADDITIONAL_INFO_PROPERTY); |
|||
alarmFieldColumnMap.put("endTs", ModelConstants.ALARM_END_TS_PROPERTY); |
|||
alarmFieldColumnMap.put("startTs", ModelConstants.ALARM_START_TS_PROPERTY); |
|||
alarmFieldColumnMap.put("status", ModelConstants.ALARM_STATUS_PROPERTY); |
|||
alarmFieldColumnMap.put("type", ModelConstants.ALARM_TYPE_PROPERTY); |
|||
alarmFieldColumnMap.put("severity", ModelConstants.ALARM_SEVERITY_PROPERTY); |
|||
alarmFieldColumnMap.put("originator_id", ModelConstants.ALARM_ORIGINATOR_ID_PROPERTY); |
|||
alarmFieldColumnMap.put("originator_type", ModelConstants.ALARM_ORIGINATOR_TYPE_PROPERTY); |
|||
} |
|||
|
|||
public static final String SELECT_ORIGINATOR_NAME = " CASE" + |
|||
" WHEN a.originator_type = 0" + |
|||
" THEN (select title from tenant where id = a.originator_id)" + |
|||
" WHEN a.originator_type = 1 " + |
|||
" THEN (select title from customer where id = a.originator_id)" + |
|||
" WHEN a.originator_type = 2" + |
|||
" THEN (select CONCAT (first_name, ' ', last_name) from tb_user where id = a.originator_id)" + |
|||
" WHEN a.originator_type = 3" + |
|||
" THEN (select title from dashboard where id = a.originator_id)" + |
|||
" WHEN a.originator_type = 4" + |
|||
" THEN (select name from asset where id = a.originator_id)" + |
|||
" WHEN a.originator_type = 5" + |
|||
" THEN (select name from device where id = a.originator_id)" + |
|||
" WHEN a.originator_type = 9" + |
|||
" THEN (select name from entity_view where id = a.originator_id)" + |
|||
" END as originator_name"; |
|||
|
|||
public static final String FIELDS_SELECTION = "select a.id as id," + |
|||
" a.created_time as created_time," + |
|||
" a.ack_ts as ack_ts," + |
|||
" a.clear_ts as clear_ts," + |
|||
" a.additional_info as additional_info," + |
|||
" a.end_ts as end_ts," + |
|||
" a.originator_id as originator_id," + |
|||
" a.originator_type as originator_type," + |
|||
" a.propagate as propagate," + |
|||
" a.severity as severity," + |
|||
" a.start_ts as start_ts," + |
|||
" a.status as status, " + |
|||
" a.tenant_id as tenant_id, " + |
|||
" a.propagate_relation_types as propagate_relation_types, " + |
|||
" a.type as type," + SELECT_ORIGINATOR_NAME + ", "; |
|||
|
|||
public static final String JOIN_RELATIONS = "left join relation r on r.relation_type_group = 'ALARM' and r.relation_type = 'ALARM_ANY' and a.id = r.to_id"; |
|||
|
|||
@Autowired |
|||
protected NamedParameterJdbcTemplate jdbcTemplate; |
|||
|
|||
|
|||
@Override |
|||
public PageData<AlarmData> findAlarmDataByQueryForEntities(TenantId tenantId, CustomerId customerId, |
|||
AlarmDataPageLink pageLink, Collection<EntityId> orderedEntityIds) { |
|||
QueryContext ctx = new QueryContext(); |
|||
|
|||
StringBuilder selectPart = new StringBuilder(FIELDS_SELECTION); |
|||
StringBuilder fromPart = new StringBuilder(" from alarm a "); |
|||
StringBuilder wherePart = new StringBuilder(" where "); |
|||
StringBuilder sortPart = new StringBuilder(" order by "); |
|||
boolean addAnd = false; |
|||
if (pageLink.isSearchPropagatedAlarms()) { |
|||
selectPart.append(" r.from_id as entity_id "); |
|||
fromPart.append(JOIN_RELATIONS); |
|||
} else { |
|||
selectPart.append(" a.originator_id as entity_id "); |
|||
} |
|||
EntityDataSortOrder sortOrder = pageLink.getSortOrder(); |
|||
if (sortOrder != null && sortOrder.getKey().getType().equals(EntityKeyType.ALARM_FIELD)) { |
|||
String sortOrderKey = sortOrder.getKey().getKey(); |
|||
sortPart.append("a.").append(alarmFieldColumnMap.getOrDefault(sortOrderKey, sortOrderKey)) |
|||
.append(" ").append(sortOrder.getDirection().name()); |
|||
ctx.addUuidListParameter("entity_ids", orderedEntityIds.stream().map(EntityId::getId).collect(Collectors.toList())); |
|||
if (pageLink.isSearchPropagatedAlarms()) { |
|||
fromPart.append(" and r.from_id in (:entity_ids)"); |
|||
} else { |
|||
wherePart.append(" a.originator_id in (:entity_ids)"); |
|||
addAnd = true; |
|||
} |
|||
} else { |
|||
fromPart.append(" left join (select * from (VALUES"); |
|||
int entityIdIdx = 0; |
|||
int lastEntityIdIdx = orderedEntityIds.size() - 1; |
|||
for (EntityId entityId : orderedEntityIds) { |
|||
fromPart.append("(uuid('").append(entityId.getId().toString()).append("'), ").append(entityIdIdx).append(")"); |
|||
if (entityIdIdx != lastEntityIdIdx) { |
|||
fromPart.append(","); |
|||
} else { |
|||
fromPart.append(")"); |
|||
} |
|||
entityIdIdx++; |
|||
} |
|||
fromPart.append(" as e(id, priority)) e "); |
|||
if (pageLink.isSearchPropagatedAlarms()) { |
|||
fromPart.append("on r.from_id = e.id"); |
|||
} else { |
|||
fromPart.append("on a.originator_id = e.id"); |
|||
} |
|||
sortPart.append("e.priority"); |
|||
} |
|||
|
|||
if (pageLink.getStartTs() > 0) { |
|||
addAndIfNeeded(wherePart, addAnd); |
|||
addAnd = true; |
|||
ctx.addLongParameter("startTime", pageLink.getStartTs()); |
|||
wherePart.append("a.created_time >= :startTime"); |
|||
} |
|||
|
|||
if (pageLink.getEndTs() > 0) { |
|||
addAndIfNeeded(wherePart, addAnd); |
|||
addAnd = true; |
|||
ctx.addLongParameter("endTime", pageLink.getEndTs()); |
|||
wherePart.append("a.created_time <= :endTime"); |
|||
} |
|||
|
|||
if (pageLink.getTypeList() != null && !pageLink.getTypeList().isEmpty()) { |
|||
addAndIfNeeded(wherePart, addAnd); |
|||
addAnd = true; |
|||
ctx.addStringListParameter("alarmTypes", pageLink.getTypeList()); |
|||
wherePart.append("a.type in (:alarmTypes)"); |
|||
} |
|||
|
|||
if (pageLink.getSeverityList() != null && !pageLink.getSeverityList().isEmpty()) { |
|||
addAndIfNeeded(wherePart, addAnd); |
|||
addAnd = true; |
|||
ctx.addStringListParameter("alarmSeverities", pageLink.getSeverityList().stream().map(AlarmSeverity::name).collect(Collectors.toList())); |
|||
wherePart.append("a.severity in (:alarmSeverities)"); |
|||
} |
|||
|
|||
if (pageLink.getStatusList() != null && !pageLink.getStatusList().isEmpty()) { |
|||
Set<AlarmStatus> statusSet = toStatusSet(pageLink.getStatusList()); |
|||
if (!statusSet.isEmpty()) { |
|||
addAndIfNeeded(wherePart, addAnd); |
|||
addAnd = true; |
|||
ctx.addStringListParameter("alarmStatuses", statusSet.stream().map(AlarmStatus::name).collect(Collectors.toList())); |
|||
wherePart.append(" a.status in (:alarmStatuses)"); |
|||
} |
|||
} |
|||
|
|||
String countQuery = fromPart.toString() + wherePart.toString(); |
|||
int totalElements = jdbcTemplate.queryForObject(String.format("select count(*) %s", countQuery), ctx, Integer.class); |
|||
|
|||
String dataQuery = selectPart.toString() + countQuery + sortPart; |
|||
|
|||
int startIndex = pageLink.getPageSize() * pageLink.getPage(); |
|||
if (pageLink.getPageSize() > 0) { |
|||
dataQuery = String.format("%s limit %s offset %s", dataQuery, pageLink.getPageSize(), startIndex); |
|||
} |
|||
List<Map<String, Object>> rows = jdbcTemplate.queryForList(dataQuery, ctx); |
|||
return AlarmDataAdapter.createAlarmData(pageLink, rows, totalElements); |
|||
} |
|||
|
|||
private Set<AlarmStatus> toStatusSet(List<AlarmSearchStatus> statusList) { |
|||
Set<AlarmStatus> result = new HashSet<>(); |
|||
for (AlarmSearchStatus searchStatus : statusList) { |
|||
switch (searchStatus) { |
|||
case ACK: |
|||
result.add(AlarmStatus.ACTIVE_ACK); |
|||
result.add(AlarmStatus.CLEARED_ACK); |
|||
break; |
|||
case UNACK: |
|||
result.add(AlarmStatus.ACTIVE_UNACK); |
|||
result.add(AlarmStatus.CLEARED_UNACK); |
|||
break; |
|||
case CLEARED: |
|||
result.add(AlarmStatus.CLEARED_ACK); |
|||
result.add(AlarmStatus.CLEARED_UNACK); |
|||
break; |
|||
case ACTIVE: |
|||
result.add(AlarmStatus.ACTIVE_ACK); |
|||
result.add(AlarmStatus.ACTIVE_UNACK); |
|||
break; |
|||
default: |
|||
break; |
|||
} |
|||
if (searchStatus == AlarmSearchStatus.ANY || result.size() == AlarmStatus.values().length) { |
|||
result.clear(); |
|||
return result; |
|||
} |
|||
} |
|||
return result; |
|||
} |
|||
|
|||
private void addAndIfNeeded(StringBuilder wherePart, boolean addAnd) { |
|||
if (addAnd) { |
|||
wherePart.append(" and "); |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue