コード例 #1
0
        /// <summary>
        /// Отправляет специфический запрос содержащий токен авторизации. Ожидает ответ.
        /// </summary>
        private async Task <bool> AuthorizeAsync(SocketQueue socketQueue, byte[] bearerToken)
        {
            // Запрос на авторизацию по токену.
            var requestToSend = Message.CreateRequest("Auth/AuthorizeToken", new Arg[] { new Arg("token", bearerToken) });

            // Отправить запрос и получить ответ.
            object result = await ExecuteRequestAsync(requestToSend, returnType : typeof(bool), socketQueue).ConfigureAwait(false);

            return((bool)result);
        }
コード例 #2
0
 /// <summary>
 /// Событие — обрыв сокета.
 /// </summary>
 private protected override void OnAtomicDisconnect(SocketQueue socketQueue)
 {
     // Установить Socket = null если в ссылке хранится экземпляр соединения в котором произошел обрыв.
     Interlocked.CompareExchange(ref _socket, null, socketQueue);
 }
コード例 #3
0
        /// <summary>
        /// Выполнить подключение сокета если еще не подключен.
        /// </summary>
        private async Task <SocketQueue> ConnectIfNeededAsync()
        {
            // Копия volatile ссылки.
            SocketQueue socketQueue = Socket;

            // Fast-path.
            if (socketQueue != null)
            {
                return(socketQueue);
            }

            using (await _asyncLock.LockAsync().ConfigureAwait(false))
            {
                // Копия volatile ссылки.
                socketQueue = _socket;

                // Необходима повторная проверка.
                if (socketQueue == null)
                {
                    if (ServiceProvider == null)
                    {
                        var ioc = new ServiceCollection();
                        _iocConfigure?.Invoke(ioc);
                        ConfigureIoC(ioc);
                    }

                    // Новый сокет.
                    var ws = new MyClientWebSocket();

                    try
                    {
                        // Простое подключение веб-сокета.
                        await ws.ConnectAsync(_uri).ConfigureAwait(false);
                    }
                    catch
                    // Не удалось подключиться (сервер не запущен?).
                    {
                        ws.Dispose();
                        throw;
                    }

                    // Управляемая обвертка для сокета.
                    socketQueue = new SocketQueue(ws);

                    // Начать бесконечное чтение из сокета.
                    StartReceivingLoop(socketQueue);

                    // Копируем ссылку на публичный токен.
                    byte[] bearerTokenCopy = BearerToken;

                    // Если токен установлен то отправить его на сервер что-бы авторизовать текущее подключение.
                    if (bearerTokenCopy != null)
                    {
                        try
                        {
                            IsAuthorized = await AuthorizeAsync(socketQueue, bearerTokenCopy).ConfigureAwait(false);
                        }
                        catch (Exception ex)
                        // Обрыв соединения.
                        {
                            // Оповестить об обрыве.
                            AtomicDisconnect(socketQueue, ex);
                            throw;
                        }
                    }

                    Debug.WriteIf(IsAuthorized, "Соединение успешно авторизовано");

                    // Открыть публичный доступ к этому сокету.
                    // Установка этого свойства должно быть самым последним действием.
                    Interlocked.CompareExchange(ref _socket, socketQueue, null); // Записать только если в ссылке Null.
                }
                return(socketQueue);
            }
        }
コード例 #4
0
 public SendJob(SocketQueue socketQueue, MemoryPoolStream memoryPoolStream)
 {
     SocketQueue      = socketQueue;
     MemoryPoolStream = memoryPoolStream;
 }