Only call Oj.mimic_JSON when in a CLI run

Shopify · Jan 8, 2021 · 5ba4d92 · 5ba4d92
1 parent bc53f2c
commit 5ba4d92
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/lib/krane.rb b/lib/krane.rb
@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'oj'
 require 'krane/deploy_task'
 require 'krane/render_task'
 require 'krane/restart_task'

diff --git a/lib/krane/cli/krane.rb b/lib/krane/cli/krane.rb
@@ -17,6 +17,11 @@ class Krane < Thor
 
       package_name "Krane"
 
+      def self.start(*)
+        Oj.mimic_JSON
+        super
+      end
+
       def self.expand_options(task_options)
         task_options.each { |option_name, config| method_option(option_name, config) }
       end

diff --git a/lib/krane/common.rb b/lib/krane/common.rb
@@ -11,7 +11,6 @@
 require 'colorized_string'
 
 require 'krane/version'
-require 'krane/oj'
 require 'krane/errors'
 require 'krane/formatted_logger'
 require 'krane/statsd'

diff --git a/lib/krane/oj.rb b/lib/krane/oj.rb