FlowAdapter.java

Go to the documentation of this file.

/* Copyright 2018 Telstra Open Source
 *
 *   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.openkilda.pce.api;

import org.openkilda.messaging.Utils;
import org.openkilda.messaging.info.event.PathInfoData;
import org.openkilda.messaging.model.Flow;
import org.openkilda.messaging.model.SwitchId;
import org.openkilda.messaging.payload.flow.FlowState;

import org.neo4j.driver.v1.Record;

import java.io.IOException;

public class FlowAdapter {
    private final Flow flow;

    public FlowAdapter(Record dbRecord) {
        String pathJson = dbRecord.get("path").asString().trim();

        if (pathJson.equals("null")) {
            pathJson = "{\"path\": [], \"latency_ns\": 0, \"timestamp\": 0}";
        }

        /*
         * The 'clazz' value is stripped when storing in the database, but we need it in the string
         * in order for MAPPER to do its thing.  So, let's add it back in at the very beginning.
         */
        String start = pathJson.substring(0, pathJson.length() - 1);
        PathInfoData path;
        pathJson = start + ", \"clazz\":\"org.openkilda.messaging.info.event.PathInfoData\"}";

        try {
            path = Utils.MAPPER.readValue(pathJson, PathInfoData.class);
        } catch (IOException e) {
            throw new IllegalArgumentException(String.format(
                    "Can\'t deserialize flow path: json=%s", pathJson), e);
        }

        flow = new Flow(
                dbRecord.get(Utils.FLOW_ID).asString(),
                dbRecord.get("bandwidth").asInt(),
                dbRecord.get("ignore_bandwidth").asBoolean(),
                dbRecord.get("cookie").asLong(),
                dbRecord.get("description").asString(),
                dbRecord.get("last_updated").asString(),
                new SwitchId(dbRecord.get("src_switch").asString()),
                new SwitchId(dbRecord.get("dst_switch").asString()),
                dbRecord.get("src_port").asInt(),
                dbRecord.get("dst_port").asInt(),
                dbRecord.get("src_vlan").asInt(),
                dbRecord.get("dst_vlan").asInt(),
                dbRecord.get("meter_id").asInt(),
                dbRecord.get("transit_vlan").asInt(),
                path, FlowState.CACHED
        );
    }

    public Flow getFlow() {
        return flow;
    }
}