dryad 0.1.0

data/dryad-consul/lib/dryad/consul/service.rb

@@ -0,0 +1,33 @@
+module Dryad
+  module Consul
+    class Service < Dryad::Core::Service
+      def to_registers
+        portals.map do |portal|
+          {
+            ID: portal.id,
+            Name: Service.full_name(portal.schema, name),
+            Address: address,
+            Port: portal.port,
+            EnableTagOverride: true,
+            Tags: tags(portal)
+          }
+        end
+      end
+
+      private
+      def tags(portal)
+        tags = [
+          "type = \"#{type_name}\"",
+          "priority = \"#{priority}\"",
+          "group = \"#{group}\"",
+          "schema = \"#{portal.schema}\"",
+          "pattern = \"#{portal.pattern}\""
+        ]
+        if portal.non_certifications.count > 0
+          tags << "non_certifications = \"#{portal.non_certifications.join(",")}\""
+        end
+        tags.concat(load_balancing.map{|lb| "load_balancing = \"#{lb}\"" })
+      end
+    end
+  end
+end

data/dryad-consul/lib/dryad/consul/service_client.rb

@@ -0,0 +1,15 @@
+module Dryad
+  module Consul
+    class ServiceClient
+      class << self
+        def method_missing(name, *args, &block)
+          Diplomat::Service.send(name, *args, &block) || super
+        end
+
+        def respond_to_missing?(meth_id, with_private = false)
+          access_method?(meth_id) || super
+        end
+      end
+    end
+  end
+end

data/dryad-consul/lib/dryad/consul/service_registry.rb

@@ -0,0 +1,37 @@
+module Dryad
+  module Consul
+    class ServiceRegistry
+      class << self
+        def register(service)
+          service.to_registers.each do |register|
+            ServiceClient.register(register)
+          end
+        end
+
+        def deregister(service)
+          service.portals.each do |portal|
+            ServiceClient.deregister(portal.id)
+          end
+        end
+
+        def service_instances(name, schema, groups)
+          all_services = ServiceClient.get(Dryad::Core::Service.full_name(schema, name), :all)
+          services = []
+          all_services.each do |service|
+            groups.each do |group|
+              services << service if service.ServiceTags.include?("group = \"#{group}\"")
+            end
+          end
+          services.map do |service|
+            Dryad::Core::ServiceInstance.new(
+              name: name,
+              schema: schema,
+              address: service.ServiceAddress,
+              port: service.ServicePort
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/dryad-consul/lib/dryad/consul/version.rb

@@ -0,0 +1,10 @@
+module Dryad
+  module Consul
+    MAJOR = 0
+    MINOR = 1
+    TINY  = 0
+    PRE   = nil
+  
+    VERSION = [MAJOR, MINOR, TINY, PRE].compact.join(".")
+  end
+end

data/dryad-consul/lib/dryad/consul.rb

@@ -0,0 +1,16 @@
+require "diplomat"
+require "dryad"
+require "dryad/consul/version"
+require "dryad/consul/service_client"
+require "dryad/consul/service_registry"
+require "dryad/consul/service"
+
+module Dryad
+  module Consul
+    class Error < StandardError; end
+
+    ::Diplomat.configure do |config|
+      config.url = "http://#{Dryad.configuration.consul[:host]}:#{Dryad.configuration.consul[:port]}"
+    end
+  end
+end

data/dryad-consul/spec/dryad/consul/service_client_spec.rb

@@ -0,0 +1,6 @@
+RSpec.describe Dryad::Consul::ServiceClient do
+  it "can get all services" do
+    services = Dryad::Consul::ServiceClient.get_all()
+    expect(services.consul.count).to be(0)
+  end
+end

data/dryad-consul/spec/dryad/consul/service_registry_spec.rb

@@ -0,0 +1,33 @@
+RSpec.describe Dryad::Consul::ServiceRegistry do
+  before do
+    @portal = Dryad::Core::Portal.new(
+      schema: Dryad::Core::Schema::HTTP,
+      port: 3000,
+      pattern: '/*',
+      non_certifications: ['/*']
+    )
+    @service = Dryad::Consul::Service.new(
+      name: 'rails',
+      address: '127.0.0.1',
+      group: 'staging',
+      portals: [@portal],
+      priority: 10,
+      load_balancing: [Dryad::Core::LoadBalancing::URL_HASH]
+    )
+  end
+
+  it "gets the instances for a service" do
+    Dryad::Consul::ServiceRegistry.register(@service)
+    registers = @service.to_registers
+    service_instances = Dryad::Consul::ServiceRegistry.service_instances(
+      @service.name,
+      Dryad::Core::Schema::HTTP,
+      [@service.group]
+    )
+    expect(service_instances.count).to eq(registers.count)
+    service_instances.zip(registers).each do |instance, register|
+      expect(instance.name).to eq(register[:Name])
+    end
+    Dryad::Consul::ServiceRegistry.deregister(@service)
+  end
+end

data/dryad-consul/spec/dryad/consul/service_spec.rb

@@ -0,0 +1,23 @@
+RSpec.describe Dryad::Consul::Service do
+  before do
+    @portal = Dryad::Core::Portal.new(
+      schema: Dryad::Core::Schema::HTTP,
+      port: 3000,
+      pattern: '/*',
+      non_certifications: ['/*']
+    )
+    @service = Dryad::Consul::Service.new(
+      name: 'http-service',
+      address: '127.0.0.1',
+      group: 'staging',
+      portals: [@portal],
+      priority: 10,
+      load_balancing: [Dryad::Core::LoadBalancing::URL_HASH]
+    )
+  end
+
+  it "gets all registers" do
+    registers = @service.to_registers
+    expect(registers.count).to eq(@service.portals.count)
+  end
+end

data/dryad-consul/spec/dryad/consul_spec.rb

@@ -0,0 +1,10 @@
+RSpec.describe Dryad::Consul do
+  it "has a version number" do
+    version = File.read(File.expand_path("../../../DRYAD_VERSION", __dir__)).strip
+    expect(Dryad::Consul::VERSION).to eq(version)
+  end
+
+  it "has configured dependencies" do
+    expect(::Diplomat.configuration.url).to eq("http://localhost:8500")
+  end
+end

data/dryad-consul/spec/spec_helper.rb

@@ -0,0 +1,14 @@
+require "bundler/setup"
+require "dryad/consul"
+
+RSpec.configure do |config|
+  # Enable flags like --only-failures and --next-failure
+  config.example_status_persistence_file_path = ".rspec_status"
+
+  # Disable RSpec exposing methods globally on `Module` and `main`
+  config.disable_monkey_patching!
+
+  config.expect_with :rspec do |c|
+    c.syntax = :expect
+  end
+end

data/dryad-core/.gitignore

@@ -0,0 +1,11 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+
+# rspec failure tracking
+.rspec_status

data/dryad-core/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/dryad-core/.travis.yml

@@ -0,0 +1,7 @@
+---
+sudo: false
+language: ruby
+cache: bundler
+rvm:
+  - 2.6.3
+before_install: gem install bundler -v 2.0.1

data/dryad-core/Gemfile

@@ -0,0 +1,4 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in dryad-core.gemspec
+gemspec

data/dryad-core/Gemfile.lock

@@ -0,0 +1,35 @@
+PATH
+  remote: .
+  specs:
+    dryad-core (0.1.0.alpha)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    diff-lcs (1.3)
+    rake (10.5.0)
+    rspec (3.8.0)
+      rspec-core (~> 3.8.0)
+      rspec-expectations (~> 3.8.0)
+      rspec-mocks (~> 3.8.0)
+    rspec-core (3.8.0)
+      rspec-support (~> 3.8.0)
+    rspec-expectations (3.8.3)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.8.0)
+    rspec-mocks (3.8.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.8.0)
+    rspec-support (3.8.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 2.0)
+  dryad-core!
+  rake (~> 10.0)
+  rspec (~> 3.0)
+
+BUNDLED WITH
+   2.0.1

data/dryad-core/LICENSE

@@ -0,0 +1,19 @@
+The MIT License (MIT)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/dryad-core/README.md

@@ -0,0 +1,35 @@
+# Dryad::Core
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/dryad/core`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'dryad-core'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install dryad-core
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/dryad-core.

data/dryad-core/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/dryad-core/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "dryad/core"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/dryad-core/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/dryad-core/dryad-core.gemspec

@@ -0,0 +1,27 @@
+
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "dryad/core/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "dryad-core"
+  spec.version       = Dryad::Core::VERSION
+  spec.authors       = ["Pan Jie"]
+  spec.email         = ["panjie@growingio.com"]
+
+  spec.summary       = %q{Dryad core library.}
+  spec.description   = %q{Dryad core includes the basic objects.}
+  spec.homepage      = "https://github.com/jack0pan/dryad.rb"
+  spec.license       = "MIT"
+
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 2.0"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+end

data/dryad-core/lib/dryad/configuration.rb

@@ -0,0 +1,22 @@
+module Dryad
+  class Configuration
+    attr_accessor :consul, :namespace, :group, :registry, :service
+
+    DEFAULT_OPTIONS = {
+      consul: {
+        host: 'localhost',
+        port: 8500
+      },
+      service: {}
+    }
+
+    def initialize(opts = {})
+      opts = DEFAULT_OPTIONS.merge(opts)
+      @consul = opts[:consul]
+      @namespace = opts[:namespace]
+      @group = opts[:group]
+      @registry = opts[:registry]
+      @service = opts[:service]
+    end
+  end
+end

data/dryad-core/lib/dryad/core/load_balancing.rb

@@ -0,0 +1,8 @@
+module Dryad
+  module Core
+    module LoadBalancing
+      URL_HASH = "url_chash"
+      ROUND_ROBIN = "round_robin"
+    end
+  end
+end

data/dryad-core/lib/dryad/core/portal.rb

@@ -0,0 +1,21 @@
+require 'securerandom'
+
+module Dryad
+  module Core
+    class Portal
+      attr_accessor :id, :schema, :port, :pattern, :check, :non_certifications
+
+      DEFAULT_OPTIONS = {
+        :non_certifications => []
+      }
+
+      def initialize(options = DEFAULT_OPTIONS)
+        @id = SecureRandom.uuid
+        @schema = options[:schema]
+        @port = options[:port]
+        @pattern = options[:pattern]
+        @non_certifications = options[:non_certifications]
+      end
+    end
+  end
+end

data/dryad-core/lib/dryad/core/schema.rb

@@ -0,0 +1,9 @@
+module Dryad
+  module Core
+    module Schema
+      HTTP = "http"
+      GRPC = "grpc"
+      WEB_SOCKET = "web-socket"
+    end
+  end
+end

data/dryad-core/lib/dryad/core/service.rb

@@ -0,0 +1,39 @@
+module Dryad
+  module Core
+    class Service
+      attr_accessor :name, :address, :group, :portals, :priority, :load_balancing
+
+      TYPE = "microservice"
+
+      DEFAULT_OPTIONS = {
+        :portals => [],
+        :load_balancing => []
+      }
+
+      def initialize(options = {})
+        options = DEFAULT_OPTIONS.merge(options)
+        @name = options[:name]
+        @address = options[:address]
+        @group = options[:group]
+        @portals = options[:portals]
+        @priority = options[:priority]
+        @load_balancing = options[:load_balancing]
+      end
+
+      def type_name
+        Dryad::Core::Service::TYPE
+      end
+
+      class << self
+        def full_name(schema, name)
+          case schema
+          when Schema::HTTP
+            name
+          else
+            "#{name}-#{schema}"
+          end
+        end
+      end
+    end
+  end
+end

data/dryad-core/lib/dryad/core/service_instance.rb

@@ -0,0 +1,14 @@
+module Dryad
+  module Core
+    class ServiceInstance
+      attr_accessor :name, :schema, :address, :port
+
+      def initialize(options = {})
+        @name = options[:name]
+        @schema = options[:schema]
+        @address = options[:address]
+        @port = options[:port]
+      end
+    end
+  end
+end

data/dryad-core/lib/dryad/core/version.rb

@@ -0,0 +1,10 @@
+module Dryad
+  module Core
+    MAJOR = 0
+    MINOR = 1
+    TINY  = 0
+    PRE   = nil
+  
+    VERSION = [MAJOR, MINOR, TINY, PRE].compact.join(".")
+  end
+end

data/dryad-core/lib/dryad/core.rb

@@ -0,0 +1,12 @@
+require "dryad/core/version"
+require "dryad/core/service_instance"
+require "dryad/core/service"
+require "dryad/core/portal"
+require "dryad/core/load_balancing"
+require "dryad/core/schema"
+
+module Dryad
+  module Core
+    class Error < StandardError; end
+  end
+end

data/dryad-core/lib/dryad.rb

@@ -0,0 +1,46 @@
+require "dryad/configuration"
+require "dryad/core"
+require "yaml"
+require "erb"
+
+DRYAD_COFNIG_FILE = ENV["DRYAD_CONFIG_FILE"] || "config/dryad.yml"
+
+module Dryad
+  class Error < StandardError; end
+
+  class << self
+    attr_accessor :configuration
+  end
+
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
+
+  def self.configure
+    yield(configuration)
+  end
+
+  def self.configure_with_file
+    environment = ENV["RAILS_ENV"] || ENV["RACK_ENV"] || "default"
+    opts = YAML.load(ERB.new(File.read(DRYAD_COFNIG_FILE)).result)[environment] || Configuration::DEFAULT_OPTIONS
+
+    if opts.respond_to? :deep_symbolize_keys!
+      opts.deep_symbolize_keys!
+    else
+      symbolize_keys_deep!(opts)
+    end
+
+    @configuration = Configuration.new(opts)
+  end
+
+  private
+  def self.symbolize_keys_deep!(hash)
+    hash.keys.each do |k|
+      symkey = k.respond_to?(:to_sym) ? k.to_sym : k
+      hash[symkey] = hash.delete k
+      symbolize_keys_deep! hash[symkey] if hash[symkey].is_a? Hash
+    end
+  end
+end
+
+Dryad.configure_with_file if File.exist?(DRYAD_COFNIG_FILE)

data/dryad-core/spec/dryad/core/load_balancing_spec.rb

@@ -0,0 +1,6 @@
+RSpec.describe Dryad::Core::LoadBalancing do
+  it "values are correct" do
+    expect(Dryad::Core::LoadBalancing::URL_HASH).to eq("url_chash")
+    expect(Dryad::Core::LoadBalancing::ROUND_ROBIN).to eq("round_robin")
+  end
+end

data/dryad-core/spec/dryad/core/portal_spec.rb

@@ -0,0 +1,13 @@
+RSpec.describe Dryad::Core::Portal do
+  it "creates with default options" do
+    portal = Dryad::Core::Portal.new
+    expect(portal.id).not_to be(nil)
+    expect(portal.non_certifications).to eq([])
+  end
+
+  it "creates portals with different id" do
+    portals = Array.new(100) { Dryad::Core::Portal.new }
+    ids_set = portals.map{|portal| portal.id }.to_set
+    expect(ids_set.count).to eq(portals.count)
+  end
+end

data/dryad-core/spec/dryad/core/schema_spec.rb

@@ -0,0 +1,7 @@
+RSpec.describe Dryad::Core::Schema do
+  it "has valid values" do
+    expect(Dryad::Core::Schema::HTTP).to eq("http")
+    expect(Dryad::Core::Schema::GRPC).to eq("grpc")
+    expect(Dryad::Core::Schema::WEB_SOCKET).to eq("web-socket")
+  end
+end

data/dryad-core/spec/dryad/core/service_instance_spec.rb

@@ -0,0 +1,7 @@
+RSpec.describe Dryad::Core::ServiceInstance do
+  it 'can be created and access' do
+    si = Dryad::Core::ServiceInstance.new(name: 'grpc-service', schema: 'grpc', address: 'localhost', port: 19000)
+    expect(si.name).to eq('grpc-service')
+    expect(si.schema).to eq('grpc')
+  end
+end

data/dryad-core/spec/dryad/core/service_spec.rb

@@ -0,0 +1,14 @@
+RSpec.describe Dryad::Core::Service do
+  it "create with default options" do
+    service = Dryad::Core::Service.new
+    expect(service.type_name).to eq(Dryad::Core::Service::TYPE)
+    expect(service.portals).to eq([])
+    expect(service.load_balancing).to eq([])
+  end
+
+  it "create with options" do
+    service = Dryad::Core::Service.new(name: "rails", address: "localhost", group: "staging", priority: 10)
+    expect(service.name).to eq("rails")
+    expect(service.address).to eq("localhost")
+  end
+end

data/dryad-core/spec/dryad/core_spec.rb

@@ -0,0 +1,6 @@
+RSpec.describe Dryad::Core do
+  it "has a version number" do
+    version = File.read(File.expand_path("../../../DRYAD_VERSION", __dir__)).strip
+    expect(Dryad::Core::VERSION).to eq(version)
+  end
+end