lib/rocket_job/cli.rb in rocketjob-3.1.0 vs lib/rocket_job/cli.rb in rocketjob-3.2.0

@@ -25,10 +25,11 @@
     end
 
     # Run a RocketJob::Server from the command line
     def run
       Thread.current.name = 'rocketjob main'
+      RocketJob.server!
       setup_environment
       setup_logger
       rails? ? boot_rails : boot_standalone
       write_pidfile
 
@@ -51,9 +52,10 @@
 
     # Initialize the Rails environment
     # Returns [true|false] whether Rails is present
     def boot_rails
       logger.info "Loading Rails environment: #{environment}"
+      RocketJob.rails!
 
       boot_file = Pathname.new(directory).join('config/environment.rb').expand_path
       require(boot_file.to_s)
 
       begin