| 
									
										
										
										
											2023-05-09 19:19:48 +02:00
										 |  |  | /* | 
					
						
							|  |  |  |  * Copyright 2021 gRPC authors. | 
					
						
							|  |  |  |  * | 
					
						
							|  |  |  |  * 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 credentials | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import ( | 
					
						
							|  |  |  | 	"context" | 
					
						
							|  |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // requestInfoKey is a struct to be used as the key to store RequestInfo in a | 
					
						
							|  |  |  | // context. | 
					
						
							|  |  |  | type requestInfoKey struct{} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // NewRequestInfoContext creates a context with ri. | 
					
						
							| 
									
										
										
										
											2023-09-18 13:47:28 +01:00
										 |  |  | func NewRequestInfoContext(ctx context.Context, ri any) context.Context { | 
					
						
							| 
									
										
										
										
											2023-05-09 19:19:48 +02:00
										 |  |  | 	return context.WithValue(ctx, requestInfoKey{}, ri) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // RequestInfoFromContext extracts the RequestInfo from ctx. | 
					
						
							| 
									
										
										
										
											2023-09-18 13:47:28 +01:00
										 |  |  | func RequestInfoFromContext(ctx context.Context) any { | 
					
						
							| 
									
										
										
										
											2023-05-09 19:19:48 +02:00
										 |  |  | 	return ctx.Value(requestInfoKey{}) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // clientHandshakeInfoKey is a struct used as the key to store | 
					
						
							|  |  |  | // ClientHandshakeInfo in a context. | 
					
						
							|  |  |  | type clientHandshakeInfoKey struct{} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // ClientHandshakeInfoFromContext extracts the ClientHandshakeInfo from ctx. | 
					
						
							| 
									
										
										
										
											2023-09-18 13:47:28 +01:00
										 |  |  | func ClientHandshakeInfoFromContext(ctx context.Context) any { | 
					
						
							| 
									
										
										
										
											2023-05-09 19:19:48 +02:00
										 |  |  | 	return ctx.Value(clientHandshakeInfoKey{}) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // NewClientHandshakeInfoContext creates a context with chi. | 
					
						
							| 
									
										
										
										
											2023-09-18 13:47:28 +01:00
										 |  |  | func NewClientHandshakeInfoContext(ctx context.Context, chi any) context.Context { | 
					
						
							| 
									
										
										
										
											2023-05-09 19:19:48 +02:00
										 |  |  | 	return context.WithValue(ctx, clientHandshakeInfoKey{}, chi) | 
					
						
							|  |  |  | } |