tops-turn 0.0.1

data/Rakefile

@@ -0,0 +1,32 @@
+# coding: utf-8
+
+require 'rspec/core/rake_task'
+require 'bundler/gem_tasks'
+require 'rubocop/rake_task'
+
+task default: [:rubocop, :spec]
+
+Rubocop::RakeTask.new
+
+task spec: :'spec:unit'
+
+namespace :spec do
+  desc 'run all tests'
+  task all: [:'spec:unit', :'spec:integration']
+
+  desc 'run unit tests'
+  task :unit do |task|
+    RSpec::Core::RakeTask.new(task) do |t|
+      t.pattern = 'spec/unit/**/*_spec.rb'
+      t.rspec_opts = %w(--format documentation --color)
+    end
+  end
+
+  desc 'run integration tests'
+  task :integration do |task|
+    RSpec::Core::RakeTask.new(task) do |t|
+      t.pattern = 'spec/integration/**/*_spec.rb'
+      t.rspec_opts = %w(--format documentation --color)
+    end
+  end
+end

data/lib/turn.rb

@@ -0,0 +1,12 @@
+# coding: utf-8
+
+require 'turn/version'
+require 'turn/term'
+require 'turn/bucket_map'
+
+require 'turn/servers/kv'
+
+module Turn
+  module Server
+  end
+end

data/lib/turn/bucket_map.rb

@@ -0,0 +1,28 @@
+# coding: utf-8
+
+require 'overlord'
+
+module Turn
+  class BucketMap
+    attr_accessor :buckets, :count
+
+    def initialize(term)
+      @term = term
+      @buckets = []
+      @count = 0
+    end
+
+    def load(dc)
+      parse(@term.dump_bucket_map(dc))
+    end
+
+    def parse(map)
+      map.each_with_index do |bucket, index|
+        if /(?<id>\S+):\s+(?<value>\d+)/ =~ bucket
+          @count = value.to_i
+          @buckets[index] = id
+        end
+      end
+    end
+  end
+end

data/lib/turn/servers/kv.rb

@@ -0,0 +1,28 @@
+# coding: utf-8
+
+module Turn::Server
+  class KV
+    def initialize(hostname, port)
+      @hostname = hostname
+      @port = port
+    end
+
+    def disk_usage
+      # ssh
+      # df -h | grep kv-data
+    end
+
+    def cpu_info
+      # ssh
+      # cat /proc/cpuinfo | grep \'model name\' | uniq | cut -d\':\' -f2
+    end
+
+    def pair_count
+      metrics = Turn::Term.new(@hostname, @port).dump_public_metrics('KeyValueStorePopulation')
+      metrics.values.reduce { |a, e| a + e }
+    end
+
+    def agent_info
+    end
+  end
+end

data/lib/turn/term.rb

@@ -0,0 +1,43 @@
+# coding: utf-8
+
+require 'rest_client'
+
+module Turn
+  class Term
+    def initialize(hostname, port)
+      @hostname = hostname
+      @port = port
+    end
+
+    def dump_bucket_map(dc)
+      hash = get('dump profile bucket map', dc: dc)
+
+      fail hash['errors'] unless hash['errors'].empty?
+
+      hash['info']
+    end
+
+    def dump_public_metrics(prefix)
+      metrics = {}
+
+      hash = get('dump public metrics', prefix: prefix)
+      hash['rows'].each do |metric|
+        metrics[metric.first] = metric.last.to_i
+      end
+
+      metrics
+    end
+
+    def get(command, query)
+      q = { command: command, 'response-formatter' => 'json' }
+      uri = URI("http://#{@hostname}:#{@port}/execute")
+      uri.query = URI.encode_www_form(q.merge(query))
+      json = RestClient.get(uri.to_s)
+      hash = JSON.parse(json)
+
+      fail unless hash['errors'].empty? && hash['warnings'].empty?
+
+      hash
+    end
+  end
+end

data/lib/turn/version.rb

@@ -0,0 +1,5 @@
+# coding: utf-8
+
+module Turn
+  VERSION = '0.0.1'
+end

data/spec/integration/term_spec.rb

@@ -0,0 +1,21 @@
+# coding: utf-8
+
+require 'spec_helper'
+
+describe Turn::Term do
+  let(:hostname) { 'prof034.ams1.turn.com' }
+  let(:port) { '4700' }
+  subject { described_class.new(hostname, port) }
+
+  describe '#dump_bucket_map' do
+    it 'should dump bucket map for a dc' do
+      expect(subject.dump_bucket_map('ams1')).to be_a(Array)
+    end
+  end
+
+  describe '#dump_public_metrics' do
+    it 'should dump public metrics' do
+      expect(subject.dump_public_metrics('KeyValueStorePopulation')).to be_a(Hash)
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,21 @@
+# coding: utf-8
+
+require 'turn'
+
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+end

data/spec/unit/bucket_map_spec.rb

@@ -0,0 +1,20 @@
+# coding: utf-8
+
+require 'spec_helper'
+
+describe Turn::BucketMap do
+  subject { described_class.new(term) }
+
+  let(:term) { double(Turn::Term) }
+  let(:bucket_map) { ['pf28-ams1: 25000', 'pf21-ams1: 25000'] }
+
+  it 'should initialize the bucket map from a server list' do
+    expect(term).to receive(:dump_bucket_map).with('dc').and_return(bucket_map)
+
+    subject.load('dc')
+
+    expect(subject.buckets.size).to eq(2)
+    expect(subject.buckets.first).to eq('pf28-ams1')
+    expect(subject.count).to eq(25_000)
+  end
+end

data/spec/unit/servers/kv_spec.rb

@@ -0,0 +1,32 @@
+# coding: utf-8
+
+require 'spec_helper'
+
+describe Turn::Server::KV do
+  let(:hostname) { 'hostname' }
+  let(:port) { 'port' }
+  subject { described_class.new(hostname, port) }
+
+  it 'should fetch disk usage' do
+    pending do
+      expect(subject.disk_usage).to eq(100)
+    end
+  end
+
+  it 'should fetch cpu info' do
+    pending do
+      expect(subject.cpu_info).to eq('foo')
+    end
+  end
+
+  describe '#pair_count' do
+    let(:metrics) { { 'x' => 1, 'y' => 2 } }
+    let(:term) { double(Turn::Term, dump_public_metrics: metrics) }
+
+    it 'should fetch kv pair count' do
+      expect(Turn::Term).to receive(:new).and_return(term)
+
+      expect(subject.pair_count).to eq(3)
+    end
+  end
+end

data/spec/unit/term_spec.rb

@@ -0,0 +1,20 @@
+# coding: utf-8
+
+require 'spec_helper'
+
+describe Turn::Term do
+  let(:hostname) { 'hostname' }
+  let(:port) { 'port' }
+  subject { described_class.new(hostname, port) }
+
+  describe '#dump_bucket_map' do
+    let(:info) { %w[foo] }
+    let(:json) { { 'errors' => [], 'info' => info } }
+
+    it 'should dump bucket map for a dc' do
+      expect(subject).to receive(:get).with('dump profile bucket map', dc: 'dc').and_return(json)
+
+      expect(subject.dump_bucket_map('dc')).to eq(info)
+    end
+  end
+end

data/turn.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'turn/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'tops-turn'
+  spec.version       = Turn::VERSION
+  spec.authors       = ['Martin Englund']
+  spec.email         = %w[menglund@turn.com]
+  spec.description   = %q{Gem to interact with the Turn terminal}
+  spec.summary       = %q{Gem to interact with the Turn terminal}
+  spec.homepage      = 'http://turn.com'
+  spec.license       = 'MIT'
+
+  spec.files         = `git ls-files`.split($RS)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = %w(lib)
+
+  spec.add_dependency 'rest_client'
+  spec.add_dependency 'tops-overlord'
+end

metadata

@@ -0,0 +1,96 @@
+--- !ruby/object:Gem::Specification
+name: tops-turn
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Martin Englund
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-02-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rest_client
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: tops-overlord
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Gem to interact with the Turn terminal
+email:
+- menglund@turn.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Rakefile
+- lib/turn.rb
+- lib/turn/bucket_map.rb
+- lib/turn/servers/kv.rb
+- lib/turn/term.rb
+- lib/turn/version.rb
+- spec/integration/term_spec.rb
+- spec/spec_helper.rb
+- spec/unit/bucket_map_spec.rb
+- spec/unit/servers/kv_spec.rb
+- spec/unit/term_spec.rb
+- turn.gemspec
+homepage: http://turn.com
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Gem to interact with the Turn terminal
+test_files:
+- spec/integration/term_spec.rb
+- spec/spec_helper.rb
+- spec/unit/bucket_map_spec.rb
+- spec/unit/servers/kv_spec.rb
+- spec/unit/term_spec.rb
+has_rdoc: