@java.lang.Override public Builder newBuilderForType() { return newBuilder(); }
private com.google.datastore.v1.CommitResponse commitMutation( List<com.google.datastore.v1.Mutation> mutationsPb) { com.google.datastore.v1.CommitRequest.Builder requestPb = com.google.datastore.v1.CommitRequest.newBuilder(); requestPb.setMode(com.google.datastore.v1.CommitRequest.Mode.NON_TRANSACTIONAL); requestPb.addAllMutations(mutationsPb); return commit(requestPb.build()); }
@Override public Batch.Response submit() { validateActive(); List<com.google.datastore.v1.Mutation> mutationsPb = toMutationPbList(); com.google.datastore.v1.CommitRequest.Builder requestPb = com.google.datastore.v1.CommitRequest.newBuilder(); requestPb.setMode(com.google.datastore.v1.CommitRequest.Mode.NON_TRANSACTIONAL); requestPb.addAllMutations(mutationsPb); com.google.datastore.v1.CommitResponse responsePb = datastore.commit(requestPb.build()); deactivate(); return new ResponseImpl(responsePb, toAddAutoId().size()); }
@Override public Transaction.Response commit() { validateActive(); List<com.google.datastore.v1.Mutation> mutationsPb = toMutationPbList(); com.google.datastore.v1.CommitRequest.Builder requestPb = com.google.datastore.v1.CommitRequest.newBuilder(); requestPb.setMode(com.google.datastore.v1.CommitRequest.Mode.TRANSACTIONAL); requestPb.setTransaction(transactionId); requestPb.addAllMutations(mutationsPb); com.google.datastore.v1.CommitResponse responsePb = datastore.commit(requestPb.build()); deactivate(); return new ResponseImpl(responsePb, toAddAutoId().size()); }
CommitRequest.Builder commitRequest = CommitRequest.newBuilder(); commitRequest.setMode(Mode.NON_TRANSACTIONAL);
public Builder newBuilderForType() { return newBuilder(); } public static Builder newBuilder() {
@java.lang.Override public Builder newBuilderForType() { return newBuilder(); }
private com.google.datastore.v1.CommitResponse commitMutation( List<com.google.datastore.v1.Mutation> mutationsPb) { com.google.datastore.v1.CommitRequest.Builder requestPb = com.google.datastore.v1.CommitRequest.newBuilder(); requestPb.setMode(com.google.datastore.v1.CommitRequest.Mode.NON_TRANSACTIONAL); requestPb.addAllMutations(mutationsPb); return commit(requestPb.build()); }
CommitRequest.Builder commitRequest = CommitRequest.newBuilder(); commitRequest.addAllMutations(mutations); commitRequest.setMode(CommitRequest.Mode.NON_TRANSACTIONAL);
@Override public Batch.Response submit() { validateActive(); List<com.google.datastore.v1.Mutation> mutationsPb = toMutationPbList(); com.google.datastore.v1.CommitRequest.Builder requestPb = com.google.datastore.v1.CommitRequest.newBuilder(); requestPb.setMode(com.google.datastore.v1.CommitRequest.Mode.NON_TRANSACTIONAL); requestPb.addAllMutations(mutationsPb); com.google.datastore.v1.CommitResponse responsePb = datastore.commit(requestPb.build()); deactivate(); return new ResponseImpl(responsePb, toAddAutoId().size()); }
@Override public Transaction.Response commit() { validateActive(); List<com.google.datastore.v1.Mutation> mutationsPb = toMutationPbList(); com.google.datastore.v1.CommitRequest.Builder requestPb = com.google.datastore.v1.CommitRequest.newBuilder(); requestPb.setMode(com.google.datastore.v1.CommitRequest.Mode.TRANSACTIONAL); requestPb.setTransaction(transactionId); requestPb.addAllMutations(mutationsPb); com.google.datastore.v1.CommitResponse responsePb = datastore.commit(requestPb.build()); deactivate(); return new ResponseImpl(responsePb, toAddAutoId().size()); }
/** * Insert an entity into the datastore. * * The entity must have no ids. * * @return The key for the inserted entity. * @throws DatastoreException on error */ private Key insert(Entity entity) throws DatastoreException { CommitRequest req = CommitRequest.newBuilder() .addMutations(Mutation.newBuilder() .setInsert(entity)) .setMode(CommitRequest.Mode.NON_TRANSACTIONAL) .build(); return datastore.commit(req).getMutationResults(0).getKey(); }
CommitRequest.Builder creq = CommitRequest.newBuilder();
private ListenableFuture<MutationResult> executeAsyncMutations(final List<Mutation> mutations, final ListenableFuture<TransactionResult> txn) { final ListenableFuture<Response> httpResponse = Futures.transformAsync(txn, result -> { final CommitRequest.Builder request = CommitRequest.newBuilder(); if (mutations != null) { request.addAllMutations(mutations); } final ByteString transaction = result.getTransaction(); if (transaction != null) { request.setTransaction(transaction); } else { request.setMode(CommitRequest.Mode.NON_TRANSACTIONAL); } final ProtoHttpContent payload = new ProtoHttpContent(request.build()); return ListenableFutureAdapter.asGuavaFuture(prepareRequest("commit", payload).execute()); }, MoreExecutors.directExecutor()); return Futures.transformAsync(httpResponse, response -> { if (!isSuccessful(response.getStatusCode())) { throw new DatastoreException(response.getStatusCode(), response.getResponseBody()); } final CommitResponse commit = CommitResponse.parseFrom(streamResponse(response)); return Futures.immediateFuture(MutationResult.build(commit)); }, MoreExecutors.directExecutor()); }
CommitRequest.Builder commitRequest = CommitRequest.newBuilder(); for (Entity entity : entities) { commitRequest.addMutations(mutationBuilder.apply(entity));
private void datastoreWriterFnTest(int numMutations) throws Exception { // Create the requested number of mutations. List<Mutation> mutations = new ArrayList<>(numMutations); for (int i = 0; i < numMutations; ++i) { mutations.add( makeUpsert(Entity.newBuilder().setKey(makeKey("key" + i, i + 1)).build()).build()); } DatastoreWriterFn datastoreWriter = new DatastoreWriterFn( StaticValueProvider.of(PROJECT_ID), null, mockDatastoreFactory, new FakeWriteBatcher()); DoFnTester<Mutation, Void> doFnTester = DoFnTester.of(datastoreWriter); doFnTester.setCloningBehavior(CloningBehavior.DO_NOT_CLONE); doFnTester.processBundle(mutations); int start = 0; while (start < numMutations) { int end = Math.min(numMutations, start + DatastoreV1.DATASTORE_BATCH_UPDATE_ENTITIES_START); CommitRequest.Builder commitRequest = CommitRequest.newBuilder(); commitRequest.setMode(CommitRequest.Mode.NON_TRANSACTIONAL); commitRequest.addAllMutations(mutations.subList(start, end)); // Verify all the batch requests were made with the expected mutations. verify(mockDatastore, times(1)).commit(commitRequest.build()); start = end; } }