Andrew Shvayka
6 years ago
committed by
GitHub
GitHub
30 changed files with 395 additions and 126 deletions
@ -0,0 +1,37 @@ |
|||
/** |
|||
* Copyright © 2016-2021 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.data.transport.resource; |
|||
|
|||
import lombok.Data; |
|||
import org.thingsboard.server.common.data.HasTenantId; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
|
|||
@Data |
|||
public class Resource implements HasTenantId { |
|||
private TenantId tenantId; |
|||
private ResourceType resourceType; |
|||
private String resourceId; |
|||
private String value; |
|||
|
|||
@Override |
|||
public String toString() { |
|||
return "Resource{" + |
|||
"tenantId=" + tenantId + |
|||
", resourceType=" + resourceType + |
|||
", resourceId='" + resourceId + '\'' + |
|||
'}'; |
|||
} |
|||
} |
|||
@ -0,0 +1,20 @@ |
|||
/** |
|||
* Copyright © 2016-2021 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.data.transport.resource; |
|||
|
|||
public enum ResourceType { |
|||
LWM2M_MODEL, JKS, PKCS_12 |
|||
} |
|||
@ -0,0 +1,136 @@ |
|||
/** |
|||
* Copyright © 2016-2021 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; |
|||
|
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.eclipse.leshan.core.model.DefaultDDFFileValidator; |
|||
import org.eclipse.leshan.core.model.LwM2mModel; |
|||
import org.eclipse.leshan.core.model.ObjectModel; |
|||
import org.eclipse.leshan.core.model.ResourceModel; |
|||
import org.eclipse.leshan.server.model.LwM2mModelProvider; |
|||
import org.eclipse.leshan.server.registration.Registration; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.transport.lwm2m.server.client.LwM2mClientContext; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.Base64; |
|||
import java.util.Collection; |
|||
import java.util.Iterator; |
|||
import java.util.Map; |
|||
|
|||
import static org.thingsboard.server.common.data.ResourceType.LWM2M_MODEL; |
|||
|
|||
@Slf4j |
|||
public class LwM2mVersionedModelProvider implements LwM2mModelProvider { |
|||
|
|||
/** |
|||
* int objectId |
|||
* String version ("1.01") |
|||
* Key = objectId + "##" + version |
|||
* Value = TenantId |
|||
*/ |
|||
private final LwM2mClientContext lwM2mClientContext; |
|||
private final LwM2mTransportContextServer lwM2mTransportContextServer; |
|||
|
|||
public LwM2mVersionedModelProvider(LwM2mClientContext lwM2mClientContext, LwM2mTransportContextServer lwM2mTransportContextServer) { |
|||
this.lwM2mClientContext = lwM2mClientContext; |
|||
this.lwM2mTransportContextServer = lwM2mTransportContextServer; |
|||
} |
|||
private String getIdVer(ObjectModel objectModel) { |
|||
return objectModel.id + "##" + ((objectModel.getVersion() == null || objectModel.getVersion().isEmpty()) ? ObjectModel.DEFAULT_VERSION : objectModel.getVersion()); |
|||
} |
|||
|
|||
private String getIdVer(Integer objectId, String version) { |
|||
return objectId != null ? objectId + "##" + ((version == null || version.isEmpty()) ? ObjectModel.DEFAULT_VERSION : version) : null; |
|||
} |
|||
|
|||
/** |
|||
* Update repository if need |
|||
* |
|||
* @param registration |
|||
* @return |
|||
*/ |
|||
@Override |
|||
public LwM2mModel getObjectModel(Registration registration) { |
|||
return new DynamicModel(registration |
|||
); |
|||
} |
|||
|
|||
private class DynamicModel implements LwM2mModel { |
|||
|
|||
private final Registration registration; |
|||
private final TenantId tenantId; |
|||
|
|||
public DynamicModel(Registration registration) { |
|||
this.registration = registration; |
|||
this.tenantId = lwM2mClientContext.getProfile(registration).getTenantId(); |
|||
} |
|||
|
|||
@Override |
|||
public ResourceModel getResourceModel(int objectId, int resourceId) { |
|||
try { |
|||
ObjectModel objectModel = getObjectModel(objectId); |
|||
if (objectModel != null) |
|||
return objectModel.resources.get(resourceId); |
|||
else |
|||
return null; |
|||
} catch (Exception e) { |
|||
log.error("", e); |
|||
return null; |
|||
} |
|||
} |
|||
|
|||
@Override |
|||
public ObjectModel getObjectModel(int objectId) { |
|||
String version = registration.getSupportedVersion(objectId); |
|||
if (version != null) { |
|||
return this.getObjectModelDynamic(objectId, version); |
|||
} |
|||
return null; |
|||
} |
|||
|
|||
@Override |
|||
public Collection<ObjectModel> getObjectModels() { |
|||
Map<Integer, String> supportedObjects = this.registration.getSupportedObject(); |
|||
Collection<ObjectModel> result = new ArrayList(supportedObjects.size()); |
|||
Iterator i$ = supportedObjects.entrySet().iterator(); |
|||
|
|||
while (i$.hasNext()) { |
|||
Map.Entry<Integer, String> supportedObject = (Map.Entry) i$.next(); |
|||
ObjectModel objectModel = this.getObjectModelDynamic((Integer) supportedObject.getKey(), (String) supportedObject.getValue()); |
|||
if (objectModel != null) { |
|||
result.add(objectModel); |
|||
} |
|||
} |
|||
return result; |
|||
} |
|||
|
|||
private ObjectModel getObjectModelDynamic(Integer objectId, String version) { |
|||
String key = getIdVer(objectId, version); |
|||
String xmlB64 = lwM2mTransportContextServer.getTransportResourceCache().get( |
|||
this.tenantId, |
|||
LWM2M_MODEL, |
|||
key). |
|||
getValue(); |
|||
return xmlB64 != null && !xmlB64.isEmpty() ? |
|||
lwM2mTransportContextServer.parseFromXmlToObjectModel( |
|||
Base64.getDecoder().decode(xmlB64), |
|||
key + ".xml", |
|||
new DefaultDDFFileValidator()) : |
|||
null; |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue