src/lib/fsl/socket/socket_drainer.h

// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SRC_LIB_FSL_SOCKET_SOCKET_DRAINER_H_
#define SRC_LIB_FSL_SOCKET_SOCKET_DRAINER_H_

#include <lib/async/cpp/wait.h>
#include <lib/async/default.h>
#include <lib/zx/socket.h>

#include "src/lib/fxl/fxl_export.h"
#include "src/lib/fxl/macros.h"

namespace fsl {

class FXL_EXPORT SocketDrainer {
 public:
  class Client {
   public:
    virtual void OnDataAvailable(const void* data, size_t num_bytes) = 0;
    virtual void OnDataComplete() = 0;

   protected:
    virtual ~Client();
  };

  explicit SocketDrainer(Client* client,
                         async_dispatcher_t* dispatcher = async_get_default_dispatcher());
  ~SocketDrainer();

  void Start(zx::socket source);

 private:
  void OnHandleReady(async_dispatcher_t* dispatcher, async::WaitBase* wait, zx_status_t status,
                     const zx_packet_signal_t* signal);

  Client* client_;
  async_dispatcher_t* dispatcher_;
  zx::socket source_;
  async::WaitMethod<SocketDrainer, &SocketDrainer::OnHandleReady> wait_{this};
  bool* destruction_sentinel_;

  FXL_DISALLOW_COPY_AND_ASSIGN(SocketDrainer);
};

}  // namespace fsl

#endif  // SRC_LIB_FSL_SOCKET_SOCKET_DRAINER_H_