/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#pragma once

#include <quic/server/QuicUDPSocketFactory.h>

namespace quic {

class QuicReusePortUDPSocketFactory : public QuicUDPSocketFactory {
 public:
  ~QuicReusePortUDPSocketFactory() override {}

  QuicReusePortUDPSocketFactory(bool reusePort = true, bool reuseAddr = false)
      : reusePort_(reusePort), reuseAddr_(reuseAddr) {}

  std::unique_ptr<FollyAsyncUDPSocketAlias> make(folly::EventBase* evb, int)
      override {
    auto sock = std::make_unique<FollyAsyncUDPSocketAlias>(evb);
    sock->setReusePort(reusePort_);
    sock->setReuseAddr(reuseAddr_);
    return sock;
  }

 private:
  bool reusePort_;
  bool reuseAddr_;
};
} // namespace quic
