mirrors/core-lightning
1
0
Fork 0
mirror of https://github.com/ElementsProject/lightning.git synced 2025-01-19 05:44:12 +01:00
Code Issues Projects Releases Wiki Activity

patch libplugin-batch.patch

Browse Source
This commit is contained in:
Rusty Russell 2024-09-18 17:11:27 +09:30
parent 1362448352
commit 17191259ea
2 changed files with 138 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

99
plugins/libplugin.c
View File

@ -1037,6 +1037,105 @@ send_outreq(struct plugin *plugin, const struct out_req *req)
return &pending;
}
struct request_batch {
size_t num_remaining;
struct command_result *(*cb)(struct command *,
const char *,
const jsmntok_t *,
void *);
struct command_result *(*errcb)(struct command *,
const char *,
const jsmntok_t *,
void *);
struct command_result *(*finalcb)(struct command *,
void *);
void *arg;
};
struct request_batch *request_batch_new_(const tal_t *ctx,
struct command_result *(*cb)(struct command *,
const char *,
const jsmntok_t *,
void *),
struct command_result *(*errcb)(struct command *,
const char *,
const jsmntok_t *,
void *),
struct command_result *(*finalcb)(struct command *,
void *),
void *arg)
{
struct request_batch *batch = tal(ctx, struct request_batch);
batch->num_remaining = 0;
batch->cb = cb;
batch->errcb = errcb;
batch->finalcb = finalcb;
batch->arg = arg;
return batch;
}
static struct command_result *batch_one_complete(struct command *cmd,
struct request_batch *batch)
{
void *arg;
struct command_result *(*finalcb)(struct command *, void *);
assert(batch->num_remaining);
if (--batch->num_remaining != 0)
return command_still_pending(cmd);
arg = batch->arg;
finalcb = batch->finalcb;
tal_free(batch);
return finalcb(cmd, arg);
}
static struct command_result *batch_one_success(struct command *cmd,
const char *buf,
const jsmntok_t *result,
struct request_batch *batch)
{
/* If this frees stuff (e.g. fails), just return */
if (batch->cb && batch->cb(cmd, buf, result, batch->arg) == &complete)
return command_done();
return batch_one_complete(cmd, batch);
}
static struct command_result *batch_one_failed(struct command *cmd,
const char *buf,
const jsmntok_t *result,
struct request_batch *batch)
{
/* If this frees stuff (e.g. fails), just return */
if (batch->errcb && batch->errcb(cmd, buf, result, batch->arg) == &complete)
return command_done();
return batch_one_complete(cmd, batch);
}
struct out_req *add_to_batch(struct command *cmd,
struct request_batch *batch,
const char *cmdname)
{
batch->num_remaining++;
return jsonrpc_request_start(cmd->plugin, cmd, cmdname,
batch_one_success,
batch_one_failed,
batch);
}
/* Runs finalcb immediately if batch is empty. */
struct command_result *batch_done(struct command *cmd,
struct request_batch *batch)
{
/* Same path as completion */
batch->num_remaining++;
return batch_one_complete(cmd, batch);
}
static void json_add_deprecated(struct json_stream *js,
const char *fieldname,
const char *depr_start, const char *depr_end)

39
plugins/libplugin.h
View File

@ -155,6 +155,45 @@ struct out_req *jsonrpc_request_start_(struct plugin *plugin,
NULL, \
(arg))
/* Batch of requests: cb and errcb are optional, finalcb is called when all complete. */
struct request_batch *request_batch_new_(const tal_t *ctx,
struct command_result *(*cb)(struct command *,
const char *,
const jsmntok_t *,
void *),
struct command_result *(*errcb)(struct command *,
const char *,
const jsmntok_t *,
void *),
struct command_result *(*finalcb)(struct command *,
void *),
void *arg);
#define request_batch_new(ctx, cb, errcb, finalcb, arg) \
request_batch_new_((ctx), \
typesafe_cb_preargs(struct command_result *, void *, \
(cb), (arg), \
struct command *command, \
const char *buf, \
const jsmntok_t *result), \
typesafe_cb_preargs(struct command_result *, void *, \
(errcb), (arg), \
struct command *command, \
const char *buf, \
const jsmntok_t *result), \
typesafe_cb_preargs(struct command_result *, void *, \
(finalcb), (arg), \
struct command *command), \
(arg))
struct out_req *add_to_batch(struct command *cmd,
struct request_batch *batch,
const char *cmdname);
/* Runs finalcb immediately if batch is empty. */
struct command_result *batch_done(struct command *cmd,
struct request_batch *batch);
/* Helper to create a JSONRPC2 response stream with a "result" object. */
struct json_stream *jsonrpc_stream_success(struct command *cmd);