|  | @@ -16,12 +16,6 @@
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  #endregion
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -using System;
 | 
	
		
			
				|  |  | -using System.Threading;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -using Grpc.Core.Internal;
 | 
	
		
			
				|  |  | -using Grpc.Core.Utils;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  |  namespace Grpc.Core
 | 
	
		
			
				|  |  |  {
 | 
	
		
			
				|  |  |      /// <summary>
 | 
	
	
		
			
				|  | @@ -29,127 +23,13 @@ namespace Grpc.Core
 | 
	
		
			
				|  |  |      /// In situations when a backend is making calls to another backend,
 | 
	
		
			
				|  |  |      /// it makes sense to propagate properties like deadline and cancellation 
 | 
	
		
			
				|  |  |      /// token of the server call to the child call.
 | 
	
		
			
				|  |  | -    /// The gRPC native layer provides some other contexts (like tracing context) that
 | 
	
		
			
				|  |  | -    /// are not accessible to explicitly C# layer, but this token still allows propagating them.
 | 
	
		
			
				|  |  | -    /// </summary>
 | 
	
		
			
				|  |  | -    public class ContextPropagationToken
 | 
	
		
			
				|  |  | -    {
 | 
	
		
			
				|  |  | -        /// <summary>
 | 
	
		
			
				|  |  | -        /// Default propagation mask used by C core.
 | 
	
		
			
				|  |  | -        /// </summary>
 | 
	
		
			
				|  |  | -        private const ContextPropagationFlags DefaultCoreMask = (ContextPropagationFlags)0xffff;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /// <summary>
 | 
	
		
			
				|  |  | -        /// Default propagation mask used by C# - we want to propagate deadline 
 | 
	
		
			
				|  |  | -        /// and cancellation token by our own means.
 | 
	
		
			
				|  |  | -        /// </summary>
 | 
	
		
			
				|  |  | -        internal const ContextPropagationFlags DefaultMask = DefaultCoreMask
 | 
	
		
			
				|  |  | -            & ~ContextPropagationFlags.Deadline & ~ContextPropagationFlags.Cancellation;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        readonly CallSafeHandle parentCall;
 | 
	
		
			
				|  |  | -        readonly DateTime deadline;
 | 
	
		
			
				|  |  | -        readonly CancellationToken cancellationToken;
 | 
	
		
			
				|  |  | -        readonly ContextPropagationOptions options;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        internal ContextPropagationToken(CallSafeHandle parentCall, DateTime deadline, CancellationToken cancellationToken, ContextPropagationOptions options)
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            this.parentCall = GrpcPreconditions.CheckNotNull(parentCall);
 | 
	
		
			
				|  |  | -            this.deadline = deadline;
 | 
	
		
			
				|  |  | -            this.cancellationToken = cancellationToken;
 | 
	
		
			
				|  |  | -            this.options = options ?? ContextPropagationOptions.Default;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /// <summary>
 | 
	
		
			
				|  |  | -        /// Gets the native handle of the parent call.
 | 
	
		
			
				|  |  | -        /// </summary>
 | 
	
		
			
				|  |  | -        internal CallSafeHandle ParentCall
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            get
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                return this.parentCall;
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /// <summary>
 | 
	
		
			
				|  |  | -        /// Gets the parent call's deadline.
 | 
	
		
			
				|  |  | -        /// </summary>
 | 
	
		
			
				|  |  | -        internal DateTime ParentDeadline
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            get
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                return this.deadline;
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /// <summary>
 | 
	
		
			
				|  |  | -        /// Gets the parent call's cancellation token.
 | 
	
		
			
				|  |  | -        /// </summary>
 | 
	
		
			
				|  |  | -        internal CancellationToken ParentCancellationToken
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            get
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                return this.cancellationToken;
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /// <summary>
 | 
	
		
			
				|  |  | -        /// Get the context propagation options.
 | 
	
		
			
				|  |  | -        /// </summary>
 | 
	
		
			
				|  |  | -        internal ContextPropagationOptions Options
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            get
 | 
	
		
			
				|  |  | -            {
 | 
	
		
			
				|  |  | -                return this.options;
 | 
	
		
			
				|  |  | -            }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    /// <summary>
 | 
	
		
			
				|  |  | -    /// Options for <see cref="ContextPropagationToken"/>.
 | 
	
		
			
				|  |  | +    /// Underlying gRPC implementation may provide other "opaque" contexts (like tracing context) that
 | 
	
		
			
				|  |  | +    /// are not explicitly accesible via the public C# API, but this token still allows propagating them.
 | 
	
		
			
				|  |  |      /// </summary>
 | 
	
		
			
				|  |  | -    public class ContextPropagationOptions
 | 
	
		
			
				|  |  | +    public abstract class ContextPropagationToken
 | 
	
		
			
				|  |  |      {
 | 
	
		
			
				|  |  | -        /// <summary>
 | 
	
		
			
				|  |  | -        /// The context propagation options that will be used by default.
 | 
	
		
			
				|  |  | -        /// </summary>
 | 
	
		
			
				|  |  | -        public static readonly ContextPropagationOptions Default = new ContextPropagationOptions();
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        bool propagateDeadline;
 | 
	
		
			
				|  |  | -        bool propagateCancellation;
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /// <summary>
 | 
	
		
			
				|  |  | -        /// Creates new context propagation options.
 | 
	
		
			
				|  |  | -        /// </summary>
 | 
	
		
			
				|  |  | -        /// <param name="propagateDeadline">If set to <c>true</c> parent call's deadline will be propagated to the child call.</param>
 | 
	
		
			
				|  |  | -        /// <param name="propagateCancellation">If set to <c>true</c> parent call's cancellation token will be propagated to the child call.</param>
 | 
	
		
			
				|  |  | -        public ContextPropagationOptions(bool propagateDeadline = true, bool propagateCancellation = true)
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            this.propagateDeadline = propagateDeadline;
 | 
	
		
			
				|  |  | -            this.propagateCancellation = propagateCancellation;
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -            
 | 
	
		
			
				|  |  | -        /// <summary><c>true</c> if parent call's deadline should be propagated to the child call.</summary>
 | 
	
		
			
				|  |  | -        public bool IsPropagateDeadline
 | 
	
		
			
				|  |  | +        internal ContextPropagationToken()
 | 
	
		
			
				|  |  |          {
 | 
	
		
			
				|  |  | -            get { return this.propagateDeadline; }
 | 
	
		
			
				|  |  |          }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -        /// <summary><c>true</c> if parent call's cancellation token should be propagated to the child call.</summary>
 | 
	
		
			
				|  |  | -        public bool IsPropagateCancellation
 | 
	
		
			
				|  |  | -        {
 | 
	
		
			
				|  |  | -            get { return this.propagateCancellation; }
 | 
	
		
			
				|  |  | -        }
 | 
	
		
			
				|  |  | -    }
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -    /// <summary>
 | 
	
		
			
				|  |  | -    /// Context propagation flags from grpc/grpc.h.
 | 
	
		
			
				|  |  | -    /// </summary>
 | 
	
		
			
				|  |  | -    [Flags]
 | 
	
		
			
				|  |  | -    internal enum ContextPropagationFlags
 | 
	
		
			
				|  |  | -    {
 | 
	
		
			
				|  |  | -        Deadline = 1,
 | 
	
		
			
				|  |  | -        CensusStatsContext = 2,
 | 
	
		
			
				|  |  | -        CensusTracingContext = 4,
 | 
	
		
			
				|  |  | -        Cancellation = 8
 | 
	
		
			
				|  |  |      }
 | 
	
		
			
				|  |  |  }
 |