Andrew Shvayka
9 years ago
12 changed files with 194 additions and 27 deletions
@ -1,31 +1,102 @@ |
|||||
|
/** |
||||
|
* 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.extensions.core.plugin.rpc.handlers; |
package org.thingsboard.server.extensions.core.plugin.rpc.handlers; |
||||
|
|
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.util.StringUtils; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
import org.thingsboard.server.common.data.id.RuleId; |
import org.thingsboard.server.common.data.id.RuleId; |
||||
import org.thingsboard.server.common.data.id.TenantId; |
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
||||
|
import org.thingsboard.server.extensions.api.plugins.PluginCallback; |
||||
import org.thingsboard.server.extensions.api.plugins.PluginContext; |
import org.thingsboard.server.extensions.api.plugins.PluginContext; |
||||
import org.thingsboard.server.extensions.api.plugins.handlers.RuleMsgHandler; |
import org.thingsboard.server.extensions.api.plugins.handlers.RuleMsgHandler; |
||||
import org.thingsboard.server.extensions.api.plugins.msg.RuleToPluginMsg; |
import org.thingsboard.server.extensions.api.plugins.msg.RuleToPluginMsg; |
||||
import org.thingsboard.server.extensions.api.plugins.msg.ToDeviceRpcRequest; |
import org.thingsboard.server.extensions.api.plugins.msg.ToDeviceRpcRequest; |
||||
|
import org.thingsboard.server.extensions.api.plugins.msg.ToDeviceRpcRequestBody; |
||||
import org.thingsboard.server.extensions.api.rules.RuleException; |
import org.thingsboard.server.extensions.api.rules.RuleException; |
||||
import org.thingsboard.server.extensions.core.action.rpc.ServerSideRpcCallActionMsg; |
import org.thingsboard.server.extensions.core.action.rpc.ServerSideRpcCallActionMsg; |
||||
|
import org.thingsboard.server.extensions.core.action.rpc.ServerSideRpcCallRuleToPluginActionMsg; |
||||
|
|
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
/** |
/** |
||||
* Created by ashvayka on 14.09.17. |
* Created by ashvayka on 14.09.17. |
||||
*/ |
*/ |
||||
|
@Slf4j |
||||
public class RpcRuleMsgHandler implements RuleMsgHandler { |
public class RpcRuleMsgHandler implements RuleMsgHandler { |
||||
|
|
||||
@Override |
@Override |
||||
public void process(PluginContext ctx, TenantId tenantId, RuleId ruleId, RuleToPluginMsg<?> msg) throws RuleException { |
public void process(PluginContext ctx, TenantId tenantId, RuleId ruleId, RuleToPluginMsg<?> msg) throws RuleException { |
||||
if (msg instanceof ServerSideRpcCallActionMsg) { |
if (msg instanceof ServerSideRpcCallRuleToPluginActionMsg) { |
||||
handle(ctx, tenantId, ruleId, (ServerSideRpcCallActionMsg) msg); |
handle(ctx, tenantId, ruleId, ((ServerSideRpcCallRuleToPluginActionMsg) msg).getPayload()); |
||||
} else { |
} else { |
||||
throw new RuntimeException("Not supported msg: " + msg + "!"); |
throw new RuntimeException("Not supported msg: " + msg + "!"); |
||||
} |
} |
||||
} |
} |
||||
|
|
||||
private void handle(PluginContext ctx, TenantId tenantId, RuleId ruleId, ServerSideRpcCallActionMsg msg) { |
private void handle(final PluginContext ctx, TenantId tenantId, RuleId ruleId, ServerSideRpcCallActionMsg msg) { |
||||
// TODO: implement
|
DeviceId deviceId = new DeviceId(UUID.fromString(msg.getDeviceId())); |
||||
// ToDeviceRpcRequest request = new ToDeviceRpcRequest();
|
ctx.checkAccess(deviceId, new PluginCallback<Void>() { |
||||
// ctx.sendRpcRequest(request);
|
@Override |
||||
|
public void onSuccess(PluginContext dummy, Void value) { |
||||
|
try { |
||||
|
List<EntityId> deviceIds; |
||||
|
if (StringUtils.isEmpty(msg.getFromDeviceRelation()) && StringUtils.isEmpty(msg.getToDeviceRelation())) { |
||||
|
deviceIds = Collections.singletonList(deviceId); |
||||
|
} else if (!StringUtils.isEmpty(msg.getFromDeviceRelation())) { |
||||
|
List<EntityRelation> relations = ctx.findByFromAndType(deviceId, msg.getFromDeviceRelation()).get(); |
||||
|
deviceIds = relations.stream().map(EntityRelation::getTo).collect(Collectors.toList()); |
||||
|
} else { |
||||
|
List<EntityRelation> relations = ctx.findByToAndType(deviceId, msg.getFromDeviceRelation()).get(); |
||||
|
deviceIds = relations.stream().map(EntityRelation::getFrom).collect(Collectors.toList()); |
||||
|
} |
||||
|
ToDeviceRpcRequestBody body = new ToDeviceRpcRequestBody(msg.getRpcCallMethod(), msg.getRpcCallBody()); |
||||
|
long expirationTime = System.currentTimeMillis() + msg.getRpcCallTimeoutInSec(); |
||||
|
for (EntityId address : deviceIds) { |
||||
|
DeviceId tmpId = new DeviceId(address.getId()); |
||||
|
ctx.checkAccess(tmpId, new PluginCallback<Void>() { |
||||
|
@Override |
||||
|
public void onSuccess(PluginContext ctx, Void value) { |
||||
|
ctx.sendRpcRequest(new ToDeviceRpcRequest(UUID.randomUUID(), |
||||
|
tenantId, tmpId, true, expirationTime, body) |
||||
|
); |
||||
|
log.trace("[{}] Sent RPC Call Action msg", tmpId); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(PluginContext ctx, Exception e) { |
||||
|
log.info("[{}] Failed to process RPC Call Action msg", tmpId, e); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.info("Failed to process RPC Call Action msg", e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(PluginContext dummy, Exception e) { |
||||
|
log.info("[{}] Failed to process RPC Call Action msg", deviceId, e); |
||||
|
} |
||||
|
}); |
||||
} |
} |
||||
} |
} |
||||
|
|||||
Loading…
Reference in new issue