viktorbasanets
8 years ago
12 changed files with 197 additions and 13 deletions
@ -0,0 +1,137 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2018 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.rule.engine.action; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.ObjectMapper; |
||||
|
import com.google.common.base.Function; |
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import com.google.gson.JsonParser; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.thingsboard.rule.engine.api.EmptyNodeConfiguration; |
||||
|
import org.thingsboard.rule.engine.api.RuleNode; |
||||
|
import org.thingsboard.rule.engine.api.TbContext; |
||||
|
import org.thingsboard.rule.engine.api.TbNode; |
||||
|
import org.thingsboard.rule.engine.api.TbNodeConfiguration; |
||||
|
import org.thingsboard.rule.engine.api.TbNodeException; |
||||
|
import org.thingsboard.rule.engine.api.TbRelationTypes; |
||||
|
import org.thingsboard.rule.engine.api.util.TbNodeUtils; |
||||
|
import org.thingsboard.server.common.data.DataConstants; |
||||
|
import org.thingsboard.server.common.data.EntityView; |
||||
|
import org.thingsboard.server.common.data.kv.AttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.TsKvEntry; |
||||
|
import org.thingsboard.server.common.data.plugin.ComponentType; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.session.SessionMsgType; |
||||
|
import org.thingsboard.server.common.transport.adaptor.JsonConverter; |
||||
|
|
||||
|
import javax.annotation.Nullable; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
import java.util.Set; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
import static org.thingsboard.rule.engine.api.util.DonAsynchron.withCallback; |
||||
|
|
||||
|
@Slf4j |
||||
|
@RuleNode( |
||||
|
type = ComponentType.ACTION, |
||||
|
name = "copy attributes", |
||||
|
configClazz = EmptyNodeConfiguration.class, |
||||
|
nodeDescription = "Copy attributes from asset/device to entity view", |
||||
|
nodeDetails = "Copy attributes from asset/device to related entity view according to entity view configuration. \n " + |
||||
|
"Copy will be done only for attributes that are between start and end dates and according to attribute keys configuration", |
||||
|
uiResources = {"static/rulenode/rulenode-core-config.js"}, |
||||
|
configDirective = "tbNodeEmptyConfig", |
||||
|
icon = "content_copy" |
||||
|
) |
||||
|
public class TbCopyAttributesToEntityViewNode implements TbNode { |
||||
|
|
||||
|
EmptyNodeConfiguration config; |
||||
|
|
||||
|
@Override |
||||
|
public void init(TbContext ctx, TbNodeConfiguration configuration) throws TbNodeException { |
||||
|
this.config = TbNodeUtils.convert(configuration, EmptyNodeConfiguration.class); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onMsg(TbContext ctx, TbMsg msg) throws ExecutionException, InterruptedException, TbNodeException { |
||||
|
if (msg.getType().equals(SessionMsgType.POST_ATTRIBUTES_REQUEST.name()) || |
||||
|
msg.getType().equals(DataConstants.ATTRIBUTES_DELETED) || |
||||
|
msg.getType().equals(DataConstants.ATTRIBUTES_UPDATED)) { |
||||
|
long now = System.currentTimeMillis(); |
||||
|
String scope; |
||||
|
if (msg.getType().equals(SessionMsgType.POST_ATTRIBUTES_REQUEST.name())) { |
||||
|
scope = DataConstants.CLIENT_SCOPE; |
||||
|
} else { |
||||
|
scope = msg.getMetaData().getValue("scope"); |
||||
|
} |
||||
|
ListenableFuture<List<EntityView>> entityViewsFuture = |
||||
|
ctx.getEntityViewService().findEntityViewsByTenantIdAndEntityIdAsync(ctx.getTenantId(), msg.getOriginator()); |
||||
|
withCallback(entityViewsFuture, |
||||
|
entityViews -> { |
||||
|
List<ListenableFuture<List<Void>>> saveFutures = new ArrayList<>(); |
||||
|
for (EntityView entityView : entityViews) { |
||||
|
if ((entityView.getEndTimeMs() != 0 && entityView.getEndTimeMs() > now && entityView.getStartTimeMs() < now) || |
||||
|
(entityView.getEndTimeMs() == 0 && entityView.getStartTimeMs() < now)) { |
||||
|
Set<AttributeKvEntry> attributes = JsonConverter.convertToAttributes(new JsonParser().parse(msg.getData())).getAttributes(); |
||||
|
List<AttributeKvEntry> filteredAttributes = |
||||
|
attributes.stream() |
||||
|
.filter(attr -> { |
||||
|
switch (scope) { |
||||
|
case DataConstants.CLIENT_SCOPE: |
||||
|
if (entityView.getKeys().getAttributes().getCs().isEmpty()) { |
||||
|
return true; |
||||
|
} |
||||
|
return entityView.getKeys().getAttributes().getCs().contains(attr.getKey()); |
||||
|
case DataConstants.SERVER_SCOPE: |
||||
|
if (entityView.getKeys().getAttributes().getSs().isEmpty()) { |
||||
|
return true; |
||||
|
} |
||||
|
return entityView.getKeys().getAttributes().getSs().contains(attr.getKey()); |
||||
|
case DataConstants.SHARED_SCOPE: |
||||
|
if (entityView.getKeys().getAttributes().getSh().isEmpty()) { |
||||
|
return true; |
||||
|
} |
||||
|
return entityView.getKeys().getAttributes().getSh().contains(attr.getKey()); |
||||
|
} |
||||
|
return false; |
||||
|
}) |
||||
|
.collect(Collectors.toList()); |
||||
|
saveFutures.add(ctx.getAttributesService().save(entityView.getId(), scope, new ArrayList<>(filteredAttributes))); |
||||
|
} |
||||
|
} |
||||
|
Futures.transform(Futures.allAsList(saveFutures), new Function<List<List<Void>>, Object>() { |
||||
|
@Nullable |
||||
|
@Override |
||||
|
public Object apply(@Nullable List<List<Void>> lists) { |
||||
|
ctx.tellNext(msg, TbRelationTypes.SUCCESS); |
||||
|
return null; |
||||
|
} |
||||
|
}); |
||||
|
}, |
||||
|
t -> ctx.tellFailure(msg, t)); |
||||
|
} else { |
||||
|
ctx.tellNext(msg, TbRelationTypes.FAILURE); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void destroy() { |
||||
|
|
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue