aboutsummaryrefslogtreecommitdiff
path: root/script/jepsen.garage/src/jepsen
diff options
context:
space:
mode:
authorAlex Auvolat <alex@adnab.me>2023-04-18 16:10:07 +0200
committerAlex Auvolat <alex@adnab.me>2023-10-18 12:03:11 +0200
commitbc11701999ececdb4571119082945cbe86f098aa (patch)
tree46b79664947521dd848f686668538e7a3f73b6e4 /script/jepsen.garage/src/jepsen
parentca4cc7e44f6143ec24be626c683ed5b14ff40295 (diff)
downloadgarage-bc11701999ececdb4571119082945cbe86f098aa.tar.gz
garage-bc11701999ececdb4571119082945cbe86f098aa.zip
jepsen: s3 gets and puts
Diffstat (limited to 'script/jepsen.garage/src/jepsen')
-rw-r--r--script/jepsen.garage/src/jepsen/garage.clj140
1 files changed, 136 insertions, 4 deletions
diff --git a/script/jepsen.garage/src/jepsen/garage.clj b/script/jepsen.garage/src/jepsen/garage.clj
index 6617064e..26c58097 100644
--- a/script/jepsen.garage/src/jepsen/garage.clj
+++ b/script/jepsen.garage/src/jepsen/garage.clj
@@ -1,14 +1,146 @@
(ns jepsen.garage
- (:require [jepsen.cli :as cli]
- [jepsen.tests :as tests]))
+ (:require [clojure.tools.logging :refer :all]
+ [clojure.string :as str]
+ [jepsen [cli :as cli]
+ [client :as client]
+ [control :as c]
+ [db :as db]
+ [generator :as gen]
+ [tests :as tests]]
+ [jepsen.control.util :as cu]
+ [jepsen.os.debian :as debian]
+ [amazonica.aws.s3 :as s3]
+ [amazonica.aws.s3transfer :as s3transfer]))
+
+(def dir "/opt/garage")
+(def binary (str dir "/garage"))
+(def logfile (str dir "/garage.log"))
+(def pidfile (str dir "/garage.pid"))
+
+(def grg-admin-token "icanhazadmin")
+(def grg-key "jepsen")
+(def grg-bucket "jepsen")
+(def grg-object "1")
+
+(defn db
+ "Garage DB for a particular version"
+ [version]
+ (reify db/DB
+ (setup! [_ test node]
+ (info node "installing garage" version)
+ (c/su
+ (c/exec :mkdir :-p dir)
+ (let [url (str "https://garagehq.deuxfleurs.fr/_releases/" version "/x86_64-unknown-linux-musl/garage")
+ cache (cu/wget! url)]
+ (c/exec :cp cache binary))
+ (c/exec :chmod :+x binary)
+ (cu/write-file!
+ (str "rpc_secret = \"0fffabe52542c2b89a56b2efb7dfd477e9dafb285c9025cbdf1de7ca21a6b372\"\n"
+ "rpc_bind_addr = \"0.0.0.0:3901\"\n"
+ "rpc_public_addr = \"" node ":3901\"\n"
+ "db_engine = \"lmdb\"\n"
+ "replication_mode = \"3\"\n"
+ "data_dir = \"" dir "/data\"\n"
+ "metadata_dir = \"" dir "/meta\"\n"
+ "[s3_api]\n"
+ "s3_region = \"us-east-1\"\n"
+ "api_bind_addr = \"0.0.0.0:3900\"\n"
+ "[k2v_api]\n"
+ "api_bind_addr = \"0.0.0.0:3902\"\n"
+ "[admin]\n"
+ "api_bind_addr = \"0.0.0.0:3903\"\n"
+ "admin_token = \"" grg-admin-token "\"\n")
+ "/etc/garage.toml")
+ (cu/start-daemon!
+ {:logfile logfile
+ :pidfile pidfile
+ :chdir dir}
+ binary
+ :server)
+ (Thread/sleep 100)
+ (let [node-id (c/exec binary :node :id :-q)]
+ (info node "node id:" node-id)
+ (c/on-many (:nodes test)
+ (c/exec binary :node :connect node-id))
+ (c/exec binary :layout :assign (subs node-id 0 16) :-c 1 :-z :dc1 :-t node))
+ (if (= node (first (:nodes test)))
+ (do
+ (Thread/sleep 2000)
+ (c/exec binary :layout :apply :--version 1)
+ (info node "garage status:" (c/exec binary :status))
+ (c/exec binary :key :new :--name grg-key)
+ (c/exec binary :bucket :create grg-bucket)
+ (c/exec binary :bucket :allow :--read :--write grg-bucket :--key grg-key)
+ (info node "key info: " (c/exec binary :key :info grg-key))))))
+ (teardown! [_ test node]
+ (info node "tearing down garage" version)
+ (c/su
+ (cu/stop-daemon! binary pidfile)
+ (c/exec :rm :-rf dir)))
+ db/LogFiles
+ (log-files [_ test node]
+ [logfile])))
+
+(defn op-get [_ _] {:type :invoke, :f :get-object, :value nil})
+(defn op-put [_ _] {:type :invoke, :f :put-object, :value (str (rand-int 50))})
+(defn op-del [_ _] {:type :invoke, :f :del-object, :value nil})
+
+(defrecord Client [creds]
+ client/Client
+ (open! [this test node]
+ (let [key-info (c/on node (c/exec binary :key :info grg-key))
+ [_ ak sk] (re-matches
+ #"(?s).*Key ID: (.*)\nSecret key: (.*)\nCan create.*"
+ key-info)
+ creds {:access-key ak
+ :secret-key sk
+ :endpoint (str "http://" node ":3900")
+ :client-config {:path-style-access-enabled true}}]
+ (info node "s3 credentials:" creds)
+ (assoc this :creds creds)))
+ (setup! [this test])
+ (invoke! [this test op]
+ (case (:f op)
+ :get-object
+ (let [value-bytes (try
+ (-> (s3/get-object (:creds this) grg-bucket grg-object)
+ :input-stream
+ slurp)
+ (catch Exception e nil))]
+ (assoc op :type :ok, :value value-bytes))
+ :put-object
+ (let [some-bytes (.getBytes (:value op) "UTF-8")
+ bytes-stream (java.io.ByteArrayInputStream. some-bytes)]
+ (s3/put-object (:creds this)
+ :bucket-name grg-bucket
+ :key grg-object
+ :input-stream bytes-stream
+ :metadata {:content-length (count some-bytes)})
+ (assoc op :type :ok))
+ :del-object
+ (do
+ (s3/delete-object (:creds this)
+ :bucket-name grg-bucket
+ :key grg-object)
+ (assoc op :type :ok, :value nil))))
+ (teardown! [this test])
+ (close! [this test]))
(defn garage-test
"Given an options map from the command line runner (e.g. :nodes, :ssh,
:concurrency, ...), constructs a test map."
[opts]
(merge tests/noop-test
- {:pure-generators true}
- opts))
+ opts
+ {:pure-generators true
+ :name "garage"
+ :os debian/os
+ :db (db "v0.8.2")
+ :client (Client. nil)
+ :generator (->> (gen/mix [op-get op-put])
+ (gen/stagger 1)
+ (gen/nemesis nil)
+ (gen/time-limit 15))}))
(defn -main
"Handles command line arguments. Can either run a test, or a web server for