Adding protoversion in config file (cadence-workflow#4263)

avmi · Jun 23, 2021 · c4f2c09 · c4f2c09
1 parent 12ddac4
commit c4f2c09
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/docker/README.md b/docker/README.md
@@ -104,6 +104,7 @@ docker run -e CASSANDRA_SEEDS=10.x.x.x                  -- csv of cassandra serv
     -e KEYSPACE=<keyspace>                              -- Cassandra keyspace
     -e VISIBILITY_KEYSPACE=<visibility_keyspace>        -- Cassandra visibility keyspace, if using basic visibility 
     -e KAFKA_SEEDS=10.x.x.x                             -- Kafka broker seed, if using ElasticSearch + Kafka for advanced visibility feature
+    -e CASSANDRA_PROTOCOL_VERSION=<protocol_version>    -- Cassandra protocol version
     -e ES_SEEDS=10.x.x.x                                -- ElasticSearch seed , if using ElasticSearch + Kafka for advanced visibility feature
     -e RINGPOP_SEEDS=10.x.x.x,10.x.x.x                  -- csv of ipaddrs for gossip bootstrap
     -e STATSD_ENDPOINT=10.x.x.x:8125                    -- statsd server endpoint

diff --git a/docker/config_template.yaml b/docker/config_template.yaml
@@ -20,13 +20,15 @@ persistence:
                 keyspace: {{ default .Env.KEYSPACE "cadence" }}
                 user: {{ default .Env.CASSANDRA_USER "" }}
                 password: {{ default .Env.CASSANDRA_PASSWORD "" }}
+                protocolVersion: {{ default .Env.CASSANDRA_PROTOCOL_VERSION "4" }}
         visibility:
             nosql:
                 pluginName: "cassandra"
                 hosts: {{ default .Env.CASSANDRA_SEEDS "" }}
                 keyspace: {{ default .Env.VISIBILITY_KEYSPACE "cadence_visibility" }}
                 user: {{ default .Env.CASSANDRA_USER "" }}
                 password: {{ default .Env.CASSANDRA_PASSWORD "" }}
+                protocolVersion: {{ default .Env.CASSANDRA_PROTOCOL_VERSION "4" }}
         {{- else if eq $db "mysql" }}
         default:
             sql:

diff --git a/docker/start.sh b/docker/start.sh
@@ -13,6 +13,7 @@ export CASSANDRA_USER="${CASSANDRA_USER:-cassandra}"
 export CASSANDRA_PASSWORD="${CASSANDRA_PASSWORD:-cassandra}"
 export KEYSPACE="${KEYSPACE:-cadence}"
 export VISIBILITY_KEYSPACE="${VISIBILITY_KEYSPACE:-cadence_visibility}"
+export CASSANDRA_PROTOCOL_VERSION="${CASSANDRA_PROTOCOL_VERSION:-4}"
 
 # mysql env
 export DBNAME="${DBNAME:-cadence}"
@@ -84,7 +85,7 @@ setup_schema() {
 
 wait_for_cassandra() {
     server=`echo $CASSANDRA_SEEDS | awk -F ',' '{print $1}'`
-    until cqlsh -u $CASSANDRA_USER -p $CASSANDRA_PASSWORD --cqlversion=3.4.4 $server < /dev/null; do
+    until cqlsh -u $CASSANDRA_USER -p $CASSANDRA_PASSWORD --cqlversion=3.4.4 --protocol-version=$CASSANDRA_PROTOCOL_VERSION $server < /dev/null; do
         echo 'waiting for cassandra to start up'
         sleep 1
     done
@@ -93,7 +94,7 @@ wait_for_cassandra() {
 
 wait_for_scylla() {
     server=`echo $CASSANDRA_SEEDS | awk -F ',' '{print $1}'`
-    until cqlsh -u $CASSANDRA_USER -p $CASSANDRA_PASSWORD --cqlversion=3.3.1 $server < /dev/null; do
+    until cqlsh -u $CASSANDRA_USER -p $CASSANDRA_PASSWORD --cqlversion=3.3.1 --protocol-version=$CASSANDRA_PROTOCOL_VERSION $server < /dev/null; do
         echo 'waiting for scylla to start up'
         sleep 1
     done