ruby 示例脚本em ruby

Posted 2021-05-15

require 'faraday'
require 'parallel'

class Params
  def initialize args
    @args = args
  end

  def opt? key
    @args.each { |arg| return true if arg.include?("-#{key}") }
    false
  end

  def include? key
    @args.each { |arg| return true if arg.include?("--#{key}") }
    false
  end

  def get key
    @args.each { |arg| return arg.split('=').last if arg.include?("--#{key}") }
    nil
  end

  def get_s key
    get(key)
  end

  def get_a key, separator = ','
    get(key).to_s.split(separator)
  end

  def get_i key
    get(key).to_i
  end
 
  def get_f key
    get(key).to_f
  end

  def get_sym key
    value = get(key)
    value.to_sym if value
  end
end 

params = Params.new(ARGV)

class Skyhub
  attr_accessor :email, :token

  def initialize email, token
    @email = email
    @token = token
  end

  def connection
    @connection ||= get_connection
  end

  def get_connection
    Faraday.new('https://in.skyhub.com.br') do |faraday|
      faraday.request  :url_encoded             # form-encode POST params
      #faraday.response :logger                  # log requests to STDOUT
      faraday.adapter  Faraday.default_adapter  # make requests with Net::HTTP
      faraday.headers['X-User-Email'] = @email  #'[e-mail]'
      faraday.headers['X-User-Token'] = @token #'9zqEvdqRjRQVJU-JjjPL'
      faraday.headers['Content-Type'] = 'application/json'
      faraday.headers['Accept'] = 'application/json'
    end
  end
end

class Product
  def initialize skyhub
    @skyhub = skyhub
  end

  def delete code
    @skyhub.connection.delete("/products/#{code}")
  end
end

product = Product.new(Skyhub.new(params.get_s(:email), params.get_s(:token)))

codes = []

sleep_time = [1,2,3,4,5]

Parallel.each(codes, in_threads: 20) do |code|
  response = product.delete(code)
  sleep(sleep_time[(rand*5).to_i])
  puts "sku: #{code};status: #{response.status};body: #{response.body}"
end