mirror of
https://github.com/dragonflydb/dragonfly.git
synced 2025-05-11 10:25:47 +02:00
c29db83b7e
Introduces squashed executor that allows squashing single-shard commands within multi transactions
79 lines
2.4 KiB
C++
79 lines
2.4 KiB
C++
// Copyright 2023, DragonflyDB authors. All rights reserved.
|
|
// See LICENSE for licensing terms.
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include "base/logging.h"
|
|
#include "facade/reply_capture.h"
|
|
#include "server/conn_context.h"
|
|
|
|
namespace dfly {
|
|
|
|
// MultiCommandSquasher allows executing a series of commands under a multi transaction
|
|
// and squashing multiple consecutive single-shard commands into one hop whenever it's possible,
|
|
// thus greatly decreasing the dispatch overhead for them.
|
|
class MultiCommandSquasher {
|
|
public:
|
|
static void Execute(absl::Span<StoredCmd> cmds, ConnectionContext* cntx) {
|
|
MultiCommandSquasher{cmds, cntx}.Run();
|
|
}
|
|
|
|
private:
|
|
using ReplyChan = ::util::fibers_ext::SimpleChannel<
|
|
facade::CapturingReplyBuilder::Payload,
|
|
base::mpmc_bounded_queue<facade::CapturingReplyBuilder::Payload>>;
|
|
|
|
// Per-shard exection info.
|
|
struct ShardExecInfo {
|
|
ShardExecInfo() : had_writes{false}, cmds{}, reply_chan{nullptr}, local_tx{nullptr} {
|
|
}
|
|
|
|
bool had_writes;
|
|
std::vector<StoredCmd*> cmds; // accumulated commands
|
|
std::unique_ptr<ReplyChan> reply_chan;
|
|
boost::intrusive_ptr<Transaction> local_tx; // stub-mode tx for use inside shard
|
|
};
|
|
|
|
enum class SquashResult { SQUASHED, SQUASHED_FULL, NOT_SQUASHED, ERROR };
|
|
|
|
static constexpr int kChanBufferSize = 32;
|
|
|
|
private:
|
|
MultiCommandSquasher(absl::Span<StoredCmd> cmds, ConnectionContext* cntx);
|
|
|
|
// Lazy initialize shard info.
|
|
ShardExecInfo& PrepareShardInfo(ShardId sid);
|
|
|
|
// Retrun squash flags
|
|
SquashResult TrySquash(StoredCmd* cmd);
|
|
|
|
// Execute separate non-squashed cmd.
|
|
void ExecuteStandalone(StoredCmd* cmd);
|
|
|
|
// Callback that runs on shards during squashed hop.
|
|
facade::OpStatus SquashedHopCb(Transaction* parent_tx, EngineShard* es);
|
|
|
|
// Execute all currently squashed commands.
|
|
void ExecuteSquashed();
|
|
|
|
// Run all commands until completion.
|
|
void Run();
|
|
|
|
private:
|
|
absl::Span<StoredCmd> cmds_; // Input range of stored commands
|
|
ConnectionContext* cntx_; // Underlying context
|
|
const CommandId* base_cid_; // either EVAL or EXEC, used for squashed hops
|
|
|
|
std::vector<ShardExecInfo> sharded_;
|
|
std::vector<ShardId> order_; // reply order for squashed cmds
|
|
|
|
// multi modes that lock on hops (non-atomic, incremental) need keys for squashed hops.
|
|
// track_keys_ stores whether to populate collected_keys_
|
|
bool track_keys_;
|
|
absl::flat_hash_set<MutableSlice> collected_keys_;
|
|
|
|
std::vector<MutableSlice> tmp_keylist_;
|
|
};
|
|
|
|
} // namespace dfly
|