lib/dryad/consul/service_registry.rb in dryad-consul-0.2.6 vs lib/dryad/consul/service_registry.rb in dryad-consul-0.2.7

@@ -2,16 +2,16 @@
   module Consul
     class ServiceRegistry
       class << self
         def register(service)
           service.to_registers.each do |register|
-            ServiceClient.register(register)
+            ::Diplomat::Service.register(register)
           end
         end
 
         def deregister(service)
           service.portals.each do |portal|
-            ServiceClient.deregister(portal.id)
+            ::Diplomat::Service.deregister(portal.id)
           end
         end
 
         def service_instances(name, schema, groups)
           all_services = ::Diplomat::Health.service(Dryad::Core::Service.full_name(schema, name))