lib/protobuf/nats.rb in protobuf-nats-0.11.0.pre1 vs lib/protobuf/nats.rb in protobuf-nats-0.12.0.pre0

@@ -4,10 +4,11 @@
 # We don't need this, but the CLI attempts to terminate.
 require "protobuf/rpc/service_directory"
 
 require "nats/io/client"
 
+require "protobuf/nats/platform"
 require "protobuf/nats/errors"
 require "protobuf/nats/client"
 require "protobuf/nats/server"
 require "protobuf/nats/runner"
 require "protobuf/nats/config"
@@ -21,10 +22,10 @@
     module Messages
       ACK  = "\1".freeze
       NACK = "\2".freeze
     end
 
-    NatsClient = if defined? JRUBY_VERSION
+    NatsClient = if jruby?
                    require "protobuf/nats/jnats"
                    ::Protobuf::Nats::JNats
                  else
                    ::NATS::IO::Client
                  end