CCOL-2039: Evaluate schemaclass schema and namespace when setting dec…

…oder
flipp-oss · Dec 18, 2023 · b61a403 · b61a403
1 parent ceff055
commit b61a403
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/lib/deimos/consumer.rb b/lib/deimos/consumer.rb
@@ -19,8 +19,16 @@ class Consumer
     class << self
       # @return [Deimos::SchemaBackends::Base]
       def decoder
-        @decoder ||= Deimos.schema_backend(schema: config[:schema],
-                                           namespace: config[:namespace])
+        # @decoder ||= Deimos.schema_backend(schema: config[:schema], namespace: config[:namespace])
+
+        return @decoder if @decoder
+
+        @decoder = if Utils::SchemaClass.use?(config.to_h)
+                     schema_class = "Schemas::#{config[:schema]}".constantize.new
+                     Deimos.schema_backend(schema: schema_class.schema, namespace: schema_class.namespace)
+                   else
+                     Deimos.schema_backend(schema: config[:schema], namespace: config[:namespace])
+                   end
       end
 
       # @return [Deimos::SchemaBackends::Base]