Ejemplo n.º 1
0
        public static async Task <TcpConnectionHeaderMessage.SupportedFeatures> NegotiateProtocolVersionAsync(JsonOperationContext documentsContext, Stream stream, TcpNegotiateParamaters parameters)
        {
            using (var writer = new BlittableJsonTextWriter(documentsContext, stream))
            {
                var currentVersion = parameters.Version;
                while (true)
                {
                    if (parameters.CancellationToken.IsCancellationRequested)
                    {
                        throw new OperationCanceledException($"Stoped Tcp negotiation for {parameters.Operation} because of cancelation request");
                    }

                    documentsContext.Write(writer, new DynamicJsonValue
                    {
                        [nameof(TcpConnectionHeaderMessage.DatabaseName)]     = parameters.Database, // _parent.Database.Name,
                        [nameof(TcpConnectionHeaderMessage.Operation)]        = parameters.Operation.ToString(),
                        [nameof(TcpConnectionHeaderMessage.SourceNodeTag)]    = parameters.NodeTag,
                        [nameof(TcpConnectionHeaderMessage.OperationVersion)] = currentVersion
                    });
                    writer.Flush();
                    int version;
                    if (parameters.ReadResponseAndGetVersionAsync == null)
                    {
                        version = parameters.ReadResponseAndGetVersion(documentsContext, writer, stream, parameters.Url);
                    }
                    else
                    {
                        version = await parameters.ReadResponseAndGetVersionAsync(documentsContext, writer, stream, parameters.Url, parameters.CancellationToken).ConfigureAwait(false);
                    }
                    //In this case we usally throw internaly but for completeness we better handle it
                    if (version == -2)
                    {
                        return(TcpConnectionHeaderMessage.GetSupportedFeaturesFor(TcpConnectionHeaderMessage.OperationTypes.Drop, TcpConnectionHeaderMessage.DropBaseLine40000));
                    }
                    var(supported, prevSupported) = TcpConnectionHeaderMessage.OperationVersionSupported(parameters.Operation, version);
                    if (supported)
                    {
                        //We are done
                        if (currentVersion == version)
                        {
                            return(TcpConnectionHeaderMessage.GetSupportedFeaturesFor(parameters.Operation, version));
                        }
                        //Here we support the requested version but need to inform the otherside we agree
                        currentVersion = version;
                        continue;
                    }
                    if (prevSupported == -1)
                    {
                        return(TcpConnectionHeaderMessage.GetSupportedFeaturesFor(TcpConnectionHeaderMessage.OperationTypes.None, TcpConnectionHeaderMessage.NoneBaseLine40000));
                    }
                    currentVersion = prevSupported;
                }
            }
        }
Ejemplo n.º 2
0
 public static TcpConnectionHeaderMessage.SupportedFeatures NegotiateProtocolVersion(JsonOperationContext documentsContext, Stream stream, TcpNegotiateParamaters parameters)
 {
     using (var writer = new BlittableJsonTextWriter(documentsContext, stream))
     {
         var currentVersion = parameters.Version;
         while (true)
         {
             documentsContext.Write(writer, new DynamicJsonValue
             {
                 [nameof(TcpConnectionHeaderMessage.DatabaseName)]     = parameters.Database, // _parent.Database.Name,
                 [nameof(TcpConnectionHeaderMessage.Operation)]        = parameters.Operation.ToString(),
                 [nameof(TcpConnectionHeaderMessage.SourceNodeTag)]    = parameters.NodeTag,
                 [nameof(TcpConnectionHeaderMessage.OperationVersion)] = currentVersion
             });
             writer.Flush();
             var version = parameters.ReadResponseAndGetVersion(documentsContext, writer, stream, parameters.Url);
             //In this case we usally throw internaly but for completeness we better handle it
             if (version == -2)
             {
                 return(TcpConnectionHeaderMessage.GetSupportedFeaturesFor(TcpConnectionHeaderMessage.OperationTypes.Drop, TcpConnectionHeaderMessage.DropBaseLine40000));
             }
             var(supported, prevSupported) = TcpConnectionHeaderMessage.OperationVersionSupported(parameters.Operation, version);
             if (supported)
             {
                 //We are done
                 if (currentVersion == version)
                 {
                     return(TcpConnectionHeaderMessage.GetSupportedFeaturesFor(parameters.Operation, version));
                 }
                 //Here we support the requested version but need to inform the otherside we agree
                 currentVersion = version;
                 continue;
             }
             if (prevSupported == -1)
             {
                 return(TcpConnectionHeaderMessage.GetSupportedFeaturesFor(TcpConnectionHeaderMessage.OperationTypes.None, TcpConnectionHeaderMessage.NoneBaseLine40000));
             }
             currentVersion = prevSupported;
         }
     }
 }