cockroachdb: use CONFIGURE ZONE in set-replication-zone!

Fixes cockroachdb/cockroach#39185. Fixes cockroachdb/cockroach#39208. Fixes cockroachdb/cockroach#39209. Fixes cockroachdb/cockroach#39210. Fixes cockroachdb/cockroach#39212. Fixes cockroachdb/cockroach#39214. Fixes cockroachdb/cockroach#39215. Fixes cockroachdb/cockroach#39217. Fixes cockroachdb/cockroach#39216. Fixes cockroachdb/cockroach#39218. Fixes cockroachdb/cockroach#39219. Fixes cockroachdb/cockroach#39220. Fixes cockroachdb/cockroach#39221. Fixes cockroachdb/cockroach#39222. Fixes cockroachdb/cockroach#39223. Fixes cockroachdb/cockroach#39224. Fixes cockroachdb/cockroach#39226. Fixes cockroachdb/cockroach#39225. Fixes cockroachdb/cockroach#39227. Fixes cockroachdb/cockroach#39228. Fixes cockroachdb/cockroach#39229. Fixes cockroachdb/cockroach#39230. Fixes cockroachdb/cockroach#39231. Fixes cockroachdb/cockroach#39234. Fixes cockroachdb/cockroach#39233. Fixes cockroachdb/cockroach#39235. Fixes cockroachdb/cockroach#39236. Fixes cockroachdb/cockroach#39238. Fixes cockroachdb/cockroach#39237. The `cockroach zone` command was deprecated in https://github.com/cockroachdb/cockroach/pull/39177a.
aliher1911 · Dec 24, 2021 · 5369ce1 · 5369ce1
1 parent b10f046
commit 5369ce1
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 25 deletions.
diff --git a/cockroachdb/project.clj b/cockroachdb/project.clj
@@ -6,7 +6,6 @@
   :dependencies [[org.clojure/clojure "1.8.0"]
                  [jepsen "0.1.9-SNAPSHOT"]
                  [org.clojure/java.jdbc "0.6.1"]
-                 [circleci/clj-yaml "0.5.5"]
                  [org.postgresql/postgresql "9.4.1211"]]
   :jvm-opts ["-Xmx12g"
              "-XX:+UseConcMarkSweepGC"

diff --git a/cockroachdb/src/jepsen/cockroach.clj b/cockroachdb/src/jepsen/cockroach.clj
@@ -72,7 +72,7 @@
 
                 (jepsen/synchronize test)
                 (when (= node (jepsen/primary test))
-                  (auto/set-replication-zone! ".default"
+                  (auto/set-replication-zone! "default"
                                               {:range_min_bytes 1024
                                                :range_max_bytes 1048576})
                   (info node "Creating database...")

diff --git a/cockroachdb/src/jepsen/cockroach/auto.clj b/cockroachdb/src/jepsen/cockroach/auto.clj
@@ -3,7 +3,6 @@
   (:require [clojure.tools.logging :refer :all]
             [clojure.java.io :as io]
             [clojure.string :as str]
-            [clj-yaml.core :as yaml]
             [jepsen.util :as util]
             [jepsen [core :as jepsen]
                     [control :as c :refer [|]]]
@@ -100,29 +99,13 @@
            [:-e ~@body]
            [:>> errlog (c/lit "2>&1")]))))
 
-(defn replication-zone
-  "Gets the replication zone on a given node. With no args, fetches the default
-  zone."
-  ([]
-   (replication-zone ".default"))
-  ([name]
-   (yaml/parse-string
-     (c/cd working-path
-           (c/exec cockroach :zone :get name (when insecure :--insecure))))))
-
 (defn set-replication-zone!
-  "Sets the replication zone on the given node. Returns the new replication
-  zone."
-  [name zone]
-  (c/sudo cockroach-user
-          (-> (c/cd working-path
-                    (c/exec :echo (yaml/generate-string zone) |
-                            cockroach :zone :set
-                            (when insecure :--insecure)
-                            :--file=-
-                            name))
-              (str/replace #"UPDATE .+\n" "")
-              (yaml/parse-string))))
+  "Sets the replication zone on the given node."
+  [zone config]
+  (csql! (str
+    "alter range " zone " configure zone using "
+    (str/join ", "
+      (map (fn [[k v]] (str (name k) " = " v)) config)))))
 
 (defn install-bumptime!
   "Install time adjusting binary"