Andrii Shvaika
3 years ago
17 changed files with 292 additions and 99 deletions
@ -0,0 +1,109 @@ |
|||||
|
<!-- |
||||
|
|
||||
|
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. |
||||
|
|
||||
|
--> |
||||
|
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" |
||||
|
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> |
||||
|
<modelVersion>4.0.0</modelVersion> |
||||
|
<parent> |
||||
|
<groupId>org.thingsboard</groupId> |
||||
|
<version>3.6.0-SNAPSHOT</version> |
||||
|
<artifactId>common</artifactId> |
||||
|
</parent> |
||||
|
<groupId>org.thingsboard.common</groupId> |
||||
|
<artifactId>proto</artifactId> |
||||
|
<packaging>jar</packaging> |
||||
|
|
||||
|
<name>Thingsboard Server Common Protobuf and gRPC structures</name> |
||||
|
<url>https://thingsboard.io</url> |
||||
|
|
||||
|
<properties> |
||||
|
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding> |
||||
|
<main.dir>${basedir}/../..</main.dir> |
||||
|
</properties> |
||||
|
|
||||
|
<dependencies> |
||||
|
<dependency> |
||||
|
<groupId>org.thingsboard.common</groupId> |
||||
|
<artifactId>data</artifactId> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>org.thingsboard.common</groupId> |
||||
|
<artifactId>message</artifactId> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>com.google.protobuf</groupId> |
||||
|
<artifactId>protobuf-java</artifactId> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>com.google.protobuf</groupId> |
||||
|
<artifactId>protobuf-java-util</artifactId> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>io.grpc</groupId> |
||||
|
<artifactId>grpc-netty-shaded</artifactId> |
||||
|
<scope>provided</scope> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>io.grpc</groupId> |
||||
|
<artifactId>grpc-protobuf</artifactId> |
||||
|
<scope>provided</scope> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>io.grpc</groupId> |
||||
|
<artifactId>grpc-stub</artifactId> |
||||
|
<scope>provided</scope> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>org.springframework.boot</groupId> |
||||
|
<artifactId>spring-boot-starter-web</artifactId> |
||||
|
<scope>provided</scope> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>org.springframework.boot</groupId> |
||||
|
<artifactId>spring-boot-starter-test</artifactId> |
||||
|
<scope>test</scope> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>org.junit.vintage</groupId> |
||||
|
<artifactId>junit-vintage-engine</artifactId> |
||||
|
<scope>test</scope> |
||||
|
</dependency> |
||||
|
<dependency> |
||||
|
<groupId>org.awaitility</groupId> |
||||
|
<artifactId>awaitility</artifactId> |
||||
|
<scope>test</scope> |
||||
|
</dependency> |
||||
|
</dependencies> |
||||
|
|
||||
|
<build> |
||||
|
<plugins> |
||||
|
<plugin> |
||||
|
<groupId>org.xolstice.maven.plugins</groupId> |
||||
|
<artifactId>protobuf-maven-plugin</artifactId> |
||||
|
</plugin> |
||||
|
</plugins> |
||||
|
</build> |
||||
|
|
||||
|
<distributionManagement> |
||||
|
<repository> |
||||
|
<id>thingsboard-repo-deploy</id> |
||||
|
<name>ThingsBoard Repo Deployment</name> |
||||
|
<url>https://repo.thingsboard.io/artifactory/libs-release-public</url> |
||||
|
</repository> |
||||
|
</distributionManagement> |
||||
|
|
||||
|
</project> |
||||
@ -0,0 +1,156 @@ |
|||||
|
/** |
||||
|
* 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.common.transport.util; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.kv.AttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.BaseAttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.BasicTsKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.BooleanDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.DataType; |
||||
|
import org.thingsboard.server.common.data.kv.DoubleDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.JsonDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.KvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.LongDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.StringDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.TsKvEntry; |
||||
|
import org.thingsboard.server.gen.transport.TransportProtos; |
||||
|
|
||||
|
import javax.annotation.Nullable; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
|
||||
|
public class KvProtoUtil { |
||||
|
|
||||
|
public static List<TransportProtos.TsKvProto> attrToTsKvProtos(List<AttributeKvEntry> result) { |
||||
|
List<TransportProtos.TsKvProto> clientAttributes; |
||||
|
if (result == null || result.isEmpty()) { |
||||
|
clientAttributes = Collections.emptyList(); |
||||
|
} else { |
||||
|
clientAttributes = new ArrayList<>(result.size()); |
||||
|
for (AttributeKvEntry attrEntry : result) { |
||||
|
clientAttributes.add(toTsKvProto(attrEntry.getLastUpdateTs(), attrEntry)); |
||||
|
} |
||||
|
} |
||||
|
return clientAttributes; |
||||
|
} |
||||
|
|
||||
|
|
||||
|
public static List<TransportProtos.TsKvProto> tsToTsKvProtos(List<TsKvEntry> result) { |
||||
|
List<TransportProtos.TsKvProto> ts; |
||||
|
if (result == null || result.isEmpty()) { |
||||
|
ts = Collections.emptyList(); |
||||
|
} else { |
||||
|
ts = new ArrayList<>(result.size()); |
||||
|
for (TsKvEntry attrEntry : result) { |
||||
|
ts.add(toTsKvProto(attrEntry.getTs(), attrEntry)); |
||||
|
} |
||||
|
} |
||||
|
return ts; |
||||
|
} |
||||
|
|
||||
|
public static TransportProtos.TsKvProto toTsKvProto(long ts, KvEntry kvEntry) { |
||||
|
return TransportProtos.TsKvProto.newBuilder().setTs(ts) |
||||
|
.setKv(KvProtoUtil.toKeyValueProto(kvEntry)).build(); |
||||
|
} |
||||
|
|
||||
|
public static TransportProtos.KeyValueProto toKeyValueProto(KvEntry kvEntry) { |
||||
|
TransportProtos.KeyValueProto.Builder builder = TransportProtos.KeyValueProto.newBuilder(); |
||||
|
builder.setKey(kvEntry.getKey()); |
||||
|
switch (kvEntry.getDataType()) { |
||||
|
case BOOLEAN: |
||||
|
builder.setType(TransportProtos.KeyValueType.BOOLEAN_V); |
||||
|
builder.setBoolV(kvEntry.getBooleanValue().get()); |
||||
|
break; |
||||
|
case DOUBLE: |
||||
|
builder.setType(TransportProtos.KeyValueType.DOUBLE_V); |
||||
|
builder.setDoubleV(kvEntry.getDoubleValue().get()); |
||||
|
break; |
||||
|
case LONG: |
||||
|
builder.setType(TransportProtos.KeyValueType.LONG_V); |
||||
|
builder.setLongV(kvEntry.getLongValue().get()); |
||||
|
break; |
||||
|
case STRING: |
||||
|
builder.setType(TransportProtos.KeyValueType.STRING_V); |
||||
|
builder.setStringV(kvEntry.getStrValue().get()); |
||||
|
break; |
||||
|
case JSON: |
||||
|
builder.setType(TransportProtos.KeyValueType.JSON_V); |
||||
|
builder.setJsonV(kvEntry.getJsonValue().get()); |
||||
|
break; |
||||
|
} |
||||
|
return builder.build(); |
||||
|
} |
||||
|
|
||||
|
public static TransportProtos.TsKvProto.Builder toKeyValueProto(long ts, KvEntry attr) { |
||||
|
TransportProtos.KeyValueProto.Builder dataBuilder = TransportProtos.KeyValueProto.newBuilder(); |
||||
|
dataBuilder.setKey(attr.getKey()); |
||||
|
dataBuilder.setType(TransportProtos.KeyValueType.forNumber(attr.getDataType().ordinal())); |
||||
|
switch (attr.getDataType()) { |
||||
|
case BOOLEAN: |
||||
|
attr.getBooleanValue().ifPresent(dataBuilder::setBoolV); |
||||
|
break; |
||||
|
case LONG: |
||||
|
attr.getLongValue().ifPresent(dataBuilder::setLongV); |
||||
|
break; |
||||
|
case DOUBLE: |
||||
|
attr.getDoubleValue().ifPresent(dataBuilder::setDoubleV); |
||||
|
break; |
||||
|
case JSON: |
||||
|
attr.getJsonValue().ifPresent(dataBuilder::setJsonV); |
||||
|
break; |
||||
|
case STRING: |
||||
|
attr.getStrValue().ifPresent(dataBuilder::setStringV); |
||||
|
break; |
||||
|
} |
||||
|
return TransportProtos.TsKvProto.newBuilder().setTs(ts).setKv(dataBuilder); |
||||
|
} |
||||
|
|
||||
|
public static List<TsKvEntry> toTsKvEntityList(List<TransportProtos.TsKvProto> dataList) { |
||||
|
List<TsKvEntry> result = new ArrayList<>(dataList.size()); |
||||
|
dataList.forEach(proto -> result.add(new BasicTsKvEntry(proto.getTs(), getKvEntry(proto.getKv())))); |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
public static List<AttributeKvEntry> toAttributeKvList(List<TransportProtos.TsKvProto> dataList) { |
||||
|
List<AttributeKvEntry> result = new ArrayList<>(dataList.size()); |
||||
|
dataList.forEach(proto -> result.add(new BaseAttributeKvEntry(getKvEntry(proto.getKv()), proto.getTs()))); |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
private static KvEntry getKvEntry(TransportProtos.KeyValueProto proto) { |
||||
|
KvEntry entry = null; |
||||
|
DataType type = DataType.values()[proto.getType().getNumber()]; |
||||
|
switch (type) { |
||||
|
case BOOLEAN: |
||||
|
entry = new BooleanDataEntry(proto.getKey(), proto.getBoolV()); |
||||
|
break; |
||||
|
case LONG: |
||||
|
entry = new LongDataEntry(proto.getKey(), proto.getLongV()); |
||||
|
break; |
||||
|
case DOUBLE: |
||||
|
entry = new DoubleDataEntry(proto.getKey(), proto.getDoubleV()); |
||||
|
break; |
||||
|
case STRING: |
||||
|
entry = new StringDataEntry(proto.getKey(), proto.getStringV()); |
||||
|
break; |
||||
|
case JSON: |
||||
|
entry = new JsonDataEntry(proto.getKey(), proto.getJsonV()); |
||||
|
break; |
||||
|
} |
||||
|
return entry; |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue