traffiction 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a3843f7b8bc6cba2375f23c92d91209f35feb945
+  data.tar.gz: 733387c16f219ef1e682db312705f251e9324d4c
+SHA512:
+  metadata.gz: 3759b22c3da1359b8f504386bd52f450e1fd2b4b2a6f1db0ea12067df72e51f586101772cf38f3233838a8bd470b475762dd501860c6e3565b7443c0474e3c56
+  data.tar.gz: 81bb59b893370b2b196f058992044ac21237ae9f79096251823f44d15e56a40ca19548898851fe74e13a32929e41a9b98f49249992c4ce924b077ff51b064929

data/.gitignore

@@ -0,0 +1,15 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log
+*.gem

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Forrest Fleming
+
+MIT License
+
+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/README.md

@@ -0,0 +1,36 @@
+# Traffiction
+
+This command-line tool simply generates traffic for a specific URL.  It was
+created for Oversee.net's QA team to simulate traffic in QA and staging
+environments in order to test our concurrent-user-tracking functionality.
+
+This is a simple wrapper - it's not intended for programmatic use, although
+nothing's stopping anyone from using Traffiction as a regular old Ruby module.
+
+## Installation
+
+```
+    $ gem install traffiction
+```
+
+## Usage
+
+```
+% traffiction
+    -v, --verbose                    Verbose output
+    -p, --port PORT                  Port
+    -c, --count COUNT                Number of visits to perform
+    -t, --threads THREADS            Number of threads to use
+    -h, --help                       Display this screen
+```
+
+## Contributing
+
+1. Fork it ( https://github.com/oversee-ffleming/traffiction/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request
+
+## Todo
+* Tests

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+

data/bin/traffiction

@@ -0,0 +1,32 @@
+#!/usr/bin/env ruby
+require 'traffiction'
+require 'optparse'
+def opts_from_cli
+  options = {}
+  opt_parser = OptionParser.new do |opts|
+    opts.banner = "#{$0} [options] url"
+    opts.on('-v', '--verbose', 'Verbose output') { options[:verbose] = true }
+    opts.on('-p PORT', '--port PORT', 'Port') { |port| options[:port] = port }
+    opts.on('-c COUNT', '--count COUNT', 'Number of visits to perform') { |count| options[:count] = count.to_i }
+    opts.on('-t THREADS', '--threads THREADS', 'Number of threads to use') { |threads| options[:threads] = threads.to_i }
+    opts.on('-h', '--help', 'Display this screen') { puts opts ; exit }
+  end
+  opt_parser.parse!
+
+  options[:count] = 5 if options[:count].to_i == 0
+  options[:threads] = 4 if options[:threads].to_i == 0
+  options[:url] = ARGV.pop
+  abort opt_parser.to_s if options[:url].nil?
+
+  options[:url] = options[:url].sub('http://', '')
+  if options[:url] =~ /^(.*?):(\d+)$/
+    options[:url] = $1
+    options[:port] = $2
+  else
+    options[:port] = 80
+  end
+  puts options.inspect if options[:verbose]
+  options
+end
+
+Traffiction.generate(opts_from_cli)

data/lib/traffiction.rb

@@ -0,0 +1,60 @@
+require "traffiction/version"
+require 'ruby-progressbar'
+require 'httparty'
+module Traffiction
+  class << self
+    # Generates threads, runs the sweep, and returns
+    # @param [Hash] Parameters hash of the following format:
+    #     threads: Fixnum, number of threads to create
+    #     count:   Fixnum, number of times to visit the site
+    #     url:     String, url to visit
+    #     port:    Fixnum, port
+    #     verbose: Boolean, whether to print verbose output
+    # @return nil
+    def generate(options={})
+      @options = options
+      puts "#{visits_array.count} threads will be created with visit distribution #{visits_array}" if options[:verbose]
+      visits_array.each do |times|
+        threads.push Thread.new { visit("#{options[:url]}:#{options[:port]}", times) }
+      end
+      begin
+        threads.each { |t| t.join }
+      rescue => e
+        abort "#{e}"
+      end
+    end
+
+    private
+    attr_reader :options
+
+    def progress_bar
+      @progress_bar ||= ProgressBar.create(
+        title: "Visits",
+        total: options[:count],
+        format: '|%B| %E',
+        width: 20,
+        progress_mark: '='
+      )
+    end
+
+    def threads
+      @threads ||= []
+    end
+
+    def visit(url_string, times=1)
+      (1..times).map do
+        progress_bar.increment
+        HTTParty.get("http://#{url_string}")
+      end
+    end
+
+    def visits_array
+      return @visits_array unless @visits_array.nil?
+      ret = Array.new(options[:threads], 0)
+      (1..options[:count]).map do |i|
+        ret[i % options[:threads]] += 1
+      end
+      @visits_array = ret.select { |el| el > 0 }
+    end
+  end
+end

data/lib/traffiction/version.rb

@@ -0,0 +1,3 @@
+module Traffiction
+  VERSION = "0.1.0"
+end

data/traffiction.gemspec

@@ -0,0 +1,25 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'traffiction/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "traffiction"
+  spec.version       = Traffiction::VERSION
+  spec.authors       = ['Forrest Fleming']
+  spec.email         = ['ffleming@oversee.net']
+  spec.summary       = %q(Hits a url a specified number of times.  It's like a curl loop, but faster.)
+  spec.description   = %q(Performs an HTTParty::get request in a reasonably threaded manner.) <<
+                       %q(Primarily useful for manipulating concurrent users displays on staging environments)
+  spec.homepage      = 'https://github.com/oversee-ffleming/traffiction'
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.7"
+  spec.add_runtime_dependency 'ruby-progressbar', '~> 1.7'
+  spec.add_runtime_dependency 'httparty', '~> 0.13'
+end

metadata

@@ -0,0 +1,97 @@
+--- !ruby/object:Gem::Specification
+name: traffiction
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Forrest Fleming
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-04-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: ruby-progressbar
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.13'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.13'
+description: Performs an HTTParty::get request in a reasonably threaded manner.Primarily
+  useful for manipulating concurrent users displays on staging environments
+email:
+- ffleming@oversee.net
+executables:
+- traffiction
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/traffiction
+- lib/traffiction.rb
+- lib/traffiction/version.rb
+- traffiction.gemspec
+homepage: https://github.com/oversee-ffleming/traffiction
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Hits a url a specified number of times.  It's like a curl loop, but faster.
+test_files: []