bdev: Use spdk_for_each_bdev() for bdev_get_io_stat and get_bdevs RPCs
spdk_for_each_bdev() can be applied simply to rpc_bdev_get_bdevs() because the callback function rpc_dump_bdev_info() is synchronous. spdk_for_each_bdev() cannot be applied simply to rpc_bdev_get_iostat(). The factored-out callback function _bdev_get_device_stat() is asynchronous. Add desc pointer to struct spdk_bdev_io_stat and open before and close after executing spdk_bdev_get_device_stat(). Replace spdk_bdev_get_by_name() by spdk_bdev_open_ext() when a bdev name is specified. spdk_bdev_register() checks if the name of each bdev is set and each bdev is opened while collecting its stats now. Hence it is not possible that spdk_bdev_get_name() returns NULL. Simplify rpc_bdev_get_iostat_cb() based on this fact. Furthermore, we want to fail the RPC for all failures. The callback function to spdk_bdev_get_device_stat() is executed after stack unwinding if successful. Defer starting RPC response until it is ensured that all spdk_bdev_get_device_stat() calls will succeed or there is no bdev. Signed-off-by: Shuhei Matsumoto <smatsumoto@nvidia.com> Change-Id: I7b036d6d707c49d19c8922a159b12b5b5ce7ca41 Reviewed-on: https://review.spdk.io/gerrit/c/spdk/spdk/+/12089 Community-CI: Broadcom CI <spdk-ci.pdl@broadcom.com> Community-CI: Mellanox Build Bot Tested-by: SPDK CI Jenkins <sys_sgci@intel.com> Reviewed-by: Jim Harris <james.r.harris@intel.com> Reviewed-by: Tomasz Zawadzki <tomasz.zawadzki@intel.com>
This commit is contained in:
parent
e749fa9c27
commit
0d29a988be
@ -188,27 +188,71 @@ SPDK_RPC_REGISTER("bdev_examine", rpc_bdev_examine_bdev, SPDK_RPC_RUNTIME)
|
||||
|
||||
struct rpc_bdev_get_iostat_ctx {
|
||||
int bdev_count;
|
||||
int rc;
|
||||
struct spdk_jsonrpc_request *request;
|
||||
struct spdk_json_write_ctx *w;
|
||||
};
|
||||
|
||||
struct rpc_bdev_iostat {
|
||||
struct spdk_bdev_io_stat stat;
|
||||
struct rpc_bdev_get_iostat_ctx *ctx;
|
||||
struct spdk_bdev_desc *desc;
|
||||
};
|
||||
|
||||
static void
|
||||
rpc_bdev_get_iostat_started(struct rpc_bdev_get_iostat_ctx *ctx)
|
||||
{
|
||||
ctx->w = spdk_jsonrpc_begin_result(ctx->request);
|
||||
|
||||
spdk_json_write_object_begin(ctx->w);
|
||||
spdk_json_write_named_uint64(ctx->w, "tick_rate", spdk_get_ticks_hz());
|
||||
spdk_json_write_named_uint64(ctx->w, "ticks", spdk_get_ticks());
|
||||
|
||||
spdk_json_write_named_array_begin(ctx->w, "bdevs");
|
||||
}
|
||||
|
||||
static void
|
||||
rpc_bdev_get_iostat_done(struct rpc_bdev_get_iostat_ctx *ctx)
|
||||
{
|
||||
if (--ctx->bdev_count != 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (ctx->rc == 0) {
|
||||
spdk_json_write_array_end(ctx->w);
|
||||
spdk_json_write_object_end(ctx->w);
|
||||
spdk_jsonrpc_end_result(ctx->request, ctx->w);
|
||||
} else {
|
||||
/* Return error response after processing all specified bdevs
|
||||
* completed or failed.
|
||||
*/
|
||||
spdk_jsonrpc_send_error_response(ctx->request, ctx->rc,
|
||||
spdk_strerror(-ctx->rc));
|
||||
}
|
||||
|
||||
free(ctx);
|
||||
}
|
||||
|
||||
static void
|
||||
rpc_bdev_get_iostat_cb(struct spdk_bdev *bdev,
|
||||
struct spdk_bdev_io_stat *stat, void *cb_arg, int rc)
|
||||
{
|
||||
struct rpc_bdev_get_iostat_ctx *ctx = cb_arg;
|
||||
struct rpc_bdev_iostat *_stat = cb_arg;
|
||||
struct rpc_bdev_get_iostat_ctx *ctx = _stat->ctx;
|
||||
struct spdk_json_write_ctx *w = ctx->w;
|
||||
const char *bdev_name;
|
||||
|
||||
if (rc != 0) {
|
||||
if (rc != 0 || ctx->rc != 0) {
|
||||
if (ctx->rc == 0) {
|
||||
ctx->rc = rc;
|
||||
}
|
||||
goto done;
|
||||
}
|
||||
|
||||
bdev_name = spdk_bdev_get_name(bdev);
|
||||
if (bdev_name != NULL) {
|
||||
assert(stat == &_stat->stat);
|
||||
|
||||
spdk_json_write_object_begin(w);
|
||||
|
||||
spdk_json_write_named_string(w, "name", bdev_name);
|
||||
spdk_json_write_named_string(w, "name", spdk_bdev_get_name(bdev));
|
||||
|
||||
spdk_json_write_named_uint64(w, "bytes_read", stat->bytes_read);
|
||||
|
||||
@ -241,16 +285,12 @@ rpc_bdev_get_iostat_cb(struct spdk_bdev *bdev,
|
||||
}
|
||||
|
||||
spdk_json_write_object_end(w);
|
||||
}
|
||||
|
||||
done:
|
||||
free(stat);
|
||||
if (--ctx->bdev_count == 0) {
|
||||
spdk_json_write_array_end(ctx->w);
|
||||
spdk_json_write_object_end(w);
|
||||
spdk_jsonrpc_end_result(ctx->request, ctx->w);
|
||||
free(ctx);
|
||||
}
|
||||
rpc_bdev_get_iostat_done(ctx);
|
||||
|
||||
spdk_bdev_close(_stat->desc);
|
||||
free(_stat);
|
||||
}
|
||||
|
||||
struct rpc_bdev_get_iostat {
|
||||
@ -267,15 +307,42 @@ static const struct spdk_json_object_decoder rpc_bdev_get_iostat_decoders[] = {
|
||||
{"name", offsetof(struct rpc_bdev_get_iostat, name), spdk_json_decode_string, true},
|
||||
};
|
||||
|
||||
static int
|
||||
_bdev_get_device_stat(void *_ctx, struct spdk_bdev *bdev)
|
||||
{
|
||||
struct rpc_bdev_get_iostat_ctx *ctx = _ctx;
|
||||
struct rpc_bdev_iostat *_stat;
|
||||
int rc;
|
||||
|
||||
_stat = calloc(1, sizeof(struct rpc_bdev_iostat));
|
||||
if (_stat == NULL) {
|
||||
SPDK_ERRLOG("Failed to allocate rpc_bdev_iostat struct\n");
|
||||
return -ENOMEM;
|
||||
}
|
||||
|
||||
rc = spdk_bdev_open_ext(spdk_bdev_get_name(bdev), false, dummy_bdev_event_cb, NULL, &_stat->desc);
|
||||
if (rc != 0) {
|
||||
free(_stat);
|
||||
SPDK_ERRLOG("Failed to open bdev\n");
|
||||
return rc;
|
||||
}
|
||||
|
||||
ctx->bdev_count++;
|
||||
_stat->ctx = ctx;
|
||||
spdk_bdev_get_device_stat(bdev, &_stat->stat, rpc_bdev_get_iostat_cb, _stat);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void
|
||||
rpc_bdev_get_iostat(struct spdk_jsonrpc_request *request,
|
||||
const struct spdk_json_val *params)
|
||||
{
|
||||
struct rpc_bdev_get_iostat req = {};
|
||||
struct spdk_bdev *bdev = NULL;
|
||||
struct spdk_json_write_ctx *w;
|
||||
struct spdk_bdev_io_stat *stat;
|
||||
struct spdk_bdev_desc *desc = NULL;
|
||||
struct rpc_bdev_get_iostat_ctx *ctx;
|
||||
struct rpc_bdev_iostat *_stat;
|
||||
int rc;
|
||||
|
||||
if (params != NULL) {
|
||||
if (spdk_json_decode_object(params, rpc_bdev_get_iostat_decoders,
|
||||
@ -289,10 +356,10 @@ rpc_bdev_get_iostat(struct spdk_jsonrpc_request *request,
|
||||
}
|
||||
|
||||
if (req.name) {
|
||||
bdev = spdk_bdev_get_by_name(req.name);
|
||||
if (bdev == NULL) {
|
||||
SPDK_ERRLOG("bdev '%s' does not exist\n", req.name);
|
||||
spdk_jsonrpc_send_error_response(request, -ENODEV, spdk_strerror(ENODEV));
|
||||
rc = spdk_bdev_open_ext(req.name, false, dummy_bdev_event_cb, NULL, &desc);
|
||||
if (rc != 0) {
|
||||
SPDK_ERRLOG("Failed to open bdev '%s': %d\n", req.name, rc);
|
||||
spdk_jsonrpc_send_error_response(request, rc, spdk_strerror(-rc));
|
||||
free_rpc_bdev_get_iostat(&req);
|
||||
return;
|
||||
}
|
||||
@ -308,56 +375,53 @@ rpc_bdev_get_iostat(struct spdk_jsonrpc_request *request,
|
||||
return;
|
||||
}
|
||||
|
||||
w = spdk_jsonrpc_begin_result(request);
|
||||
/*
|
||||
* Increment initial bdev_count so that it will never reach 0 in the middle
|
||||
* of iterating.
|
||||
*/
|
||||
ctx->bdev_count++;
|
||||
ctx->request = request;
|
||||
ctx->w = w;
|
||||
|
||||
if (desc != NULL) {
|
||||
_stat = calloc(1, sizeof(struct rpc_bdev_iostat));
|
||||
if (_stat == NULL) {
|
||||
SPDK_ERRLOG("Failed to allocate rpc_bdev_iostat struct\n");
|
||||
ctx->rc = -ENOMEM;
|
||||
|
||||
spdk_json_write_object_begin(w);
|
||||
spdk_json_write_named_uint64(w, "tick_rate", spdk_get_ticks_hz());
|
||||
spdk_json_write_named_uint64(w, "ticks", spdk_get_ticks());
|
||||
|
||||
spdk_json_write_named_array_begin(w, "bdevs");
|
||||
|
||||
if (bdev != NULL) {
|
||||
stat = calloc(1, sizeof(struct spdk_bdev_io_stat));
|
||||
if (stat == NULL) {
|
||||
SPDK_ERRLOG("Failed to allocate rpc_bdev_get_iostat_ctx struct\n");
|
||||
spdk_bdev_close(desc);
|
||||
} else {
|
||||
_stat->desc = desc;
|
||||
|
||||
ctx->bdev_count++;
|
||||
spdk_bdev_get_device_stat(bdev, stat, rpc_bdev_get_iostat_cb, ctx);
|
||||
_stat->ctx = ctx;
|
||||
spdk_bdev_get_device_stat(spdk_bdev_desc_get_bdev(desc), &_stat->stat,
|
||||
rpc_bdev_get_iostat_cb, _stat);
|
||||
}
|
||||
} else {
|
||||
for (bdev = spdk_bdev_first(); bdev != NULL; bdev = spdk_bdev_next(bdev)) {
|
||||
stat = calloc(1, sizeof(struct spdk_bdev_io_stat));
|
||||
if (stat == NULL) {
|
||||
SPDK_ERRLOG("Failed to allocate spdk_bdev_io_stat struct\n");
|
||||
break;
|
||||
}
|
||||
ctx->bdev_count++;
|
||||
spdk_bdev_get_device_stat(bdev, stat, rpc_bdev_get_iostat_cb, ctx);
|
||||
rc = spdk_for_each_bdev(ctx, _bdev_get_device_stat);
|
||||
if (rc != 0 && ctx->rc == 0) {
|
||||
ctx->rc = rc;
|
||||
}
|
||||
}
|
||||
|
||||
if (--ctx->bdev_count == 0) {
|
||||
spdk_json_write_array_end(w);
|
||||
spdk_json_write_object_end(w);
|
||||
spdk_jsonrpc_end_result(request, w);
|
||||
free(ctx);
|
||||
if (ctx->rc == 0) {
|
||||
/* We want to fail the RPC for all failures. The callback function to
|
||||
* spdk_bdev_get_device_stat() is executed after stack unwinding if successful.
|
||||
* Hence defer starting RPC response until it is ensured that all
|
||||
* spdk_bdev_get_device_stat() calls will succeed or there is no bdev.
|
||||
*/
|
||||
rpc_bdev_get_iostat_started(ctx);
|
||||
}
|
||||
|
||||
rpc_bdev_get_iostat_done(ctx);
|
||||
}
|
||||
SPDK_RPC_REGISTER("bdev_get_iostat", rpc_bdev_get_iostat, SPDK_RPC_RUNTIME)
|
||||
SPDK_RPC_REGISTER_ALIAS_DEPRECATED(bdev_get_iostat, get_bdevs_iostat)
|
||||
|
||||
static void
|
||||
rpc_dump_bdev_info(struct spdk_json_write_ctx *w,
|
||||
struct spdk_bdev *bdev)
|
||||
static int
|
||||
rpc_dump_bdev_info(void *ctx, struct spdk_bdev *bdev)
|
||||
{
|
||||
struct spdk_json_write_ctx *w = ctx;
|
||||
struct spdk_bdev_alias *tmp;
|
||||
uint64_t qos_limits[SPDK_BDEV_QOS_NUM_RATE_LIMIT_TYPES];
|
||||
struct spdk_memory_domain **domains;
|
||||
@ -470,6 +534,8 @@ rpc_dump_bdev_info(struct spdk_json_write_ctx *w,
|
||||
spdk_json_write_object_end(w);
|
||||
|
||||
spdk_json_write_object_end(w);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
struct rpc_bdev_get_bdevs {
|
||||
@ -500,22 +566,25 @@ get_bdevs_poller(void *_ctx)
|
||||
{
|
||||
struct rpc_bdev_get_bdevs_ctx *ctx = _ctx;
|
||||
struct spdk_json_write_ctx *w;
|
||||
struct spdk_bdev *bdev;
|
||||
struct spdk_bdev_desc *desc;
|
||||
int rc;
|
||||
|
||||
bdev = spdk_bdev_get_by_name(ctx->rpc.name);
|
||||
if (bdev == NULL && spdk_get_ticks() < ctx->timeout_ticks) {
|
||||
rc = spdk_bdev_open_ext(ctx->rpc.name, false, dummy_bdev_event_cb, NULL, &desc);
|
||||
if (rc != 0 && spdk_get_ticks() < ctx->timeout_ticks) {
|
||||
return SPDK_POLLER_BUSY;
|
||||
}
|
||||
|
||||
if (bdev == NULL) {
|
||||
if (rc != 0) {
|
||||
SPDK_ERRLOG("Timed out while waiting for bdev '%s' to appear\n", ctx->rpc.name);
|
||||
spdk_jsonrpc_send_error_response(ctx->request, -ENODEV, spdk_strerror(ENODEV));
|
||||
} else {
|
||||
w = spdk_jsonrpc_begin_result(ctx->request);
|
||||
spdk_json_write_array_begin(w);
|
||||
rpc_dump_bdev_info(w, bdev);
|
||||
rpc_dump_bdev_info(w, spdk_bdev_desc_get_bdev(desc));
|
||||
spdk_json_write_array_end(w);
|
||||
spdk_jsonrpc_end_result(ctx->request, w);
|
||||
|
||||
spdk_bdev_close(desc);
|
||||
}
|
||||
|
||||
spdk_poller_unregister(&ctx->poller);
|
||||
@ -532,7 +601,8 @@ rpc_bdev_get_bdevs(struct spdk_jsonrpc_request *request,
|
||||
struct rpc_bdev_get_bdevs req = {};
|
||||
struct rpc_bdev_get_bdevs_ctx *ctx;
|
||||
struct spdk_json_write_ctx *w;
|
||||
struct spdk_bdev *bdev = NULL;
|
||||
struct spdk_bdev_desc *desc = NULL;
|
||||
int rc;
|
||||
|
||||
if (params && spdk_json_decode_object(params, rpc_bdev_get_bdevs_decoders,
|
||||
SPDK_COUNTOF(rpc_bdev_get_bdevs_decoders),
|
||||
@ -545,8 +615,8 @@ rpc_bdev_get_bdevs(struct spdk_jsonrpc_request *request,
|
||||
}
|
||||
|
||||
if (req.name) {
|
||||
bdev = spdk_bdev_get_by_name(req.name);
|
||||
if (bdev == NULL) {
|
||||
rc = spdk_bdev_open_ext(req.name, false, dummy_bdev_event_cb, NULL, &desc);
|
||||
if (rc != 0) {
|
||||
if (req.timeout == 0) {
|
||||
SPDK_ERRLOG("bdev '%s' does not exist\n", req.name);
|
||||
spdk_jsonrpc_send_error_response(request, -ENODEV, spdk_strerror(ENODEV));
|
||||
@ -583,12 +653,11 @@ rpc_bdev_get_bdevs(struct spdk_jsonrpc_request *request,
|
||||
w = spdk_jsonrpc_begin_result(request);
|
||||
spdk_json_write_array_begin(w);
|
||||
|
||||
if (bdev != NULL) {
|
||||
rpc_dump_bdev_info(w, bdev);
|
||||
if (desc != NULL) {
|
||||
rpc_dump_bdev_info(w, spdk_bdev_desc_get_bdev(desc));
|
||||
spdk_bdev_close(desc);
|
||||
} else {
|
||||
for (bdev = spdk_bdev_first(); bdev != NULL; bdev = spdk_bdev_next(bdev)) {
|
||||
rpc_dump_bdev_info(w, bdev);
|
||||
}
|
||||
spdk_for_each_bdev(w, rpc_dump_bdev_info);
|
||||
}
|
||||
|
||||
spdk_json_write_array_end(w);
|
||||
|
Loading…
Reference in New Issue
Block a user