|  | // Copyright 2021 The Chromium Authors. All rights reserved. | 
|  | // Use of this source code is governed by a BSD-style license that can be | 
|  | // found in the LICENSE file. | 
|  |  | 
|  | #include "quic/core/quic_connection_context.h" | 
|  |  | 
|  | #include "common/platform/api/quiche_thread_local.h" | 
|  |  | 
|  | namespace quic { | 
|  | namespace { | 
|  | DEFINE_QUICHE_THREAD_LOCAL_POINTER(CurrentContext, QuicConnectionContext); | 
|  | }  // namespace | 
|  |  | 
|  | // static | 
|  | QuicConnectionContext* QuicConnectionContext::Current() { | 
|  | return GET_QUICHE_THREAD_LOCAL_POINTER(CurrentContext); | 
|  | } | 
|  |  | 
|  | QuicConnectionContextSwitcher::QuicConnectionContextSwitcher( | 
|  | QuicConnectionContext* new_context) | 
|  | : old_context_(QuicConnectionContext::Current()) { | 
|  | SET_QUICHE_THREAD_LOCAL_POINTER(CurrentContext, new_context); | 
|  | } | 
|  |  | 
|  | QuicConnectionContextSwitcher::~QuicConnectionContextSwitcher() { | 
|  | SET_QUICHE_THREAD_LOCAL_POINTER(CurrentContext, old_context_); | 
|  | } | 
|  |  | 
|  | }  // namespace quic |