Home | History | Annotate | Download | only in census
      1 /*
      2  *
      3  * Copyright 2015 gRPC authors.
      4  *
      5  * Licensed under the Apache License, Version 2.0 (the "License");
      6  * you may not use this file except in compliance with the License.
      7  * You may obtain a copy of the License at
      8  *
      9  *     http://www.apache.org/licenses/LICENSE-2.0
     10  *
     11  * Unless required by applicable law or agreed to in writing, software
     12  * distributed under the License is distributed on an "AS IS" BASIS,
     13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     14  * See the License for the specific language governing permissions and
     15  * limitations under the License.
     16  *
     17  */
     18 
     19 #include <grpc/support/port_platform.h>
     20 
     21 #include <grpc/census.h>
     22 #include <grpc/grpc.h>
     23 #include "src/core/lib/surface/api_trace.h"
     24 #include "src/core/lib/surface/call.h"
     25 
     26 void grpc_census_call_set_context(grpc_call* call, census_context* context) {
     27   GRPC_API_TRACE("grpc_census_call_set_context(call=%p, census_context=%p)", 2,
     28                  (call, context));
     29   if (context != nullptr) {
     30     grpc_call_context_set(call, GRPC_CONTEXT_TRACING, context, nullptr);
     31   }
     32 }
     33 
     34 census_context* grpc_census_call_get_context(grpc_call* call) {
     35   GRPC_API_TRACE("grpc_census_call_get_context(call=%p)", 1, (call));
     36   return static_cast<census_context*>(
     37       grpc_call_context_get(call, GRPC_CONTEXT_TRACING));
     38 }
     39