/*
* Copyright (c) 2002-2017 "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<GraphRowListModel> {
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;
}
}