com.arangodb.impl.InternalCursorDriverImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of arangodb-java-driver Show documentation
ArangoDB Java Driver
There is a newer version: 7.15.0
/*
 * Copyright (C) 2012 tamtam180
 *
 * 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.arangodb.impl;

import java.util.Collections;
import java.util.Map;

import com.arangodb.ArangoConfigure;
import com.arangodb.ArangoException;
import com.arangodb.CursorResultSet;
import com.arangodb.entity.CursorEntity;
import com.arangodb.entity.DefaultEntity;
import com.arangodb.entity.EntityFactory;
import com.arangodb.http.HttpManager;
import com.arangodb.http.HttpResponseEntity;
import com.arangodb.util.MapBuilder;

/**
 * @author tamtam180 - kirscheless at gmail.com
 *
 */
public class InternalCursorDriverImpl extends BaseArangoDriverImpl implements com.arangodb.InternalCursorDriver {

  InternalCursorDriverImpl(ArangoConfigure configure, HttpManager httpManager) {
    super(configure, httpManager);
  }

  @Override
  public CursorEntity validateQuery(String database, String query) throws ArangoException {

    HttpResponseEntity res = httpManager.doPost(
      createEndpointUrl(baseUrl, database, "/_api/query"),
      null,
      EntityFactory.toJsonString(new MapBuilder("query", query).get()));
    try {
      CursorEntity entity = createEntity(res, CursorEntity.class);
      return entity;
    } catch (ArangoException e) {
      return (CursorEntity) e.getEntity();
    }

  }

  // ※Iteratorで綺麗に何回もRoundtripもしてくれる処理はClientのレイヤーで行う。
  // ※ここでは単純にコールするだけ

  @Override
  public  CursorEntity executeQuery(
    String database,
    String query,
    Map bindVars,
    Class clazz,
    Boolean calcCount,
    Integer batchSize,
    Boolean fullCount) throws ArangoException {

    HttpResponseEntity res = httpManager.doPost(
      createEndpointUrl(baseUrl, database, "/_api/cursor"),
      null,
      EntityFactory.toJsonString(new MapBuilder().put("query", query)
          .put("bindVars", bindVars == null ? Collections.emptyMap() : bindVars).put("count", calcCount)
          .put("batchSize", batchSize).put("options", new MapBuilder().put("fullCount", fullCount).get()).get()));
    try {
      CursorEntity entity = createEntity(res, CursorEntity.class, clazz);
      // resultを処理する
      // EntityFactory.createResult(entity, clazz);
      return entity;
    } catch (ArangoException e) {
      throw e;
    }

  }

  // ※Iteratorで綺麗に何回もRoundtripもしてくれる処理はClientのレイヤーで行う。
  // ※ここでは単純にコールするだけ

  @Override
  public  CursorEntity executeQuery(
    String database,
    String query,
    Map bindVars,
    Class clazz,
    Boolean calcCount,
    Integer batchSize) throws ArangoException {

    return executeQuery(database, query, bindVars, clazz, calcCount, batchSize, false);

  }

  @Override
  public  CursorEntity continueQuery(String database, long cursorId, Class... clazz) throws ArangoException {

    HttpResponseEntity res = httpManager.doPut(
      createEndpointUrl(baseUrl, database, "/_api/cursor", cursorId),
      null,
      null);

    try {
      CursorEntity entity = createEntity(res, CursorEntity.class, clazz);
      // resultを処理する
      // EntityFactory.createResult(entity, clazz);
      return entity;
    } catch (ArangoException e) {
      throw e;
    }

  }

  @Override
  public DefaultEntity finishQuery(String database, long cursorId) throws ArangoException {
    HttpResponseEntity res = httpManager
        .doDelete(createEndpointUrl(baseUrl, database, "/_api/cursor/", cursorId), null);

    try {
      DefaultEntity entity = createEntity(res, DefaultEntity.class);
      return entity;
    } catch (ArangoException e) {
      // TODO Mode
      if (e.getErrorNumber() == 1600) {
        // 既に削除されている
        return (DefaultEntity) e.getEntity();
      }
      throw e;
    }
  }

  @Override
  public  CursorResultSet executeQueryWithResultSet(
    String database,
    String query,
    Map bindVars,
    Class clazz,
    Boolean calcCount,
    Integer batchSize,
    Boolean fullCount) throws ArangoException {

    CursorEntity entity = executeQuery(database, query, bindVars, clazz, calcCount, batchSize, fullCount);
    CursorResultSet rs = new CursorResultSet(database, this, entity, clazz);
    return rs;

  }

  @Override
  public  CursorResultSet executeQueryWithResultSet(
    String database,
    String query,
    Map bindVars,
    Class clazz,
    Boolean calcCount,
    Integer batchSize) throws ArangoException {

    CursorEntity entity = executeQuery(database, query, bindVars, clazz, calcCount, batchSize, false);
    CursorResultSet rs = new CursorResultSet(database, this, entity, clazz);
    return rs;

  }

}