diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-27 18:24:20 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-27 18:24:20 +0000 |
commit | 483eb2f56657e8e7f419ab1a4fab8dce9ade8609 (patch) | |
tree | e5d88d25d870d5dedacb6bbdbe2a966086a0a5cf /src/rgw/rgw_aio_throttle.h | |
parent | Initial commit. (diff) | |
download | ceph-483eb2f56657e8e7f419ab1a4fab8dce9ade8609.tar.xz ceph-483eb2f56657e8e7f419ab1a4fab8dce9ade8609.zip |
Adding upstream version 14.2.21.upstream/14.2.21upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/rgw/rgw_aio_throttle.h')
-rw-r--r-- | src/rgw/rgw_aio_throttle.h | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/src/rgw/rgw_aio_throttle.h b/src/rgw/rgw_aio_throttle.h new file mode 100644 index 00000000..751d7f98 --- /dev/null +++ b/src/rgw/rgw_aio_throttle.h @@ -0,0 +1,83 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab + +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2018 Red Hat, Inc. + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ + +#pragma once + +#include "include/rados/librados_fwd.hpp" +#include <memory> +#include "common/ceph_mutex.h" +#include "services/svc_rados.h" +#include "rgw_aio.h" + +namespace rgw { + +// a throttle for aio operations that enforces a maximum window on outstanding +// bytes. only supports a single waiter, so all public functions must be called +// from the same thread +class AioThrottle : public Aio { + protected: + const uint64_t window; + uint64_t pending_size = 0; + + bool is_available() const { return pending_size <= window; } + bool has_completion() const { return !completed.empty(); } + bool is_drained() const { return pending.empty(); } + + struct Pending : AioResultEntry { + AioThrottle *parent = nullptr; + uint64_t cost = 0; + librados::AioCompletion *completion = nullptr; + }; + OwningList<Pending> pending; + AioResultList completed; + + enum class Wait { None, Available, Completion, Drained }; + Wait waiter = Wait::None; + + bool waiter_ready() const; + + ceph::mutex mutex = ceph::make_mutex("AioThrottle"); + ceph::condition_variable cond; + + void get(Pending& p); + void put(Pending& p); + + static void aio_cb(void *cb, void *arg); + + public: + AioThrottle(uint64_t window) : window(window) {} + + virtual ~AioThrottle() { + // must drain before destructing + ceph_assert(pending.empty()); + ceph_assert(completed.empty()); + } + + AioResultList submit(RGWSI_RADOS::Obj& obj, + librados::ObjectReadOperation *op, + uint64_t cost, uint64_t id) override; + + AioResultList submit(RGWSI_RADOS::Obj& obj, + librados::ObjectWriteOperation *op, + uint64_t cost, uint64_t id) override; + + AioResultList poll() override; + + AioResultList wait() override; + + AioResultList drain() override; +}; + +} // namespace rgw |