diff options
Diffstat (limited to '')
-rw-r--r-- | src/rgw/rgw_user.cc | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/src/rgw/rgw_user.cc b/src/rgw/rgw_user.cc new file mode 100644 index 000000000..e5e07cbc4 --- /dev/null +++ b/src/rgw/rgw_user.cc @@ -0,0 +1,127 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab ft=cpp + +#include "rgw_sal_rados.h" + +#include "include/types.h" +#include "rgw_user.h" + +// until everything is moved from rgw_common +#include "rgw_common.h" + +#define dout_subsys ceph_subsys_rgw + +using namespace std; + +int rgw_user_sync_all_stats(const DoutPrefixProvider *dpp, rgw::sal::Driver* driver, + rgw::sal::User* user, optional_yield y) +{ + rgw::sal::BucketList user_buckets; + + CephContext *cct = driver->ctx(); + size_t max_entries = cct->_conf->rgw_list_buckets_max_chunk; + string marker; + int ret; + + do { + ret = user->list_buckets(dpp, marker, string(), max_entries, false, user_buckets, y); + if (ret < 0) { + ldpp_dout(dpp, 0) << "failed to read user buckets: ret=" << ret << dendl; + return ret; + } + auto& buckets = user_buckets.get_buckets(); + for (auto i = buckets.begin(); i != buckets.end(); ++i) { + marker = i->first; + + auto& bucket = i->second; + + ret = bucket->load_bucket(dpp, y); + if (ret < 0) { + ldpp_dout(dpp, 0) << "ERROR: could not read bucket info: bucket=" << bucket << " ret=" << ret << dendl; + continue; + } + ret = bucket->sync_user_stats(dpp, y); + if (ret < 0) { + ldout(cct, 0) << "ERROR: could not sync bucket stats: ret=" << ret << dendl; + return ret; + } + ret = bucket->check_bucket_shards(dpp); + if (ret < 0) { + ldpp_dout(dpp, 0) << "ERROR in check_bucket_shards: " << cpp_strerror(-ret)<< dendl; + } + } + } while (user_buckets.is_truncated()); + + ret = user->complete_flush_stats(dpp, y); + if (ret < 0) { + cerr << "ERROR: failed to complete syncing user stats: ret=" << ret << std::endl; + return ret; + } + + return 0; +} + +int rgw_user_get_all_buckets_stats(const DoutPrefixProvider *dpp, + rgw::sal::Driver* driver, + rgw::sal::User* user, + map<string, bucket_meta_entry>& buckets_usage_map, + optional_yield y) +{ + CephContext *cct = driver->ctx(); + size_t max_entries = cct->_conf->rgw_list_buckets_max_chunk; + bool done; + string marker; + int ret; + + do { + rgw::sal::BucketList buckets; + ret = user->list_buckets(dpp, marker, string(), max_entries, false, buckets, y); + if (ret < 0) { + ldpp_dout(dpp, 0) << "failed to read user buckets: ret=" << ret << dendl; + return ret; + } + auto& m = buckets.get_buckets(); + for (const auto& i : m) { + marker = i.first; + + auto& bucket_ent = i.second; + ret = bucket_ent->load_bucket(dpp, y, true /* load user stats */); + if (ret < 0) { + ldpp_dout(dpp, 0) << "ERROR: could not get bucket stats: ret=" << ret << dendl; + return ret; + } + bucket_meta_entry entry; + entry.size = bucket_ent->get_size(); + entry.size_rounded = bucket_ent->get_size_rounded(); + entry.creation_time = bucket_ent->get_creation_time(); + entry.count = bucket_ent->get_count(); + buckets_usage_map.emplace(bucket_ent->get_name(), entry); + } + done = (buckets.count() < max_entries); + } while (!done); + + return 0; +} + +int rgw_validate_tenant_name(const string& t) +{ + struct tench { + static bool is_good(char ch) { + return isalnum(ch) || ch == '_'; + } + }; + std::string::const_iterator it = + std::find_if_not(t.begin(), t.end(), tench::is_good); + return (it == t.end())? 0: -ERR_INVALID_TENANT_NAME; +} + +/** + * Get the anonymous (ie, unauthenticated) user info. + */ +void rgw_get_anon_user(RGWUserInfo& info) +{ + info.user_id = RGW_USER_ANON_ID; + info.display_name.clear(); + info.access_keys.clear(); +} + |