Vladyslav_Prykhodko
6 years ago
19 changed files with 256 additions and 87 deletions
@ -0,0 +1,35 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2020 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; |
||||
|
|
||||
|
import com.google.protobuf.ByteString; |
||||
|
import org.thingsboard.server.common.data.DeviceProfile; |
||||
|
import org.thingsboard.server.common.data.id.DeviceProfileId; |
||||
|
|
||||
|
import java.util.Optional; |
||||
|
|
||||
|
public interface TransportProfileCache { |
||||
|
|
||||
|
|
||||
|
DeviceProfile getOrCreate(DeviceProfileId id, ByteString profileBody); |
||||
|
|
||||
|
DeviceProfile get(DeviceProfileId id); |
||||
|
|
||||
|
void put(DeviceProfile profile); |
||||
|
|
||||
|
DeviceProfile put(ByteString profileBody); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,77 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2020 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.service; |
||||
|
|
||||
|
import com.google.protobuf.ByteString; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnExpression; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.DeviceProfile; |
||||
|
import org.thingsboard.server.common.data.id.DeviceProfileId; |
||||
|
import org.thingsboard.server.common.transport.TransportProfileCache; |
||||
|
import org.thingsboard.server.common.transport.util.DataDecodingEncodingService; |
||||
|
|
||||
|
import java.util.Optional; |
||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||
|
import java.util.concurrent.ConcurrentMap; |
||||
|
|
||||
|
@Slf4j |
||||
|
@Component |
||||
|
@ConditionalOnExpression("('${service.type:null}'=='monolith' && '${transport.api_enabled:true}'=='true') || '${service.type:null}'=='tb-transport'") |
||||
|
public class DefaultTransportProfileCache implements TransportProfileCache { |
||||
|
|
||||
|
private final ConcurrentMap<DeviceProfileId, DeviceProfile> deviceProfiles = new ConcurrentHashMap<>(); |
||||
|
|
||||
|
private final DataDecodingEncodingService dataDecodingEncodingService; |
||||
|
|
||||
|
public DefaultTransportProfileCache(DataDecodingEncodingService dataDecodingEncodingService) { |
||||
|
this.dataDecodingEncodingService = dataDecodingEncodingService; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public DeviceProfile getOrCreate(DeviceProfileId id, ByteString profileBody) { |
||||
|
DeviceProfile profile = deviceProfiles.get(id); |
||||
|
if (profile == null) { |
||||
|
Optional<DeviceProfile> deviceProfile = dataDecodingEncodingService.decode(profileBody.toByteArray()); |
||||
|
if (deviceProfile.isPresent()) { |
||||
|
profile = deviceProfile.get(); |
||||
|
deviceProfiles.put(id, profile); |
||||
|
} |
||||
|
} |
||||
|
return profile; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public DeviceProfile get(DeviceProfileId id) { |
||||
|
return deviceProfiles.get(id); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void put(DeviceProfile profile) { |
||||
|
deviceProfiles.put(profile.getId(), profile); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public DeviceProfile put(ByteString profileBody) { |
||||
|
Optional<DeviceProfile> deviceProfile = dataDecodingEncodingService.decode(profileBody.toByteArray()); |
||||
|
if (deviceProfile.isPresent()) { |
||||
|
put(deviceProfile.get()); |
||||
|
return deviceProfile.get(); |
||||
|
} else { |
||||
|
return null; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue