Add Zocalo config plugin to set default transport

DiamondLightSource · Oct 7, 2021 · 73ba66b · 73ba66b
1 parent 5425406
commit 73ba66b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -55,6 +55,7 @@ workflows.transport =
 zocalo.configuration.plugins =
  pika = workflows.util.zocalo.configuration:Pika
  stomp = workflows.util.zocalo.configuration:Stomp
+ transport = workflows.util.zocalo.configuration:DefaultTransport
 
 [options.packages.find]
 where = src

diff --git a/src/workflows/util/zocalo/configuration.py b/src/workflows/util/zocalo/configuration.py
@@ -1,6 +1,7 @@
 from marshmallow import fields
 from zocalo.configuration import PluginSchema
 
+import workflows.transport
 from workflows.transport.pika_transport import PikaTransport
 from workflows.transport.stomp_transport import StompTransport
 
@@ -47,3 +48,14 @@ def activate(configuration):
  ("vhost", "--rabbit-vhost"),
  ]:
  PikaTransport.defaults[target] = configuration[cfgoption]
+
+
+class DefaultTransport:
+ """A Zocalo configuration plugin to set a default transport class"""
+
+ class Schema(PluginSchema):
+ default = fields.Str(required=True)
+
+ @staticmethod
+ def activate(configuration):
+ workflows.transport.default_transport = configuration["default"]