com.hurence.logisland.util.kura.KuraPayloadEncoder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of logisland-utils Show documentation
There is a newer version: 1.4.1
/**
 * Copyright (C) 2016 Hurence ([email protected])
 *
 * 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 com.hurence.logisland.util.kura;
/*******************************************************************************
 * Copyright (C) 2015 - Amit Kumar Mondal 
 *
 * 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.
 *******************************************************************************/


import com.google.protobuf.ByteString;
import org.eclipse.kura.core.message.protobuf.KuraPayloadProto;

import java.io.IOException;

/**
 * Encodes an KuraPayload class using the Google ProtoBuf binary format.
 */
public class KuraPayloadEncoder {

    private static void setProtoKuraMetricValue(final KuraPayloadProto.KuraPayload.KuraMetric.Builder metric,
                                                final Object o) throws Exception {

        if (o instanceof String) {
            metric.setType(KuraPayloadProto.KuraPayload.KuraMetric.ValueType.STRING);
            metric.setStringValue((String) o);
        } else if (o instanceof Double) {
            metric.setType(KuraPayloadProto.KuraPayload.KuraMetric.ValueType.DOUBLE);
            metric.setDoubleValue((Double) o);
        } else if (o instanceof Integer) {
            metric.setType(KuraPayloadProto.KuraPayload.KuraMetric.ValueType.INT32);
            metric.setIntValue((Integer) o);
        } else if (o instanceof Float) {
            metric.setType(KuraPayloadProto.KuraPayload.KuraMetric.ValueType.FLOAT);
            metric.setFloatValue((Float) o);
        } else if (o instanceof Long) {
            metric.setType(KuraPayloadProto.KuraPayload.KuraMetric.ValueType.INT64);
            metric.setLongValue((Long) o);
        } else if (o instanceof Boolean) {
            metric.setType(KuraPayloadProto.KuraPayload.KuraMetric.ValueType.BOOL);
            metric.setBoolValue((Boolean) o);
        } else if (o instanceof byte[]) {
            metric.setType(KuraPayloadProto.KuraPayload.KuraMetric.ValueType.BYTES);
            metric.setBytesValue(ByteString.copyFrom((byte[]) o));
        } else if (o == null) {
            throw new Exception("null value");
        } else {
            throw new Exception(o.getClass().getName());
        }
    }

    private final KuraPayload m_kuraPayload;

    public KuraPayloadEncoder(final KuraPayload KuraPayload) {
        this.m_kuraPayload = KuraPayload;
    }

    private KuraPayloadProto.KuraPayload.KuraPosition buildPositionProtoBuf() {
        KuraPayloadProto.KuraPayload.KuraPosition.Builder protoPos = null;
        protoPos = KuraPayloadProto.KuraPayload.KuraPosition.newBuilder();

        final KuraPosition position = this.m_kuraPayload.getPosition();
        if (position.getLatitude() != null) {
            protoPos.setLatitude(position.getLatitude());
        }
        if (position.getLongitude() != null) {
            protoPos.setLongitude(position.getLongitude());
        }
        if (position.getAltitude() != null) {
            protoPos.setAltitude(position.getAltitude());
        }
        if (position.getPrecision() != null) {
            protoPos.setPrecision(position.getPrecision());
        }
        if (position.getHeading() != null) {
            protoPos.setHeading(position.getHeading());
        }
        if (position.getSpeed() != null) {
            protoPos.setSpeed(position.getSpeed());
        }
        if (position.getTimestamp() != null) {
            protoPos.setTimestamp(position.getTimestamp().getTime());
        }
        if (position.getSatellites() != null) {
            protoPos.setSatellites(position.getSatellites());
        }
        if (position.getStatus() != null) {
            protoPos.setStatus(position.getStatus());
        }
        return protoPos.build();
    }

    public byte[] getBytes() throws IOException {
        final KuraPayloadProto.KuraPayload.Builder protoMsg = KuraPayloadProto.KuraPayload.newBuilder();

        if (this.m_kuraPayload.getTimestamp() != null) {
            protoMsg.setTimestamp(this.m_kuraPayload.getTimestamp().getTime());
        }

        if (this.m_kuraPayload.getPosition() != null) {
            protoMsg.setPosition(this.buildPositionProtoBuf());
        }

        for (final String name : this.m_kuraPayload.metricNames()) {

            final Object value = this.m_kuraPayload.getMetric(name);
            try {
                final KuraPayloadProto.KuraPayload.KuraMetric.Builder metricB = KuraPayloadProto.KuraPayload.KuraMetric
                        .newBuilder();
                metricB.setName(name);

                setProtoKuraMetricValue(metricB, value);
                metricB.build();

                protoMsg.addMetric(metricB);
            } catch (final Exception eihte) {
                try {
                    System.err.println("During serialization, ignoring metric named:  " + name
                            + "  . Unrecognized value type: " + value.getClass().getName() + ".");
                } catch (final NullPointerException npe) {
                    System.err.println("During serialization, ignoring metric named: " + name + ". The value is null.");
                }
                throw new RuntimeException(eihte);
            }
        }

        if (this.m_kuraPayload.getBody() != null) {
            protoMsg.setBody(ByteString.copyFrom(this.m_kuraPayload.getBody()));
        }

        return protoMsg.build().toByteArray();
    }
}