mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-14 19:45:11 +00:00
51 lines
1.6 KiB
Clojure
51 lines
1.6 KiB
Clojure
(ns jepsen.clickhouse-keeper.set
|
|
(:require
|
|
[clojure.tools.logging :refer :all]
|
|
[jepsen
|
|
[checker :as checker]
|
|
[client :as client]
|
|
[generator :as gen]]
|
|
[jepsen.clickhouse-keeper.utils :refer :all]
|
|
[zookeeper :as zk])
|
|
(:import (org.apache.zookeeper ZooKeeper KeeperException KeeperException$BadVersionException)))
|
|
|
|
(defrecord SetClient [k conn nodename]
|
|
client/Client
|
|
(open! [this test node]
|
|
(assoc
|
|
(assoc this
|
|
:conn (zk-connect node 9181 30000))
|
|
:nodename node))
|
|
|
|
(setup! [this test]
|
|
(exec-with-retries 30 (fn []
|
|
(zk-create-if-not-exists conn k "#{}"))))
|
|
|
|
(invoke! [this test op]
|
|
(case (:f op)
|
|
:read (exec-with-retries 30 (fn []
|
|
(zk-sync conn)
|
|
(assoc op
|
|
:type :ok
|
|
:value (read-string (:data (zk-get-str conn k))))))
|
|
:add (try
|
|
(do
|
|
(zk-add-to-set conn k (:value op))
|
|
(assoc op :type :ok))
|
|
(catch KeeperException$BadVersionException _ (assoc op :type :fail, :error :bad-version))
|
|
(catch Exception _ (assoc op :type :info, :error :connect-error)))))
|
|
|
|
(teardown! [_ test])
|
|
|
|
(close! [_ test]
|
|
(zk/close conn)))
|
|
|
|
(defn workload
|
|
"A generator, client, and checker for a set test."
|
|
[opts]
|
|
{:client (SetClient. "/a-set" nil nil)
|
|
:checker (checker/set)
|
|
:generator (->> (range)
|
|
(map (fn [x] {:type :invoke, :f :add, :value x})))
|
|
:final-generator (gen/once {:type :invoke, :f :read, :value nil})})
|