go/libraries/doltcore/sqle/cluster: Add replication of user and grant changes from primary to standby replicas.

2026-04-23 21:59:01 -05:00 · 2023-07-31 17:17:45 -07:00
parent 98dfb0d4de
commit 08f91f74a0
13 changed files with 886 additions and 9 deletions
@@ -28,6 +28,10 @@ func TestCluster(t *testing.T) {
 	RunTestsFile(t, "tests/sql-server-cluster.yaml")
 }

+func TestClusterUsersAndGrants(t *testing.T) {
+	RunTestsFile(t, "tests/sql-server-cluster-users-and-grants.yaml")
+}
+
 func TestRemotesAPI(t *testing.T) {
 	RunTestsFile(t, "tests/sql-server-remotesapi.yaml")
 }
@@ -35,7 +39,7 @@ func TestRemotesAPI(t *testing.T) {
 // TestSingle is a convenience method for running a single test from within an IDE. Unskip and set to the file and name
 // of the test you want to debug. See README.md in the `tests` directory for more debugging info.
 func TestSingle(t *testing.T) {
-	// t.Skip()
+	t.Skip()
 	RunSingleTest(t, "tests/sql-server-cluster.yaml", "primary comes up and replicates to standby")
 }

@@ -0,0 +1,145 @@
+tests:
+- name: users and grants replicate
+  multi_repos:
+  - name: server1
+    with_files:
+    - name: server.yaml
+      contents: |
+        log_level: trace
+        listener:
+          host: 0.0.0.0
+          port: 3309
+        cluster:
+          standby_remotes:
+          - name: standby
+            remote_url_template: http://localhost:3852/{database}
+          bootstrap_role: primary
+          bootstrap_epoch: 1
+          remotesapi:
+            port: 3851
+    server:
+      args: ["--config", "server.yaml"]
+      port: 3309
+  - name: server2
+    with_files:
+    - name: server.yaml
+      contents: |
+        log_level: trace
+        listener:
+          host: 0.0.0.0
+          port: 3310
+        cluster:
+          standby_remotes:
+          - name: standby
+            remote_url_template: http://localhost:3851/{database}
+          bootstrap_role: standby
+          bootstrap_epoch: 1
+          remotesapi:
+            port: 3852
+    server:
+      args: ["--config", "server.yaml"]
+      port: 3310
+  connections:
+  - on: server1
+    queries:
+    - exec: 'SET @@PERSIST.dolt_cluster_ack_writes_timeout_secs = 10'
+    - exec: 'create database repo1'
+    - exec: "use repo1"
+    - exec: 'create table vals (i int primary key)'
+    - exec: 'insert into vals values (0),(1),(2),(3),(4)'
+    - exec: 'create user "aaron"@"%" IDENTIFIED BY "aaronspassword"'
+    - exec: 'grant ALL ON *.* to "aaron"@"%"'
+    - exec: 'insert into vals values (5),(6),(7),(8),(9)'
+  - on: server1
+    user: 'aaron'
+    password: 'aaronspassword'
+    queries:
+    - exec: "use repo1"
+    - exec: 'insert into vals values (10),(11),(12),(13),(14)'
+  - on: server2
+    user: 'aaron'
+    password: 'aaronspassword'
+    queries:
+    - exec: "use repo1"
+    - query: 'select count(*) from vals'
+      result:
+        columns: ["count(*)"]
+        rows: [["15"]]
+- name: users and grants applied to standby do not replicate
+### TODO: This test should not be possible; being able to run create user on a standby is a bug.
+  multi_repos:
+  - name: server1
+    with_files:
+    - name: server.yaml
+      contents: |
+        log_level: trace
+        listener:
+          host: 0.0.0.0
+          port: 3309
+        cluster:
+          standby_remotes:
+          - name: standby
+            remote_url_template: http://localhost:3852/{database}
+          bootstrap_role: primary
+          bootstrap_epoch: 1
+          remotesapi:
+            port: 3851
+    server:
+      args: ["--config", "server.yaml"]
+      port: 3309
+  - name: server2
+    with_files:
+    - name: server.yaml
+      contents: |
+        log_level: trace
+        listener:
+          host: 0.0.0.0
+          port: 3310
+        cluster:
+          standby_remotes:
+          - name: standby
+            remote_url_template: http://localhost:3851/{database}
+          bootstrap_role: standby
+          bootstrap_epoch: 1
+          remotesapi:
+            port: 3852
+    server:
+      args: ["--config", "server.yaml"]
+      port: 3310
+  connections:
+  - on: server1
+    queries:
+    - exec: 'SET @@PERSIST.dolt_cluster_ack_writes_timeout_secs = 10'
+    - exec: 'create database repo1'
+    - exec: "use repo1"
+    - exec: 'create table vals (i int primary key)'
+    - exec: 'insert into vals values (0),(1),(2),(3),(4)'
+    - exec: 'create user "aaron"@"%" IDENTIFIED BY "aaronspassword"'
+    - exec: 'grant ALL ON *.* to "aaron"@"%"'
+    - exec: 'insert into vals values (5),(6),(7),(8),(9)'
+  - on: server1
+    user: 'aaron'
+    password: 'aaronspassword'
+    queries:
+    - exec: "use repo1"
+    - exec: 'insert into vals values (10),(11),(12),(13),(14)'
+  - on: server2
+    user: 'aaron'
+    password: 'aaronspassword'
+    queries:
+    - exec: "use repo1"
+    - query: 'select count(*) from vals'
+      result:
+        columns: ["count(*)"]
+        rows: [["15"]]
+    - exec: 'create user "brian"@"%" IDENTIFIED BY "brianspassword"'
+    - exec: 'grant ALL ON *.* to "brian"@"%"'
+    - exec: 'select sleep(1) from dual'
+  - on: server1
+    user: 'aaron'
+    password: 'aaronspassword'
+    queries:
+    - query: "select count(*) from mysql.user where User = 'brian'"
+      result:
+        columns: ["count(*)"]
+        rows: [["0"]]