← Back to branch summary

~ubuntu-branches/ubuntu/wily/grpc/wily-proposed

« back to all changes in this revision

Viewing changes to src/cpp/client/client_context.cc

  • Committer: Package Import Robot
  • Author(s): Andrew Pollock
  • Date: 2015-05-07 13:28:11 UTC
  • Revision ID: package-import@ubuntu.com-20150507132811-ybm4hfq73tnvvd2e
Tags: upstream-0.10.0
Import upstream version 0.10.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/cpp/client/client_context.cc
 
1
/*
 
2
 *
 
3
 * Copyright 2015, Google Inc.
 
4
 * All rights reserved.
 
5
 *
 
6
 * Redistribution and use in source and binary forms, with or without
 
7
 * modification, are permitted provided that the following conditions are
 
8
 * met:
 
9
 *
 
10
 *     * Redistributions of source code must retain the above copyright
 
11
 * notice, this list of conditions and the following disclaimer.
 
12
 *     * Redistributions in binary form must reproduce the above
 
13
 * copyright notice, this list of conditions and the following disclaimer
 
14
 * in the documentation and/or other materials provided with the
 
15
 * distribution.
 
16
 *     * Neither the name of Google Inc. nor the names of its
 
17
 * contributors may be used to endorse or promote products derived from
 
18
 * this software without specific prior written permission.
 
19
 *
 
20
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 
21
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 
22
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 
23
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 
24
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 
25
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 
26
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 
27
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 
28
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 
29
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 
30
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
31
 *
 
32
 */
 
33
 
 
34
#include <grpc++/client_context.h>
 
35
 
 
36
#include <grpc/grpc.h>
 
37
#include <grpc++/credentials.h>
 
38
#include <grpc++/time.h>
 
39
#include "src/cpp/common/create_auth_context.h"
 
40
 
 
41
namespace grpc {
 
42
 
 
43
ClientContext::ClientContext()
 
44
    : initial_metadata_received_(false),
 
45
      call_(nullptr),
 
46
      cq_(nullptr),
 
47
      deadline_(gpr_inf_future) {}
 
48
 
 
49
ClientContext::~ClientContext() {
 
50
  if (call_) {
 
51
    grpc_call_destroy(call_);
 
52
  }
 
53
  if (cq_) {
 
54
    // Drain cq_.
 
55
    grpc_completion_queue_shutdown(cq_);
 
56
    while (grpc_completion_queue_next(cq_, gpr_inf_future).type !=
 
57
           GRPC_QUEUE_SHUTDOWN)
 
58
      ;
 
59
    grpc_completion_queue_destroy(cq_);
 
60
  }
 
61
}
 
62
 
 
63
void ClientContext::AddMetadata(const grpc::string& meta_key,
 
64
                                const grpc::string& meta_value) {
 
65
  send_initial_metadata_.insert(std::make_pair(meta_key, meta_value));
 
66
}
 
67
 
 
68
void ClientContext::set_call(grpc_call* call,
 
69
                             const std::shared_ptr<ChannelInterface>& channel) {
 
70
  GPR_ASSERT(call_ == nullptr);
 
71
  call_ = call;
 
72
  channel_ = channel;
 
73
  if (creds_ && !creds_->ApplyToCall(call_)) {
 
74
    grpc_call_cancel_with_status(call, GRPC_STATUS_CANCELLED,
 
75
                                 "Failed to set credentials to rpc.");
 
76
  }
 
77
}
 
78
 
 
79
std::shared_ptr<const AuthContext> ClientContext::auth_context() const {
 
80
  if (auth_context_.get() == nullptr) {
 
81
    auth_context_ = CreateAuthContext(call_);
 
82
  }
 
83
  return auth_context_;
 
84
}
 
85
 
 
86
void ClientContext::TryCancel() {
 
87
  if (call_) {
 
88
    grpc_call_cancel(call_);
 
89
  }
 
90
}
 
91
 
 
92
}  // namespace grpc