org.neo4j.ogm.drivers.embedded.response.GraphRowModelResponse Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2002-2016 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This product is licensed to you under the Apache License, Version 2.0 (the "License").
 * You may not use this product except in compliance with the License.
 *
 * This product may include a number of subcomponents with
 * separate copyright notices and license terms. Your use of the source
 * code for these subcomponents is subject to the terms and
 *  conditions of the subcomponent's license, as noted in the LICENSE file.
 */

package org.neo4j.ogm.drivers.embedded.response;

import org.neo4j.graphdb.Result;
import org.neo4j.ogm.model.GraphRowListModel;
import org.neo4j.ogm.response.model.DefaultGraphRowListModel;
import org.neo4j.ogm.transaction.TransactionManager;

/**
 * @author vince
 */
public class GraphRowModelResponse extends EmbeddedResponse {

    private EmbeddedGraphRowModelAdapter adapter = new EmbeddedGraphRowModelAdapter(new EmbeddedGraphModelAdapter());

    public GraphRowModelResponse(Result result, TransactionManager transactionManager) {
        super(result, transactionManager);
        adapter.setColumns(result.columns());
    }

    @Override
    public GraphRowListModel next() {

        if (result.hasNext()) {
            DefaultGraphRowListModel model = new DefaultGraphRowListModel();

            while (result.hasNext()) {
                model.add(adapter.adapt(result.next()));
            }
            return model;
        }
        return null;
    }
}