com.google.appengine.repackaged.com.google.datastore.v1.client.Datastore Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of appengine-api-1.0-sdk
API for Google App Engine standard environment with some of the dependencies shaded (repackaged)
There is a newer version: 2.0.27
/*
 * Copyright 2015 Google LLC
 *
 * 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.google.datastore.v1.client;

import com.google.datastore.v1.AllocateIdsRequest;
import com.google.datastore.v1.AllocateIdsResponse;
import com.google.datastore.v1.BeginTransactionRequest;
import com.google.datastore.v1.BeginTransactionResponse;
import com.google.datastore.v1.CommitRequest;
import com.google.datastore.v1.CommitResponse;
import com.google.datastore.v1.LookupRequest;
import com.google.datastore.v1.LookupResponse;
import com.google.datastore.v1.ReserveIdsRequest;
import com.google.datastore.v1.ReserveIdsResponse;
import com.google.datastore.v1.RollbackRequest;
import com.google.datastore.v1.RollbackResponse;
import com.google.datastore.v1.RunAggregationQueryRequest;
import com.google.datastore.v1.RunAggregationQueryResponse;
import com.google.datastore.v1.RunQueryRequest;
import com.google.datastore.v1.RunQueryResponse;
import com.google.rpc.Code;
import java.io.IOException;
import java.io.InputStream;

/**
 * Provides access to Cloud Datastore.
 *
 * This class is thread-safe.
 */
public class Datastore {

  final RemoteRpc remoteRpc;

  Datastore(RemoteRpc remoteRpc) {
    this.remoteRpc = remoteRpc;
  }

  /** Reset the RPC count. */
  public void resetRpcCount() {
    remoteRpc.resetRpcCount();
  }

  /**
   * Returns the number of RPC calls made since the client was created or {@link #resetRpcCount} was
   * called.
   */
  public int getRpcCount() {
    return remoteRpc.getRpcCount();
  }

  private DatastoreException invalidResponseException(String method, IOException exception) {
    return RemoteRpc.makeException(
        remoteRpc.getUrl(), method, Code.UNAVAILABLE, "Invalid response", exception);
  }

  public AllocateIdsResponse allocateIds(AllocateIdsRequest request) throws DatastoreException {
    try (InputStream is = remoteRpc.call("allocateIds", request)) {
      return AllocateIdsResponse.parseFrom(is);
    } catch (IOException exception) {
      throw invalidResponseException("allocateIds", exception);
    }
  }

  public BeginTransactionResponse beginTransaction(BeginTransactionRequest request)
      throws DatastoreException {
    try (InputStream is = remoteRpc.call("beginTransaction", request)) {
      return BeginTransactionResponse.parseFrom(is);
    } catch (IOException exception) {
      throw invalidResponseException("beginTransaction", exception);
    }
  }

  public CommitResponse commit(CommitRequest request) throws DatastoreException {
    try (InputStream is = remoteRpc.call("commit", request)) {
      return CommitResponse.parseFrom(is);
    } catch (IOException exception) {
      throw invalidResponseException("commit", exception);
    }
  }

  public LookupResponse lookup(LookupRequest request) throws DatastoreException {
    try (InputStream is = remoteRpc.call("lookup", request)) {
      return LookupResponse.parseFrom(is);
    } catch (IOException exception) {
      throw invalidResponseException("lookup", exception);
    }
  }

  public ReserveIdsResponse reserveIds(ReserveIdsRequest request) throws DatastoreException {
    try (InputStream is = remoteRpc.call("reserveIds", request)) {
      return ReserveIdsResponse.parseFrom(is);
    } catch (IOException exception) {
      throw invalidResponseException("reserveIds", exception);
    }
  }

  public RollbackResponse rollback(RollbackRequest request) throws DatastoreException {
    try (InputStream is = remoteRpc.call("rollback", request)) {
      return RollbackResponse.parseFrom(is);
    } catch (IOException exception) {
      throw invalidResponseException("rollback", exception);
    }
  }

  public RunQueryResponse runQuery(RunQueryRequest request) throws DatastoreException {
    try (InputStream is = remoteRpc.call("runQuery", request)) {
      return RunQueryResponse.parseFrom(is);
    } catch (IOException exception) {
      throw invalidResponseException("runQuery", exception);
    }
  }

  public RunAggregationQueryResponse runAggregationQuery(RunAggregationQueryRequest request)
      throws DatastoreException {
    try (InputStream is = remoteRpc.call("runAggregationQuery", request)) {
      return RunAggregationQueryResponse.parseFrom(is);
    } catch (IOException exception) {
      throw invalidResponseException("runAggregationQuery", exception);
    }
  }
}