兰染
4 years ago
committed by
GitHub
GitHub
19 changed files with 531 additions and 239 deletions
@ -1,21 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2022 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.transport.lwm2m.server.client; |
|||
|
|||
import org.eclipse.leshan.core.node.LwM2mResource; |
|||
|
|||
public interface TbLwM2MResource extends LwM2mResource { |
|||
} |
|||
@ -1,30 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2022 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.transport.lwm2m.server.client; |
|||
|
|||
import org.eclipse.leshan.core.model.ResourceModel; |
|||
import org.eclipse.leshan.core.node.LwM2mResourceInstance; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class TbLwM2MResourceInstance extends LwM2mResourceInstance implements Serializable { |
|||
|
|||
private static final long serialVersionUID = -8322290426892538345L; |
|||
|
|||
protected TbLwM2MResourceInstance(int id, Object value, ResourceModel.Type type) { |
|||
super(id, value, type); |
|||
} |
|||
} |
|||
@ -1,30 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2022 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.transport.lwm2m.server.client; |
|||
|
|||
import org.eclipse.leshan.core.model.ResourceModel; |
|||
import org.eclipse.leshan.core.node.LwM2mSingleResource; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class TbLwM2MSingleResource extends LwM2mSingleResource implements TbLwM2MResource, Serializable { |
|||
|
|||
private static final long serialVersionUID = -878078368245340809L; |
|||
|
|||
public TbLwM2MSingleResource(int id, Object value, ResourceModel.Type type) { |
|||
super(id, value, type); |
|||
} |
|||
} |
|||
@ -1,30 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2022 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.transport.lwm2m.server.client; |
|||
|
|||
import org.eclipse.leshan.core.model.ResourceModel; |
|||
import org.eclipse.leshan.core.node.LwM2mMultipleResource; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class TbLwM2mMultipleResource extends LwM2mMultipleResource implements TbLwM2MResource, Serializable { |
|||
|
|||
private static final long serialVersionUID = 4658477128628087186L; |
|||
|
|||
public TbLwM2mMultipleResource(int id, ResourceModel.Type type, TbLwM2MResourceInstance... instances) { |
|||
super(id, type, instances); |
|||
} |
|||
} |
|||
@ -1,29 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2022 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.transport.lwm2m.server.client; |
|||
|
|||
import org.eclipse.leshan.core.model.ResourceModel; |
|||
|
|||
import java.io.Serializable; |
|||
|
|||
public class TbResourceModel extends ResourceModel implements Serializable { |
|||
|
|||
private static final long serialVersionUID = -2082846558899793932L; |
|||
|
|||
public TbResourceModel(Integer id, String name, Operations operations, Boolean multiple, Boolean mandatory, Type type, String rangeEnumeration, String units, String description) { |
|||
super(id, name, operations, multiple, mandatory, type, rangeEnumeration, units, description); |
|||
} |
|||
} |
|||
@ -0,0 +1,349 @@ |
|||
/** |
|||
* Copyright © 2016-2022 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.transport.lwm2m.server.store.util; |
|||
|
|||
import com.eclipsesource.json.Json; |
|||
import com.eclipsesource.json.JsonObject; |
|||
import com.eclipsesource.json.JsonValue; |
|||
import com.google.protobuf.util.JsonFormat; |
|||
import lombok.SneakyThrows; |
|||
import org.eclipse.leshan.core.model.ResourceModel; |
|||
import org.eclipse.leshan.core.node.LwM2mMultipleResource; |
|||
import org.eclipse.leshan.core.node.LwM2mNodeException; |
|||
import org.eclipse.leshan.core.node.LwM2mResource; |
|||
import org.eclipse.leshan.core.node.LwM2mSingleResource; |
|||
import org.eclipse.leshan.core.node.ObjectLink; |
|||
import org.eclipse.leshan.core.util.datatype.ULong; |
|||
import org.eclipse.leshan.server.redis.serialization.RegistrationSerDes; |
|||
import org.thingsboard.server.common.data.device.data.PowerMode; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.gen.transport.TransportProtos; |
|||
import org.thingsboard.server.transport.lwm2m.server.client.LwM2MClientState; |
|||
import org.thingsboard.server.transport.lwm2m.server.client.LwM2mClient; |
|||
import org.thingsboard.server.transport.lwm2m.server.client.ResourceValue; |
|||
|
|||
import java.lang.reflect.Field; |
|||
import java.util.Base64; |
|||
import java.util.Date; |
|||
import java.util.HashMap; |
|||
import java.util.Map; |
|||
import java.util.UUID; |
|||
import java.util.concurrent.atomic.AtomicLong; |
|||
|
|||
public class LwM2MClientSerDes { |
|||
public static final String VALUE = "value"; |
|||
|
|||
@SneakyThrows |
|||
public static byte[] serialize(LwM2mClient client) { |
|||
JsonObject o = Json.object(); |
|||
o.add("nodeId", client.getNodeId()); |
|||
o.add("endpoint", client.getEndpoint()); |
|||
|
|||
JsonObject resources = Json.object(); |
|||
client.getResources().forEach((k, v) -> { |
|||
JsonObject resourceValue = Json.object(); |
|||
resourceValue.add("lwM2mResource", serialize(v.getLwM2mResource())); |
|||
resourceValue.add("resourceModel", serialize(v.getResourceModel())); |
|||
resources.add(k, resourceValue); |
|||
}); |
|||
o.add("resources", resources); |
|||
JsonObject sharedAttributes = Json.object(); |
|||
|
|||
for (Map.Entry<String, TransportProtos.TsKvProto> entry : client.getSharedAttributes().entrySet()) { |
|||
sharedAttributes.add(entry.getKey(), JsonFormat.printer().print(entry.getValue())); |
|||
} |
|||
|
|||
o.add("sharedAttributes", sharedAttributes); |
|||
JsonObject keyTsLatestMap = Json.object(); |
|||
client.getKeyTsLatestMap().forEach((k, v) -> { |
|||
keyTsLatestMap.add(k, v.get()); |
|||
}); |
|||
o.add("keyTsLatestMap", keyTsLatestMap); |
|||
|
|||
o.add("state", client.getState().toString()); |
|||
|
|||
if (client.getSession() != null) { |
|||
o.add("session", JsonFormat.printer().print(client.getSession())); |
|||
} |
|||
if (client.getTenantId() != null) { |
|||
o.add("tenantId", client.getTenantId().toString()); |
|||
} |
|||
if (client.getDeviceId() != null) { |
|||
o.add("deviceId", client.getDeviceId().toString()); |
|||
} |
|||
if (client.getProfileId() != null) { |
|||
o.add("profileId", client.getProfileId().toString()); |
|||
} |
|||
if (client.getPowerMode() != null) { |
|||
o.add("powerMode", client.getPowerMode().toString()); |
|||
} |
|||
if (client.getEdrxCycle() != null) { |
|||
o.add("edrxCycle", client.getEdrxCycle()); |
|||
} |
|||
if (client.getPsmActivityTimer() != null) { |
|||
o.add("psmActivityTimer", client.getPsmActivityTimer()); |
|||
} |
|||
if (client.getPagingTransmissionWindow() != null) { |
|||
o.add("pagingTransmissionWindow", client.getPagingTransmissionWindow()); |
|||
} |
|||
if (client.getRegistration() != null) { |
|||
o.add("registration", RegistrationSerDes.jSerialize(client.getRegistration())); |
|||
} |
|||
o.add("asleep", client.isAsleep()); |
|||
o.add("lastUplinkTime", client.getLastUplinkTime()); |
|||
|
|||
Field firstEdrxDownlink = LwM2mClient.class.getDeclaredField("firstEdrxDownlink"); |
|||
firstEdrxDownlink.setAccessible(true); |
|||
o.add("firstEdrxDownlink", (boolean) firstEdrxDownlink.get(client)); |
|||
o.add("retryAttempts", client.getRetryAttempts().get()); |
|||
|
|||
if (client.getLastSentRpcId() != null) { |
|||
o.add("lastSentRpcId", client.getLastSentRpcId().toString()); |
|||
} |
|||
|
|||
return o.toString().getBytes(); |
|||
} |
|||
|
|||
private static JsonObject serialize(LwM2mResource resource) { |
|||
JsonObject o = Json.object(); |
|||
o.add("id", resource.getId()); |
|||
o.add("type", resource.getType().toString()); |
|||
if (resource.isMultiInstances()) { |
|||
o.add("multiInstances", true); |
|||
JsonObject instances = Json.object(); |
|||
resource.getInstances().forEach((id, in) -> { |
|||
JsonObject instance = Json.object(); |
|||
instance.add("id", in.getId()); |
|||
addValue(instance, in.getType(), in.getValue()); |
|||
instances.add(id.toString(), instance); |
|||
}); |
|||
o.add("instances", instances); |
|||
} else { |
|||
o.add("multiInstances", false); |
|||
addValue(o, resource.getType(), resource.getValue()); |
|||
} |
|||
|
|||
return o; |
|||
} |
|||
|
|||
private static LwM2mResource parseLwM2mResource(JsonObject o) { |
|||
boolean multiInstances = o.get("multiInstances").asBoolean(); |
|||
int id = o.get("id").asInt(); |
|||
ResourceModel.Type type = ResourceModel.Type.valueOf(o.get("type").asString()); |
|||
if (multiInstances) { |
|||
Map<Integer, Object> instances = new HashMap<>(); |
|||
o.get("instances").asObject().forEach(entry -> { |
|||
instances.put(Integer.valueOf(entry.getName()), parseValue(type, entry.getValue())); |
|||
}); |
|||
return LwM2mMultipleResource.newResource(id, instances, type); |
|||
} else { |
|||
return LwM2mSingleResource.newResource(id, parseValue(type, o.get(VALUE))); |
|||
} |
|||
} |
|||
|
|||
private static Object parseValue(ResourceModel.Type type, JsonValue value) { |
|||
switch (type) { |
|||
case INTEGER: |
|||
return value.asLong(); |
|||
case FLOAT: |
|||
return value.asDouble(); |
|||
case BOOLEAN: |
|||
return value.asBoolean(); |
|||
case OPAQUE: |
|||
return Base64.getDecoder().decode(value.asString()); |
|||
case STRING: |
|||
return value.asString(); |
|||
case TIME: |
|||
return new Date(value.asLong()); |
|||
case OBJLNK: |
|||
return ObjectLink.decodeFromString(value.asString()); |
|||
case UNSIGNED_INTEGER: |
|||
return ULong.valueOf(value.asString()); |
|||
default: |
|||
throw new LwM2mNodeException(String.format("Type %s is not supported", type.name())); |
|||
} |
|||
} |
|||
|
|||
private static JsonObject serialize(ResourceModel resourceModel) { |
|||
JsonObject o = Json.object(); |
|||
o.add("id", resourceModel.id); |
|||
o.add("name", resourceModel.name); |
|||
o.add("operations", resourceModel.operations.toString()); |
|||
o.add("multiple", resourceModel.multiple); |
|||
o.add("mandatory", resourceModel.mandatory); |
|||
o.add("type", resourceModel.type.toString()); |
|||
o.add("rangeEnumeration", resourceModel.rangeEnumeration); |
|||
o.add("units", resourceModel.units); |
|||
o.add("description", resourceModel.description); |
|||
return o; |
|||
} |
|||
|
|||
private static ResourceModel parseResourceModel(JsonObject o) { |
|||
Integer id = o.get("id").asInt(); |
|||
String name = o.get("name").asString(); |
|||
ResourceModel.Operations operations = ResourceModel.Operations.valueOf(o.get("operations").asString()); |
|||
Boolean multiple = o.get("multiple").asBoolean(); |
|||
Boolean mandatory = o.get("mandatory").asBoolean(); |
|||
ResourceModel.Type type = ResourceModel.Type.valueOf(o.get("type").asString()); |
|||
String rangeEnumeration = o.get("rangeEnumeration").asString(); |
|||
String units = o.get("units").asString(); |
|||
String description = o.get("description").asString(); |
|||
return new ResourceModel(id, name, operations, multiple, mandatory, type, rangeEnumeration, units, description); |
|||
} |
|||
|
|||
private static void addValue(JsonObject o, ResourceModel.Type type, Object value) { |
|||
switch (type) { |
|||
case INTEGER: |
|||
o.add(VALUE, (Long) value); |
|||
break; |
|||
case FLOAT: |
|||
o.add(VALUE, (Double) value); |
|||
break; |
|||
case BOOLEAN: |
|||
o.add(VALUE, (Boolean) value); |
|||
break; |
|||
case OPAQUE: |
|||
o.add(VALUE, Base64.getEncoder().encodeToString((byte[]) value)); |
|||
break; |
|||
case STRING: |
|||
o.add(VALUE, (String) value); |
|||
break; |
|||
case TIME: |
|||
o.add(VALUE, ((Date) value).getTime()); |
|||
break; |
|||
case OBJLNK: |
|||
o.add(VALUE, ((ObjectLink) value).encodeToString()); |
|||
break; |
|||
case UNSIGNED_INTEGER: |
|||
o.add(VALUE, value.toString()); |
|||
break; |
|||
default: |
|||
throw new LwM2mNodeException(String.format("Type %s is not supported", type.name())); |
|||
} |
|||
} |
|||
|
|||
@SneakyThrows |
|||
public static LwM2mClient deserialize(byte[] data) { |
|||
JsonObject o = Json.parse(new String(data)).asObject(); |
|||
LwM2mClient lwM2mClient = new LwM2mClient(o.get("nodeId").asString(), o.get("endpoint").asString()); |
|||
|
|||
o.get("resources").asObject().forEach(entry -> { |
|||
JsonObject resource = entry.getValue().asObject(); |
|||
LwM2mResource lwM2mResource = parseLwM2mResource(resource.get("lwM2mResource").asObject()); |
|||
ResourceModel resourceModel = parseResourceModel(resource.get("resourceModel").asObject()); |
|||
ResourceValue resourceValue = new ResourceValue(lwM2mResource, resourceModel); |
|||
lwM2mClient.getResources().put(entry.getName(), resourceValue); |
|||
}); |
|||
|
|||
for (JsonObject.Member entry : o.get("sharedAttributes").asObject()) { |
|||
TransportProtos.TsKvProto.Builder builder = TransportProtos.TsKvProto.newBuilder(); |
|||
JsonFormat.parser().merge(entry.getValue().asString(), builder); |
|||
lwM2mClient.getSharedAttributes().put(entry.getName(), builder.build()); |
|||
} |
|||
|
|||
o.get("keyTsLatestMap").asObject().forEach(entry -> { |
|||
lwM2mClient.getKeyTsLatestMap().put(entry.getName(), new AtomicLong(entry.getValue().asLong())); |
|||
}); |
|||
|
|||
lwM2mClient.setState(LwM2MClientState.valueOf(o.get("state").asString())); |
|||
|
|||
Class<LwM2mClient> lwM2mClientClass = LwM2mClient.class; |
|||
|
|||
JsonValue session = o.get("session"); |
|||
if (session != null) { |
|||
TransportProtos.SessionInfoProto.Builder builder = TransportProtos.SessionInfoProto.newBuilder(); |
|||
JsonFormat.parser().merge(session.asString(), builder); |
|||
|
|||
Field sessionField = lwM2mClientClass.getDeclaredField("session"); |
|||
sessionField.setAccessible(true); |
|||
sessionField.set(lwM2mClient, builder.build()); |
|||
} |
|||
|
|||
JsonValue tenantId = o.get("tenantId"); |
|||
if (tenantId != null) { |
|||
Field tenantIdField = lwM2mClientClass.getDeclaredField("tenantId"); |
|||
tenantIdField.setAccessible(true); |
|||
tenantIdField.set(lwM2mClient, new TenantId(UUID.fromString(tenantId.asString()))); |
|||
} |
|||
|
|||
JsonValue deviceId = o.get("deviceId"); |
|||
if (tenantId != null) { |
|||
Field deviceIdField = lwM2mClientClass.getDeclaredField("deviceId"); |
|||
deviceIdField.setAccessible(true); |
|||
deviceIdField.set(lwM2mClient, UUID.fromString(deviceId.asString())); |
|||
} |
|||
|
|||
JsonValue profileId = o.get("profileId"); |
|||
if (tenantId != null) { |
|||
Field profileIdField = lwM2mClientClass.getDeclaredField("profileId"); |
|||
profileIdField.setAccessible(true); |
|||
profileIdField.set(lwM2mClient, UUID.fromString(profileId.asString())); |
|||
} |
|||
|
|||
JsonValue powerMode = o.get("powerMode"); |
|||
if (powerMode != null) { |
|||
Field powerModeField = lwM2mClientClass.getDeclaredField("powerMode"); |
|||
powerModeField.setAccessible(true); |
|||
powerModeField.set(lwM2mClient, PowerMode.valueOf(powerMode.asString())); |
|||
} |
|||
|
|||
JsonValue edrxCycle = o.get("edrxCycle"); |
|||
if (edrxCycle != null) { |
|||
Field edrxCycleField = lwM2mClientClass.getDeclaredField("edrxCycle"); |
|||
edrxCycleField.setAccessible(true); |
|||
edrxCycleField.set(lwM2mClient, edrxCycle.asLong()); |
|||
} |
|||
|
|||
JsonValue psmActivityTimer = o.get("psmActivityTimer"); |
|||
if (psmActivityTimer != null) { |
|||
Field psmActivityTimerField = lwM2mClientClass.getDeclaredField("psmActivityTimer"); |
|||
psmActivityTimerField.setAccessible(true); |
|||
psmActivityTimerField.set(lwM2mClient, psmActivityTimer.asLong()); |
|||
} |
|||
|
|||
JsonValue pagingTransmissionWindow = o.get("pagingTransmissionWindow"); |
|||
if (pagingTransmissionWindow != null) { |
|||
Field pagingTransmissionWindowField = lwM2mClientClass.getDeclaredField("pagingTransmissionWindow"); |
|||
pagingTransmissionWindowField.setAccessible(true); |
|||
pagingTransmissionWindowField.set(lwM2mClient, pagingTransmissionWindow.asLong()); |
|||
} |
|||
|
|||
JsonValue registration = o.get("registration"); |
|||
if (registration != null) { |
|||
lwM2mClient.setRegistration(RegistrationSerDes.deserialize(registration.asObject())); |
|||
} |
|||
|
|||
lwM2mClient.setAsleep(o.get("asleep").asBoolean()); |
|||
|
|||
Field lastUplinkTimeField = lwM2mClientClass.getDeclaredField("lastUplinkTime"); |
|||
lastUplinkTimeField.setAccessible(true); |
|||
lastUplinkTimeField.setLong(lwM2mClient, o.get("lastUplinkTime").asLong()); |
|||
|
|||
Field firstEdrxDownlinkField = lwM2mClientClass.getDeclaredField("firstEdrxDownlink"); |
|||
firstEdrxDownlinkField.setAccessible(true); |
|||
firstEdrxDownlinkField.setBoolean(lwM2mClient, o.get("firstEdrxDownlink").asBoolean()); |
|||
|
|||
lwM2mClient.getRetryAttempts().set(o.get("retryAttempts").asInt()); |
|||
|
|||
JsonValue lastSentRpcId = o.get("lastSentRpcId"); |
|||
if (lastSentRpcId != null) { |
|||
lwM2mClient.setLastSentRpcId(UUID.fromString(lastSentRpcId.asString())); |
|||
} |
|||
|
|||
return lwM2mClient; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,101 @@ |
|||
/** |
|||
* Copyright © 2016-2022 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.transport.lwm2m.server.store.util; |
|||
|
|||
import org.eclipse.leshan.core.link.Link; |
|||
import org.eclipse.leshan.core.request.Identity; |
|||
import org.eclipse.leshan.server.registration.Registration; |
|||
import org.junit.Assert; |
|||
import org.junit.jupiter.api.Test; |
|||
import org.thingsboard.server.common.data.device.data.PowerMode; |
|||
import org.thingsboard.server.common.data.id.CustomerId; |
|||
import org.thingsboard.server.common.data.id.DeviceId; |
|||
import org.thingsboard.server.common.data.id.DeviceProfileId; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.common.transport.auth.TransportDeviceInfo; |
|||
import org.thingsboard.server.common.transport.auth.ValidateDeviceCredentialsResponse; |
|||
import org.thingsboard.server.gen.transport.TransportProtos; |
|||
import org.thingsboard.server.transport.lwm2m.server.client.LwM2MClientState; |
|||
import org.thingsboard.server.transport.lwm2m.server.client.LwM2mClient; |
|||
|
|||
import java.net.InetSocketAddress; |
|||
import java.util.UUID; |
|||
|
|||
class LwM2MClientSerDesTest { |
|||
|
|||
@Test |
|||
void serializeDeserialize() { |
|||
LwM2mClient client = new LwM2mClient("nodeId", "testEndpoint"); |
|||
|
|||
TransportDeviceInfo tdi = new TransportDeviceInfo(); |
|||
tdi.setPowerMode(PowerMode.PSM); |
|||
tdi.setPsmActivityTimer(10000L); |
|||
tdi.setPagingTransmissionWindow(2000L); |
|||
tdi.setEdrxCycle(3000L); |
|||
tdi.setTenantId(TenantId.fromUUID(UUID.randomUUID())); |
|||
tdi.setCustomerId(new CustomerId(UUID.randomUUID())); |
|||
tdi.setDeviceId(new DeviceId(UUID.randomUUID())); |
|||
tdi.setDeviceProfileId(new DeviceProfileId(UUID.randomUUID())); |
|||
tdi.setDeviceName("testDevice"); |
|||
tdi.setDeviceType("testType"); |
|||
ValidateDeviceCredentialsResponse credentialsResponse = ValidateDeviceCredentialsResponse.builder() |
|||
.deviceInfo(tdi) |
|||
.build(); |
|||
|
|||
client.init(credentialsResponse, UUID.randomUUID()); |
|||
|
|||
Registration registration = |
|||
new Registration.Builder("test", "testEndpoint", Identity |
|||
.unsecure(new InetSocketAddress(1000))) |
|||
.supportedContentFormats() |
|||
.objectLinks(new Link[]{new Link("/")}) |
|||
.build(); |
|||
|
|||
client.setRegistration(registration); |
|||
client.setState(LwM2MClientState.REGISTERED); |
|||
|
|||
client.getSharedAttributes().put("key1", TransportProtos.TsKvProto.newBuilder().setTs(0).setKv(TransportProtos.KeyValueProto.newBuilder().setStringV("test").build()).build()); |
|||
client.getSharedAttributes().put("key2", TransportProtos.TsKvProto.newBuilder().setTs(1).setKv(TransportProtos.KeyValueProto.newBuilder().setDoubleV(1.02).build()).build()); |
|||
|
|||
byte[] bytes = LwM2MClientSerDes.serialize(client); |
|||
Assert.assertNotNull(bytes); |
|||
|
|||
LwM2mClient desClient = LwM2MClientSerDes.deserialize(bytes); |
|||
|
|||
Assert.assertEquals(client.getNodeId(), desClient.getNodeId()); |
|||
Assert.assertEquals(client.getEndpoint(), desClient.getEndpoint()); |
|||
Assert.assertEquals(client.getResources(), desClient.getResources()); |
|||
Assert.assertEquals(client.getSharedAttributes(), desClient.getSharedAttributes()); |
|||
Assert.assertEquals(client.getKeyTsLatestMap(), desClient.getKeyTsLatestMap()); |
|||
Assert.assertEquals(client.getTenantId(), desClient.getTenantId()); |
|||
Assert.assertEquals(client.getProfileId(), desClient.getProfileId()); |
|||
Assert.assertEquals(client.getDeviceId(), desClient.getDeviceId()); |
|||
Assert.assertEquals(client.getState(), desClient.getState()); |
|||
Assert.assertEquals(client.getSession(), desClient.getSession()); |
|||
Assert.assertEquals(client.getPowerMode(), desClient.getPowerMode()); |
|||
Assert.assertEquals(client.getPsmActivityTimer(), desClient.getPsmActivityTimer()); |
|||
Assert.assertEquals(client.getPagingTransmissionWindow(), desClient.getPagingTransmissionWindow()); |
|||
Assert.assertEquals(client.getEdrxCycle(), desClient.getEdrxCycle()); |
|||
Assert.assertEquals(client.getRegistration(), desClient.getRegistration()); |
|||
Assert.assertEquals(client.isAsleep(), desClient.isAsleep()); |
|||
Assert.assertEquals(client.getLastUplinkTime(), desClient.getLastUplinkTime()); |
|||
Assert.assertEquals(client.getSleepTask(), desClient.getSleepTask()); |
|||
Assert.assertEquals(client.getClientSupportContentFormats(), desClient.getClientSupportContentFormats()); |
|||
Assert.assertEquals(client.getDefaultContentFormat(), desClient.getDefaultContentFormat()); |
|||
Assert.assertEquals(client.getRetryAttempts().get(), desClient.getRetryAttempts().get()); |
|||
Assert.assertEquals(client.getLastSentRpcId(), desClient.getLastSentRpcId()); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue