Add support for TLS and namespace options

Signed-off-by: Greg Haskins <[email protected]>

Author: ghaskins

src/manetu/temporal_benchmark/main.clj

@@ -34,6 +34,11 @@
    ["-v" "--version" "Print version info and exit"]
    [nil "--temporal-target HOSTPORT" "The host:port of the Temporal cluster"
     :default "localhost:7233"]
+   [nil "--temporal-namespace NAMESPACE" "The Temporal namespace to use"
+    :default "default"]
+   [nil "--temporal-tls ENABLED" "Enable TLS for Temporal connection"
+    :default false]
+   [nil "--temporal-ca PATH" "Path to a PEM for CA trust-root for Temporal cluster"]
    [nil "--temporal-taskqueue QUEUENAME" "The temporal taskqueue to use"
     :default "temporal-benchmark"]
    ["-l" "--log-level LEVEL" loglevel-description

src/manetu/temporal_benchmark/utils.clj

@@ -4,7 +4,8 @@
   (:require [clojure.string :as string]
             [clojure.tools.cli :refer [parse-opts]]
             [slingshot.slingshot :refer [throw+ try+]]
-            [temporal.client.core :as c]))
+            [temporal.client.core :as c]
+            [temporal.tls :as tls]))
 
 (def tool-name "temporal-benchmark")
 
@@ -35,18 +36,24 @@
 (defn error-exit [error-msg summary]
   (exit -1 (str "Error: " error-msg "\n\n") summary))
 
-(defn global-option-error [field usage-summary]
-  (error-exit (str "global option '" field "' required") usage-summary))
-
 (defn round2
   "Round a double to the given precision (number of significant digits)"
   [precision ^double d]
   (let [factor (Math/pow 10 precision)]
     (/ (Math/round (* d factor)) factor)))
 
+(defn new-ssl-context [{:keys [temporal-ca] :as global-options}]
+  (-> {}
+      (cond-> temporal-ca (assoc :ca-path temporal-ca))
+      (tls/new-ssl-context)))
+
+(defn create-client [{:keys [temporal-target temporal-namespace temporal-tls] :as global-options}]
+  (c/create-client (-> {:target temporal-target
+                        :namespace temporal-namespace}
+                       (cond-> temporal-tls (assoc :ssl-context (new-ssl-context global-options))))))
+
 (defn exec-command
-  [{:keys [command description options-spec fn]} global-summary
-   {:keys [temporal-target] :as global-options} args]
+  [{:keys [command description options-spec fn]} global-summary global-options args]
   (let [{{:keys [help] :as local-options} :options
          :keys [errors summary]} (parse-opts args (cons ["-h" "--help"] options-spec))
         summary (subcommand-usage command description global-summary summary)]
@@ -60,5 +67,5 @@
 
       :else
       (let [options (merge global-options local-options)
-            client (c/create-client {:target temporal-target})]
+            client (create-client global-options)]
         (fn options client)))))