Igor Kulikov
6 years ago
18 changed files with 771 additions and 217 deletions
@ -0,0 +1,21 @@ |
|||||
|
/** |
||||
|
* 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; |
||||
|
|
||||
|
public enum DataUpdateType { |
||||
|
ENTITY_DATA, |
||||
|
ALARM_DATA |
||||
|
} |
||||
@ -0,0 +1,176 @@ |
|||||
|
///
|
||||
|
/// 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.
|
||||
|
///
|
||||
|
|
||||
|
import { |
||||
|
AlarmDataCmd, |
||||
|
DataKeyType, |
||||
|
TelemetryService, |
||||
|
TelemetrySubscriber |
||||
|
} from '@shared/models/telemetry/telemetry.models'; |
||||
|
import { DatasourceType } from '@shared/models/widget.models'; |
||||
|
import { |
||||
|
AlarmData, |
||||
|
AlarmDataPageLink, |
||||
|
EntityFilter, |
||||
|
EntityKey, |
||||
|
EntityKeyType, |
||||
|
KeyFilter |
||||
|
} from '@shared/models/query/query.models'; |
||||
|
import { SubscriptionTimewindow } from '@shared/models/time/time.models'; |
||||
|
import { AlarmDataListener } from '@core/api/alarm-data.service'; |
||||
|
import { UtilsService } from '@core/services/utils.service'; |
||||
|
import { PageData } from '@shared/models/page/page-data'; |
||||
|
import { deepClone, isDefined, isDefinedAndNotNull, isObject } from '@core/utils'; |
||||
|
import { simulatedAlarm } from '@shared/models/alarm.models'; |
||||
|
|
||||
|
export interface AlarmSubscriptionDataKey { |
||||
|
name: string; |
||||
|
type: DataKeyType; |
||||
|
} |
||||
|
|
||||
|
export interface AlarmDataSubscriptionOptions { |
||||
|
datasourceType: DatasourceType; |
||||
|
dataKeys: Array<AlarmSubscriptionDataKey>; |
||||
|
entityFilter?: EntityFilter; |
||||
|
pageLink?: AlarmDataPageLink; |
||||
|
keyFilters?: Array<KeyFilter>; |
||||
|
additionalKeyFilters?: Array<KeyFilter>; |
||||
|
subscriptionTimewindow?: SubscriptionTimewindow; |
||||
|
} |
||||
|
|
||||
|
export class AlarmDataSubscription { |
||||
|
|
||||
|
private datasourceType: DatasourceType = this.alarmDataSubscriptionOptions.datasourceType; |
||||
|
|
||||
|
private history: boolean; |
||||
|
private realtime: boolean; |
||||
|
|
||||
|
private subscriber: TelemetrySubscriber; |
||||
|
private alarmDataCommand: AlarmDataCmd; |
||||
|
|
||||
|
private pageData: PageData<AlarmData>; |
||||
|
private alarmIdToDataIndex: {[id: string]: number}; |
||||
|
|
||||
|
private subsTw: SubscriptionTimewindow; |
||||
|
|
||||
|
constructor(public alarmDataSubscriptionOptions: AlarmDataSubscriptionOptions, |
||||
|
private listener: AlarmDataListener, |
||||
|
private telemetryService: TelemetryService, |
||||
|
private utils: UtilsService) { |
||||
|
} |
||||
|
|
||||
|
public unsubscribe() { |
||||
|
if (this.datasourceType === DatasourceType.entity) { |
||||
|
if (this.subscriber) { |
||||
|
this.subscriber.unsubscribe(); |
||||
|
this.subscriber = null; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public subscribe() { |
||||
|
this.subsTw = this.alarmDataSubscriptionOptions.subscriptionTimewindow; |
||||
|
this.history = this.alarmDataSubscriptionOptions.subscriptionTimewindow && |
||||
|
isObject(this.alarmDataSubscriptionOptions.subscriptionTimewindow.fixedWindow); |
||||
|
this.realtime = this.alarmDataSubscriptionOptions.subscriptionTimewindow && |
||||
|
isDefinedAndNotNull(this.alarmDataSubscriptionOptions.subscriptionTimewindow.realtimeWindowMs); |
||||
|
if (this.datasourceType === DatasourceType.entity) { |
||||
|
this.subscriber = new TelemetrySubscriber(this.telemetryService); |
||||
|
this.alarmDataCommand = new AlarmDataCmd(); |
||||
|
|
||||
|
const entityFields: Array<EntityKey> = |
||||
|
this.alarmDataSubscriptionOptions.dataKeys.filter(dataKey => dataKey.type === DataKeyType.entityField).map( |
||||
|
dataKey => ({ type: EntityKeyType.ENTITY_FIELD, key: dataKey.name }) |
||||
|
); |
||||
|
|
||||
|
const attrFields = this.alarmDataSubscriptionOptions.dataKeys.filter(dataKey => dataKey.type === DataKeyType.attribute).map( |
||||
|
dataKey => ({ type: EntityKeyType.ATTRIBUTE, key: dataKey.name }) |
||||
|
); |
||||
|
const tsFields = this.alarmDataSubscriptionOptions.dataKeys.filter(dataKey => dataKey.type === DataKeyType.timeseries).map( |
||||
|
dataKey => ({ type: EntityKeyType.TIME_SERIES, key: dataKey.name }) |
||||
|
); |
||||
|
const latestValues = attrFields.concat(tsFields); |
||||
|
|
||||
|
let keyFilters = this.alarmDataSubscriptionOptions.keyFilters; |
||||
|
if (this.alarmDataSubscriptionOptions.additionalKeyFilters) { |
||||
|
if (keyFilters) { |
||||
|
keyFilters = keyFilters.concat(this.alarmDataSubscriptionOptions.additionalKeyFilters); |
||||
|
} else { |
||||
|
keyFilters = this.alarmDataSubscriptionOptions.additionalKeyFilters; |
||||
|
} |
||||
|
} |
||||
|
this.alarmDataCommand.query = { |
||||
|
entityFilter: this.alarmDataSubscriptionOptions.entityFilter, |
||||
|
pageLink: deepClone(this.alarmDataSubscriptionOptions.pageLink), |
||||
|
keyFilters, |
||||
|
entityFields, |
||||
|
latestValues |
||||
|
}; |
||||
|
if (this.history) { |
||||
|
this.alarmDataCommand.query.pageLink.startTs = this.subsTw.fixedWindow.startTimeMs; |
||||
|
this.alarmDataCommand.query.pageLink.endTs = this.subsTw.fixedWindow.endTimeMs; |
||||
|
} else { |
||||
|
this.alarmDataCommand.query.pageLink.timeWindow = this.subsTw.realtimeWindowMs; |
||||
|
} |
||||
|
|
||||
|
this.subscriber.subscriptionCommands.push(this.alarmDataCommand); |
||||
|
|
||||
|
this.subscriber.alarmData$.subscribe((alarmDataUpdate) => { |
||||
|
if (alarmDataUpdate.data) { |
||||
|
this.onPageData(alarmDataUpdate.data); |
||||
|
} else if (alarmDataUpdate.update) { |
||||
|
this.onDataUpdate(alarmDataUpdate.update); |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
this.subscriber.subscribe(); |
||||
|
|
||||
|
} else if (this.datasourceType === DatasourceType.function) { |
||||
|
const pageData: PageData<AlarmData> = { |
||||
|
data: [{...simulatedAlarm, entityId: '1', latest: {}}], |
||||
|
hasNext: false, |
||||
|
totalElements: 1, |
||||
|
totalPages: 1 |
||||
|
}; |
||||
|
this.onPageData(pageData); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private resetData() { |
||||
|
this.alarmIdToDataIndex = {}; |
||||
|
for (let dataIndex = 0; dataIndex < this.pageData.data.length; dataIndex++) { |
||||
|
const alarmData = this.pageData.data[dataIndex]; |
||||
|
this.alarmIdToDataIndex[alarmData.id.id] = dataIndex; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private onPageData(pageData: PageData<AlarmData>) { |
||||
|
this.pageData = pageData; |
||||
|
this.resetData(); |
||||
|
this.listener.alarmsLoaded(pageData, this.alarmDataSubscriptionOptions.pageLink); |
||||
|
} |
||||
|
|
||||
|
private onDataUpdate(update: Array<AlarmData>) { |
||||
|
for (const alarmData of update) { |
||||
|
const dataIndex = this.alarmIdToDataIndex[alarmData.id.id]; |
||||
|
if (isDefined(dataIndex) && dataIndex >= 0) { |
||||
|
this.pageData.data[dataIndex] = alarmData; |
||||
|
} |
||||
|
} |
||||
|
this.listener.alarmsUpdated(update, this.pageData); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,94 @@ |
|||||
|
///
|
||||
|
/// 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.
|
||||
|
///
|
||||
|
|
||||
|
import { SubscriptionTimewindow } from '@shared/models/time/time.models'; |
||||
|
import { Datasource, DatasourceType } from '@shared/models/widget.models'; |
||||
|
import { PageData } from '@shared/models/page/page-data'; |
||||
|
import { AlarmData, AlarmDataPageLink, KeyFilter } from '@shared/models/query/query.models'; |
||||
|
import { Injectable } from '@angular/core'; |
||||
|
import { TelemetryWebsocketService } from '@core/ws/telemetry-websocket.service'; |
||||
|
import { UtilsService } from '@core/services/utils.service'; |
||||
|
import { |
||||
|
AlarmDataSubscription, |
||||
|
AlarmDataSubscriptionOptions, |
||||
|
AlarmSubscriptionDataKey |
||||
|
} from '@core/api/alarm-data-subscription'; |
||||
|
import { deepClone } from '@core/utils'; |
||||
|
|
||||
|
export interface AlarmDataListener { |
||||
|
subscriptionTimewindow?: SubscriptionTimewindow; |
||||
|
alarmSource: Datasource; |
||||
|
alarmsLoaded: (pageData: PageData<AlarmData>, pageLink: AlarmDataPageLink) => void; |
||||
|
alarmsUpdated: (update: Array<AlarmData>, pageData: PageData<AlarmData>) => void; |
||||
|
subscription?: AlarmDataSubscription; |
||||
|
} |
||||
|
|
||||
|
@Injectable({ |
||||
|
providedIn: 'root' |
||||
|
}) |
||||
|
export class AlarmDataService { |
||||
|
|
||||
|
constructor(private telemetryService: TelemetryWebsocketService, |
||||
|
private utils: UtilsService) {} |
||||
|
|
||||
|
|
||||
|
public subscribeForAlarms(listener: AlarmDataListener, |
||||
|
pageLink: AlarmDataPageLink, |
||||
|
keyFilters: KeyFilter[]) { |
||||
|
const alarmSource = listener.alarmSource; |
||||
|
if (alarmSource.type === DatasourceType.entity && (!alarmSource.entityFilter || !pageLink)) { |
||||
|
return; |
||||
|
} |
||||
|
listener.subscription = this.createSubscription(listener, |
||||
|
pageLink, alarmSource.keyFilters, keyFilters); |
||||
|
return listener.subscription.subscribe(); |
||||
|
} |
||||
|
|
||||
|
public stopSubscription(listener: AlarmDataListener) { |
||||
|
if (listener.subscription) { |
||||
|
listener.subscription.unsubscribe(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private createSubscription(listener: AlarmDataListener, |
||||
|
pageLink: AlarmDataPageLink, |
||||
|
keyFilters: KeyFilter[], |
||||
|
additionalKeyFilters: KeyFilter[]): AlarmDataSubscription { |
||||
|
const alarmSource = listener.alarmSource; |
||||
|
const alarmSubscriptionDataKeys: Array<AlarmSubscriptionDataKey> = []; |
||||
|
alarmSource.dataKeys.forEach((dataKey) => { |
||||
|
const alarmSubscriptionDataKey: AlarmSubscriptionDataKey = { |
||||
|
name: dataKey.name, |
||||
|
type: dataKey.type |
||||
|
}; |
||||
|
alarmSubscriptionDataKeys.push(alarmSubscriptionDataKey); |
||||
|
}); |
||||
|
const alarmDataSubscriptionOptions: AlarmDataSubscriptionOptions = { |
||||
|
datasourceType: alarmSource.type, |
||||
|
dataKeys: alarmSubscriptionDataKeys, |
||||
|
subscriptionTimewindow: deepClone(listener.subscriptionTimewindow) |
||||
|
}; |
||||
|
if (alarmDataSubscriptionOptions.datasourceType === DatasourceType.entity) { |
||||
|
alarmDataSubscriptionOptions.entityFilter = alarmSource.entityFilter; |
||||
|
alarmDataSubscriptionOptions.pageLink = pageLink; |
||||
|
alarmDataSubscriptionOptions.keyFilters = keyFilters; |
||||
|
alarmDataSubscriptionOptions.additionalKeyFilters = additionalKeyFilters; |
||||
|
} |
||||
|
return new AlarmDataSubscription(alarmDataSubscriptionOptions, |
||||
|
listener, this.telemetryService, this.utils); |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue