Vladyslav_Prykhodko
4 years ago
15 changed files with 106 additions and 74 deletions
@ -0,0 +1,55 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 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.queue.notification; |
||||
|
|
||||
|
import com.google.protobuf.ByteString; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.server.common.msg.notification.NotificationRuleProcessor; |
||||
|
import org.thingsboard.server.common.msg.notification.trigger.NotificationRuleTrigger; |
||||
|
import org.thingsboard.server.common.msg.queue.ServiceType; |
||||
|
import org.thingsboard.server.common.msg.queue.TopicPartitionInfo; |
||||
|
import org.thingsboard.server.gen.transport.TransportProtos; |
||||
|
import org.thingsboard.server.queue.common.TbProtoQueueMsg; |
||||
|
import org.thingsboard.server.queue.discovery.PartitionService; |
||||
|
import org.thingsboard.server.queue.provider.TbQueueProducerProvider; |
||||
|
import org.thingsboard.server.queue.util.DataDecodingEncodingService; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
|
||||
|
@Service |
||||
|
@ConditionalOnMissingBean(NotificationRuleProcessor.class) |
||||
|
@RequiredArgsConstructor |
||||
|
public class RemoteNotificationRuleProcessor implements NotificationRuleProcessor { |
||||
|
|
||||
|
private final TbQueueProducerProvider producerProvider; |
||||
|
private final PartitionService partitionService; |
||||
|
private final DataDecodingEncodingService encodingService; |
||||
|
|
||||
|
@Override |
||||
|
public void process(NotificationRuleTrigger trigger) { |
||||
|
TransportProtos.NotificationRuleProcessorMsg.Builder msg = TransportProtos.NotificationRuleProcessorMsg.newBuilder() |
||||
|
.setTrigger(ByteString.copyFrom(encodingService.encode(trigger))); |
||||
|
|
||||
|
TopicPartitionInfo tpi = partitionService.resolve(ServiceType.TB_CORE, trigger.getTenantId(), trigger.getOriginatorEntityId()); |
||||
|
producerProvider.getTbCoreMsgProducer().send(tpi, new TbProtoQueueMsg<>(UUID.randomUUID(), |
||||
|
TransportProtos.ToCoreMsg.newBuilder() |
||||
|
.setNotificationRuleProcessorMsg(msg) |
||||
|
.build()), null); |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue