Examples with SnmpMapping org.thingsboard.server.common.data.transport.snmp.SnmpMapping used on opensource projects

Example 1 with SnmpMapping

use of org.thingsboard.server.common.data.transport.snmp.SnmpMapping in project thingsboard by thingsboard.

the class SnmpTransportService method configureResponseDataMappers.

private void configureResponseDataMappers() {
    responseDataMappers.put(SnmpCommunicationSpec.TO_DEVICE_RPC_REQUEST, (pdu, requestInfo) -> {
        JsonObject responseData = new JsonObject();
        pduService.processPdu(pdu).forEach((oid, value) -> {
            requestInfo.getResponseMappings().stream().filter(snmpMapping -> snmpMapping.getOid().equals(oid.toDottedString())).findFirst().ifPresent(snmpMapping -> {
                pduService.processValue(snmpMapping.getKey(), snmpMapping.getDataType(), value, responseData);
            });
        });
        return responseData;
    });
    ResponseDataMapper defaultResponseDataMapper = (pdu, requestInfo) -> {
        return pduService.processPdu(pdu, requestInfo.getResponseMappings());
    };
    Arrays.stream(SnmpCommunicationSpec.values()).forEach(communicationSpec -> {
        responseDataMappers.putIfAbsent(communicationSpec, defaultResponseDataMapper);
    });
}

Example 2 with SnmpMapping

use of org.thingsboard.server.common.data.transport.snmp.SnmpMapping in project thingsboard by thingsboard.

the class PduService method processPdu.

public JsonObject processPdu(PDU pdu, List<SnmpMapping> responseMappings) {
    Map<OID, String> values = processPdu(pdu);
    Map<OID, SnmpMapping> mappings = new HashMap<>();
    if (responseMappings != null) {
        for (SnmpMapping mapping : responseMappings) {
            OID oid = new OID(mapping.getOid());
            mappings.put(oid, mapping);
        }
    }
    JsonObject data = new JsonObject();
    values.forEach((oid, value) -> {
        log.trace("Processing variable binding: {} - {}", oid, value);
        SnmpMapping mapping = mappings.get(oid);
        if (mapping == null) {
            log.debug("No SNMP mapping for oid {}", oid);
            return;
        }
        processValue(mapping.getKey(), mapping.getDataType(), value, data);
    });
    return data;
}

Example 3 with SnmpMapping

use of org.thingsboard.server.common.data.transport.snmp.SnmpMapping in project thingsboard by thingsboard.

the class SnmpTransportService method onToDeviceRpcRequest.

public void onToDeviceRpcRequest(DeviceSessionContext sessionContext, TransportProtos.ToDeviceRpcRequestMsg toDeviceRpcRequestMsg) {
    SnmpMethod snmpMethod = SnmpMethod.valueOf(toDeviceRpcRequestMsg.getMethodName());
    JsonObject params = JsonConverter.parse(toDeviceRpcRequestMsg.getParams()).getAsJsonObject();
    String key = Optional.ofNullable(params.get("key")).map(JsonElement::getAsString).orElse(null);
    String value = Optional.ofNullable(params.get("value")).map(JsonElement::getAsString).orElse(null);
    if (value == null && snmpMethod == SnmpMethod.SET) {
        throw new IllegalArgumentException("Value must be specified for SNMP method 'SET'");
    }
    SnmpCommunicationConfig communicationConfig = sessionContext.getProfileTransportConfiguration().getCommunicationConfigs().stream().filter(config -> config.getSpec() == SnmpCommunicationSpec.TO_DEVICE_RPC_REQUEST).findFirst().orElseThrow(() -> new IllegalArgumentException("No communication config found with RPC spec"));
    SnmpMapping snmpMapping = communicationConfig.getAllMappings().stream().filter(mapping -> mapping.getKey().equals(key)).findFirst().orElseThrow(() -> new IllegalArgumentException("No SNMP mapping found in the config for specified key"));
    String oid = snmpMapping.getOid();
    DataType dataType = snmpMapping.getDataType();
    PDU request = pduService.createSingleVariablePdu(sessionContext, snmpMethod, oid, value, dataType);
    RequestInfo requestInfo = new RequestInfo(toDeviceRpcRequestMsg.getRequestId(), communicationConfig.getSpec(), communicationConfig.getAllMappings());
    sendRequest(sessionContext, request, requestInfo);
}