Skip to content

Commit

Permalink
replication: implement an instance id filter for relay
Browse files Browse the repository at this point in the history
Add a filter for relay to skip rows coming from unwanted instances.
A list of instance ids whose rows replica doesn't want to fetch is encoded
together with SUBSCRIBE request after a freshly introduced flag IPROTO_ID_MASK.

Filtering rows is needed to prevent an instance from fetching its own
rows from a remote master, which is useful on initial configuration and
harmful on resubscribe.

Prerequisite #4739, #3294

@TarantoolBot document

Title: document new binary protocol key and subscribe request changes

Add key `IPROTO_ID_MASK = 0x51` to the internals reference.
This is an optional key used in SUBSCRIBE request followed by an array
of ids of instances whose rows won't be relayed to the replica.

SUBSCRIBE request is supplemented with an optional field of the
following structure:
```
+====================+
|      ID_MASK       |
|   0x51 : ID LIST   |
| MP_INT : MP_ARRRAY |
|                    |
+====================+
```
The field is encoded only when the id list is not empty.
  • Loading branch information
sergepetrenko committed Feb 26, 2020
1 parent f5fb54c commit ea38145
Show file tree
Hide file tree
Showing 7 changed files with 77 additions and 20 deletions.
2 changes: 1 addition & 1 deletion src/box/applier.cc
Original file line number Diff line number Diff line change
Expand Up @@ -867,7 +867,7 @@ applier_subscribe(struct applier *applier)
vclock_create(&vclock);
vclock_copy(&vclock, &replicaset.vclock);
xrow_encode_subscribe_xc(&row, &REPLICASET_UUID, &INSTANCE_UUID,
&vclock, replication_anon);
&vclock, replication_anon, 0);
coio_write_xrow(coio, &row);

/* Read SUBSCRIBE response */
Expand Down
7 changes: 4 additions & 3 deletions src/box/box.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1787,8 +1787,9 @@ box_process_subscribe(struct ev_io *io, struct xrow_header *header)
uint32_t replica_version_id;
vclock_create(&replica_clock);
bool anon;
xrow_decode_subscribe_xc(header, NULL, &replica_uuid,
&replica_clock, &replica_version_id, &anon);
unsigned int id_mask;
xrow_decode_subscribe_xc(header, NULL, &replica_uuid, &replica_clock,
&replica_version_id, &anon, &id_mask);

/* Forbid connection to itself */
if (tt_uuid_is_equal(&replica_uuid, &INSTANCE_UUID))
Expand Down Expand Up @@ -1871,7 +1872,7 @@ box_process_subscribe(struct ev_io *io, struct xrow_header *header)
* indefinitely).
*/
relay_subscribe(replica, io->fd, header->sync, &replica_clock,
replica_version_id);
replica_version_id, id_mask);
}

void
Expand Down
1 change: 1 addition & 0 deletions src/box/iproto_constants.h
Original file line number Diff line number Diff line change
Expand Up @@ -125,6 +125,7 @@ enum iproto_key {
IPROTO_STMT_ID = 0x43,
/* Leave a gap between SQL keys and additional request keys */
IPROTO_REPLICA_ANON = 0x50,
IPROTO_ID_MASK = 0x51,
IPROTO_KEY_MAX
};

Expand Down
9 changes: 8 additions & 1 deletion src/box/relay.cc
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,7 @@ struct relay {
struct vclock recv_vclock;
/** Replicatoin slave version. */
uint32_t version_id;
unsigned int id_mask;
/**
* Local vclock at the moment of subscribe, used to check
* dataset on the other side and send missing data rows if any.
Expand Down Expand Up @@ -676,7 +677,8 @@ relay_subscribe_f(va_list ap)
/** Replication acceptor fiber handler. */
void
relay_subscribe(struct replica *replica, int fd, uint64_t sync,
struct vclock *replica_clock, uint32_t replica_version_id)
struct vclock *replica_clock, uint32_t replica_version_id,
unsigned int replica_id_mask)
{
assert(replica->anon || replica->id != REPLICA_ID_NIL);
struct relay *relay = replica->relay;
Expand Down Expand Up @@ -705,6 +707,8 @@ relay_subscribe(struct replica *replica, int fd, uint64_t sync,
vclock_copy(&relay->tx.vclock, replica_clock);
relay->version_id = replica_version_id;

relay->id_mask = replica_id_mask;

int rc = cord_costart(&relay->cord, "subscribe",
relay_subscribe_f, relay);
if (rc == 0)
Expand Down Expand Up @@ -763,6 +767,9 @@ relay_send_row(struct xstream *stream, struct xrow_header *packet)
packet->group_id = GROUP_DEFAULT;
packet->bodycnt = 0;
}
/* Check if the rows from the instance are filtered. */
if (1 << packet->replica_id & relay->id_mask)
return;
/*
* We're feeding a WAL, thus responding to FINAL JOIN or SUBSCRIBE
* request. If this is FINAL JOIN (i.e. relay->replica is NULL),
Expand Down
3 changes: 2 additions & 1 deletion src/box/relay.h
Original file line number Diff line number Diff line change
Expand Up @@ -124,6 +124,7 @@ relay_final_join(int fd, uint64_t sync, struct vclock *start_vclock,
*/
void
relay_subscribe(struct replica *replica, int fd, uint64_t sync,
struct vclock *replica_vclock, uint32_t replica_version_id);
struct vclock *replica_vclock, uint32_t replica_version_id,
unsigned int replica_id_mask);

#endif /* TARANTOOL_REPLICATION_RELAY_H_INCLUDED */
41 changes: 38 additions & 3 deletions src/box/xrow.c
Original file line number Diff line number Diff line change
Expand Up @@ -1194,17 +1194,23 @@ int
xrow_encode_subscribe(struct xrow_header *row,
const struct tt_uuid *replicaset_uuid,
const struct tt_uuid *instance_uuid,
const struct vclock *vclock, bool anon)
const struct vclock *vclock, bool anon,
unsigned int id_mask)
{
memset(row, 0, sizeof(*row));
size_t size = XROW_BODY_LEN_MAX + mp_sizeof_vclock(vclock);
unsigned int map_size = __builtin_popcount(id_mask);
if (map_size) {
size += mp_sizeof_array(map_size) + map_size *
mp_sizeof_uint(VCLOCK_MAX);
}
char *buf = (char *) region_alloc(&fiber()->gc, size);
if (buf == NULL) {
diag_set(OutOfMemory, size, "region_alloc", "buf");
return -1;
}
char *data = buf;
data = mp_encode_map(data, 5);
data = mp_encode_map(data, map_size ? 6 : 5);
data = mp_encode_uint(data, IPROTO_CLUSTER_UUID);
data = xrow_encode_uuid(data, replicaset_uuid);
data = mp_encode_uint(data, IPROTO_INSTANCE_UUID);
Expand All @@ -1215,6 +1221,17 @@ xrow_encode_subscribe(struct xrow_header *row,
data = mp_encode_uint(data, tarantool_version_id());
data = mp_encode_uint(data, IPROTO_REPLICA_ANON);
data = mp_encode_bool(data, anon);
if (map_size) {
data = mp_encode_uint(data, IPROTO_ID_MASK);
data = mp_encode_array(data, map_size);
struct bit_iterator it;
bit_iterator_init(&it, &id_mask, sizeof(id_mask),
true);
for (size_t id = bit_iterator_next(&it); id < VCLOCK_MAX;
id = bit_iterator_next(&it)) {
data = mp_encode_uint(data, id);
}
}
assert(data <= buf + size);
row->body[0].iov_base = buf;
row->body[0].iov_len = (data - buf);
Expand All @@ -1226,7 +1243,8 @@ xrow_encode_subscribe(struct xrow_header *row,
int
xrow_decode_subscribe(struct xrow_header *row, struct tt_uuid *replicaset_uuid,
struct tt_uuid *instance_uuid, struct vclock *vclock,
uint32_t *version_id, bool *anon)
uint32_t *version_id, bool *anon,
unsigned int *id_mask)
{
if (row->bodycnt == 0) {
diag_set(ClientError, ER_INVALID_MSGPACK, "request body");
Expand All @@ -1244,6 +1262,8 @@ xrow_decode_subscribe(struct xrow_header *row, struct tt_uuid *replicaset_uuid,

if (anon)
*anon = false;
if (id_mask)
*id_mask = 0;
d = data;
uint32_t map_size = mp_decode_map(&d);
for (uint32_t i = 0; i < map_size; i++) {
Expand Down Expand Up @@ -1301,6 +1321,21 @@ xrow_decode_subscribe(struct xrow_header *row, struct tt_uuid *replicaset_uuid,
}
*anon = mp_decode_bool(&d);
break;
case IPROTO_ID_MASK:
if (id_mask == NULL)
goto skip;
if (mp_typeof(*d) != MP_ARRAY) {
decode_err: xrow_on_decode_err(data, end, ER_INVALID_MSGPACK,
"invalid id_mask");
return -1;
}
uint32_t len = mp_decode_array(&d);
for(uint32_t i = 0; i < len; ++i) {
if (mp_typeof(*d) != MP_UINT)
goto decode_err;
*id_mask |= 1 << mp_decode_uint(&d);
}
break;
default: skip:
mp_next(&d); /* value */
}
Expand Down
34 changes: 23 additions & 11 deletions src/box/xrow.h
Original file line number Diff line number Diff line change
Expand Up @@ -322,6 +322,8 @@ xrow_encode_register(struct xrow_header *row,
* @param instance_uuid Instance uuid.
* @param vclock Replication clock.
* @param anon Whether it is an anonymous subscribe request or not.
* @param id_mask A List of replica ids to skip rows from
* when feeding a replica.
*
* @retval 0 Success.
* @retval -1 Memory error.
Expand All @@ -330,7 +332,8 @@ int
xrow_encode_subscribe(struct xrow_header *row,
const struct tt_uuid *replicaset_uuid,
const struct tt_uuid *instance_uuid,
const struct vclock *vclock, bool anon);
const struct vclock *vclock, bool anon,
unsigned int id_mask);

/**
* Decode SUBSCRIBE command.
Expand All @@ -340,14 +343,17 @@ xrow_encode_subscribe(struct xrow_header *row,
* @param[out] vclock.
* @param[out] version_id.
* @param[out] anon Whether it is an anonymous subscribe.
* @param[out] id_mask A list of ids to skip rows from when
* feeding replica.
*
* @retval 0 Success.
* @retval -1 Memory or format error.
*/
int
xrow_decode_subscribe(struct xrow_header *row, struct tt_uuid *replicaset_uuid,
struct tt_uuid *instance_uuid, struct vclock *vclock,
uint32_t *version_id, bool *anon);
uint32_t *version_id, bool *anon,
unsigned int *id_mask);

/**
* Encode JOIN command.
Expand All @@ -371,7 +377,8 @@ xrow_encode_join(struct xrow_header *row, const struct tt_uuid *instance_uuid);
static inline int
xrow_decode_join(struct xrow_header *row, struct tt_uuid *instance_uuid)
{
return xrow_decode_subscribe(row, NULL, instance_uuid, NULL, NULL, NULL);
return xrow_decode_subscribe(row, NULL, instance_uuid, NULL, NULL, NULL,
NULL);
}

/**
Expand All @@ -386,7 +393,8 @@ static inline int
xrow_decode_register(struct xrow_header *row, struct tt_uuid *instance_uuid,
struct vclock *vclock)
{
return xrow_decode_subscribe(row, NULL, instance_uuid, vclock, NULL, NULL);
return xrow_decode_subscribe(row, NULL, instance_uuid, vclock, NULL,
NULL, NULL);
}

/**
Expand All @@ -411,7 +419,7 @@ xrow_encode_vclock(struct xrow_header *row, const struct vclock *vclock);
static inline int
xrow_decode_vclock(struct xrow_header *row, struct vclock *vclock)
{
return xrow_decode_subscribe(row, NULL, NULL, vclock, NULL, NULL);
return xrow_decode_subscribe(row, NULL, NULL, vclock, NULL, NULL, NULL);
}

/**
Expand Down Expand Up @@ -442,7 +450,8 @@ xrow_decode_subscribe_response(struct xrow_header *row,
struct tt_uuid *replicaset_uuid,
struct vclock *vclock)
{
return xrow_decode_subscribe(row, replicaset_uuid, NULL, vclock, NULL, NULL);
return xrow_decode_subscribe(row, replicaset_uuid, NULL, vclock, NULL,
NULL, NULL);
}

/**
Expand Down Expand Up @@ -817,22 +826,25 @@ static inline void
xrow_encode_subscribe_xc(struct xrow_header *row,
const struct tt_uuid *replicaset_uuid,
const struct tt_uuid *instance_uuid,
const struct vclock *vclock, bool anon)
const struct vclock *vclock, bool anon,
unsigned int id_mask)
{
if (xrow_encode_subscribe(row, replicaset_uuid, instance_uuid,
vclock, anon) != 0)
vclock, anon, id_mask) != 0)
diag_raise();
}

/** @copydoc xrow_decode_subscribe. */
static inline void
xrow_decode_subscribe_xc(struct xrow_header *row,
struct tt_uuid *replicaset_uuid,
struct tt_uuid *instance_uuid, struct vclock *vclock,
uint32_t *replica_version_id, bool *anon)
struct tt_uuid *instance_uuid, struct vclock *vclock,
uint32_t *replica_version_id, bool *anon,
unsigned int *id_mask)
{
if (xrow_decode_subscribe(row, replicaset_uuid, instance_uuid,
vclock, replica_version_id, anon) != 0)
vclock, replica_version_id, anon,
id_mask) != 0)
diag_raise();
}

Expand Down

0 comments on commit ea38145

Please sign in to comment.