Introduce RESET request

Signed-off-by: Cole Miller <cole.miller@canonical.com>

src/gateway.c

@@ -1268,6 +1268,42 @@ static int handle_weight(struct handle *req)
 	return 0;
 }
 
+static void resetCb(struct exec *exec, int status)
+{
+	tracef("reset cb status:%d", status);
+	struct gateway *g = exec->data;
+	struct handle *req = g->req;
+	struct response_empty response = {0};
+	g->req = NULL;
+	if (status != 0) {
+		failure(req, status, "database reset failed");
+		return;
+	}
+	/* status and exec->status should always match. */
+	assert(exec->status == 0);
+	SUCCESS(empty, EMPTY);
+}
+
+static int handle_reset(struct handle *req)
+{
+	tracef("handle reset");
+	struct cursor *cursor = &req->cursor;
+	struct gateway *g = req->gateway;
+	int rc;
+
+	START_V0(reset, empty);
+	(void)response;
+	CHECK_LEADER(req);
+	LOOKUP_DB(request.db_id);
+	FAIL_IF_CHECKPOINTING;
+
+	g->req = req;
+	rc = leader__reset(g->leader, &g->exec, resetCb);
+	/* leader__reset reports all errors through the callback. */
+	assert(rc == 0);
+	return 0;
+}
+
 int gateway__handle(struct gateway *g,
 		    struct handle *req,
 		    int type,

src/leader.c

@@ -450,3 +450,49 @@ int leader__barrier(struct leader *l, struct barrier *barrier, barrier_cb cb)
 	}
 	return 0;
 }
+
+int leader__reset(struct leader *l, struct exec *exec, exec_cb cb)
+{
+	tracef("leader reset");
+	sqlite3_vfs *vfs;
+	dqlite_vfs_frame *frames;
+	unsigned n, i;
+	int rv;
+
+	exec->leader = l;
+	memset(&exec->barrier, 0, sizeof(exec->barrier));
+	exec->stmt = NULL;
+	exec->cb = cb;
+	l->exec = exec;
+
+	rv = sqlite3_db_config(l->conn, SQLITE_DBCONFIG_RESET_DATABASE, 1, 0);
+	assert(rv == 0);
+	exec->status = sqlite3_exec(l->conn, "VACUUM", NULL, NULL, NULL);
+	rv = sqlite3_db_config(l->conn, SQLITE_DBCONFIG_RESET_DATABASE, 0, 0);
+	assert(rv == 0);
+
+	vfs = sqlite3_vfs_find(l->db->config->name);
+	assert(vfs != NULL);
+	rv = VfsPoll(vfs, l->db->filename, &frames, &n);
+	if (rv != 0 || n == 0) {
+		goto finish;
+	}
+
+	rv = leaderApplyFrames(exec, frames, n);
+	for (i = 0; i < n; i++) {
+		sqlite3_free(frames[i].data);
+	}
+	sqlite3_free(frames);
+	if (rv != 0) {
+		VfsAbort(vfs, l->db->filename);
+		goto finish;
+	}
+	return 0;
+
+finish:
+	if (rv != 0) {
+		l->exec->status = rv;
+	}
+	leaderExecDone(l->exec);
+	return 0;
+}

src/leader.h

@@ -108,4 +108,6 @@ int leader__exec(struct leader *l,
  */
 int leader__barrier(struct leader *l, struct barrier *barrier, barrier_cb cb);
 
+int leader__reset(struct leader *l, struct exec *exec, exec_cb cb);
+
 #endif /* LEADER_H_*/

src/protocol.h

@@ -48,7 +48,8 @@ enum {
     DQLITE_REQUEST_CLUSTER,
     DQLITE_REQUEST_TRANSFER,
     DQLITE_REQUEST_DESCRIBE,
-    DQLITE_REQUEST_WEIGHT
+    DQLITE_REQUEST_WEIGHT,
+    DQLITE_REQUEST_RESET
 };
 
 #define DQLITE_REQUEST_CLUSTER_FORMAT_V0 0 /* ID and address */

src/request.h

@@ -42,6 +42,7 @@
 #define REQUEST_TRANSFER(X, ...) X(uint64, id, ##__VA_ARGS__)
 #define REQUEST_DESCRIBE(X, ...) X(uint64, format, ##__VA_ARGS__)
 #define REQUEST_WEIGHT(X, ...) X(uint64, weight, ##__VA_ARGS__)
+#define REQUEST_RESET(X, ...) X(uint64, db_id, ##__VA_ARGS__)
 
 #define REQUEST__DEFINE(LOWER, UPPER, _) \
 	SERIALIZE__DEFINE(request_##LOWER, REQUEST_##UPPER);
@@ -64,7 +65,8 @@
 	X(cluster, CLUSTER, __VA_ARGS__)     \
 	X(transfer, TRANSFER, __VA_ARGS__)   \
 	X(describe, DESCRIBE, __VA_ARGS__)   \
-	X(weight, WEIGHT, __VA_ARGS__)
+	X(weight, WEIGHT, __VA_ARGS__)       \
+	X(reset, RESET, __VA_ARGS__)
 
 REQUEST__TYPES(REQUEST__DEFINE);
 

test/unit/test_gateway.c

@@ -2077,6 +2077,57 @@ TEST_CASE(request_cluster, unrecognizedFormat, NULL)
 	return MUNIT_OK;
 }
 
+/******************************************************************************
+ *
+ * reset
+ *
+ ******************************************************************************/
+
+struct reset_fixture
+{
+	FIXTURE;
+	struct request_reset request;
+	struct response_empty response;
+};
+
+TEST_SUITE(reset);
+TEST_SETUP(reset)
+{
+	struct reset_fixture *f = munit_malloc(sizeof *f);
+	SETUP;
+	CLUSTER_ELECT(0);
+	OPEN;
+	return f;
+}
+TEST_TEAR_DOWN(reset)
+{
+	struct reset_fixture *f = data;
+	TEAR_DOWN;
+	free(f);
+}
+
+TEST_CASE(reset, simple, NULL)
+{
+	struct reset_fixture *f = data;
+	(void)params;
+	EXEC("CREATE TABLE test (n INT)");
+	EXEC("INSERT INTO test VALUES (1)");
+	EXEC("INSERT INTO test VALUES (2)");
+	EXEC("INSERT INTO test VALUES (3)");
+	f->request.db_id = 0;
+	ENCODE(&f->request, reset);
+	HANDLE(RESET);
+	WAIT;
+	ASSERT_CALLBACK(0, EMPTY);
+
+	CLUSTER_DEPOSE;
+	SELECT(1);
+	CLUSTER_ELECT(1);
+	OPEN;
+	EXEC("CREATE TABLE test (n INT)");
+	return MUNIT_OK;
+}
+
 /******************************************************************************
  *
  * invalid