logstash-input-mongoprofile 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 03371f27b32269c070dd5ca2aefab1146746d318
-  data.tar.gz: f41818a757c80a411428eb7657eb755ddde1760e
+  metadata.gz: f26f651e99da27cbe03bf873ef7bb6c736db4fb2
+  data.tar.gz: bb5705c865fd804733ed18cb815a63af8b39751e
 SHA512:
-  metadata.gz: 36508a1ecbb35a64e4c25e1fd2035aba8eba0d7aac9415f4741bc3a817a9f217c1476f7bcae390665920a07c5fbba1dd400dd1414326e00823ec68137f489fc0
-  data.tar.gz: 0220babfc04743b8f249fd4e89487d085d65bb81d1b595be3d810ad5640098e7da557eef24f838af4aa1c907237c244a25f56ff993983f358ba59e34ade162d6
+  metadata.gz: 64467f27ef79e98ad9f299168c09822727c562c69ea3d3ca5098630266fb7454d4225348e0a74131e078c0063bd1b959d221a4f0a9e73a2e99c979ac194bf6a7
+  data.tar.gz: fed81e42dc1464f287804c922513c5fbd1749dab5dd91bd515c920e7820360a686f7e6af562d06ca42ffebd8950eae5b729c7cd184f1cbddc130f1c87665ea5a

data/lib/logstash/inputs/mongoprofile.rb

@@ -3,7 +3,7 @@ require "logstash/inputs/base"
 require "logstash/namespace"
 require "stud/interval"
 require "socket" # for Socket.gethostname
-require '../../../lib/mongo/mongo'
+require "mongo"
 
 # Generate a repeating message.
 #
@@ -53,3 +53,105 @@ class LogStash::Inputs::Mongoprofile < LogStash::Inputs::Base
     #  * terminate spawned threads
   end
 end # class LogStash::Inputs::Mongoprofile
+
+class MongoAccessor
+  def initialize(url, collection, client_host)
+    connection = Mongo::Client.new(url)
+
+    @mongodb = connection.database
+    @collection = @mongodb.collection(collection)
+    @client_host = client_host
+  end
+
+  def get_documents_by_ts(date, limit)
+    @collection.find({:ts => {:$gt => date}, :client => {:$ne => @client_host}}).limit(limit)
+  end
+
+  def get_documents(limit)
+    @collection.find({:client => {:$ne => @client_host}}).limit(limit)
+  end
+end
+
+class ProfileCollection
+  def initialize(documents, parser)
+    @documents = documents
+    @parser = parser
+  end
+
+  def each
+    @documents.each do |document|
+      document['_id'] = generate_id
+      yield @parser.parse(document)
+    end
+  end
+
+  def get_last_document_date
+    @documents[-1]['ts']
+  end
+
+  private
+  def generate_id
+    # noinspection RubyArgCount
+    BSON::ObjectId.new
+  end
+end
+
+class DocumentParser
+  def initialize(host)
+    @host = host
+  end
+
+  def parse(document)
+    event = LogStash::Event.new('host' => @host)
+
+    document.each do |key, value|
+      event.set(key, value)
+    end
+
+    event
+  end
+end
+
+class LastValueStore
+  def initialize(path, name)
+    @file_full_name = "#{path}/#{name}"
+  end
+
+  def save_last_value(value)
+    file = File.open(@file_full_name, 'a+')
+
+    file.truncate(0)
+    file.puts(value)
+
+    file.close
+  end
+
+  def get_last_value
+    File.read(@file_full_name)
+  end
+end
+
+class Controller
+  def initialize(event, url, collection, limit, path, client_host)
+    @mongo_accessor = MongoAccessor.new(url, collection, client_host)
+    @last_value_store = LastValueStore.new(path, collection)
+    @document_parser = DocumentParser.new(event)
+    @limit = limit
+  end
+
+  def get_next_events
+    last_date_value = @last_value_store.get_last_value
+
+    if last_date_value == ''
+      documents = @mongo_accessor.get_documents(@limit)
+    else
+      documents = @mongo_accessor.get_documents_by_ts(last_date_value, @limit)
+    end
+
+    profile_collection = ProfileCollection.new(documents, @document_parser)
+
+    @last_value_store.save_last_value(profile_collection.get_last_document_date)
+
+    profile_collection
+  end
+end

data/logstash-input-mongoprofile.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name          = 'logstash-input-mongoprofile'
-  s.version       = '0.1.2'
+  s.version       = '0.1.3'
   s.licenses      = ['Apache License (2.0)']
   s.summary       = 'MongoDB system.profile input plugin'
   s.description   = 'MongoDB system.profile input plugin'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: logstash-input-mongoprofile
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Artem Antonov
@@ -93,7 +93,6 @@ files:
 - LICENSE
 - README.md
 - lib/logstash/inputs/mongoprofile.rb
-- lib/mongo/mongo.rb
 - logstash-input-mongoprofile.gemspec
 - spec/inputs/mongoprofile_spec.rb
 homepage: https://github.com/aantonovdevelop/logstash-input-mongoprofile

data/lib/mongo/mongo.rb

@@ -1,117 +0,0 @@
-require 'mongo'
-require 'logstash/inputs/base'
-
-class MongoAccessor
-  def initialize(url, collection, client_host)
-    connection = Mongo::Client.new(url)
-
-    @mongodb = connection.database
-    @collection = @mongodb.collection(collection)
-    @client_host = client_host
-  end
-
-  def get_documents_by_ts(date, limit)
-    @collection.find({:ts => {:$gt => date}, :client => {:$ne => @client_host}}).limit(limit)
-  end
-
-  def get_documents(limit)
-    @collection.find({:client => {:$ne => @client_host}}).limit(limit)
-  end
-end
-
-class ProfileCollection
-  def initialize(documents, parser)
-    @documents = documents
-    @parser = parser
-  end
-
-  def each
-    @documents.each do |document|
-      document['_id'] = generate_id
-      yield @parser.parse(document)
-    end
-  end
-
-  def get_last_document_date
-    @documents[-1]['ts']
-  end
-
-  private
-  def generate_id
-    # noinspection RubyArgCount
-    BSON::ObjectId.new
-  end
-end
-
-class DocumentParser
-  def initialize(host)
-    @host = host
-  end
-
-  def parse(document)
-    event = LogStash::Event.new('host' => @host)
-
-    document.each do |key, value|
-      event.set(key, value)
-    end
-
-    event
-  end
-end
-
-class LastValueStore
-  def initialize(path, name)
-    @file_full_name = "#{path}/#{name}"
-  end
-
-  def save_last_value(value)
-    file = File.open(@file_full_name, 'a+')
-
-    file.truncate(0)
-    file.puts(value)
-
-    file.close
-  end
-
-  def get_last_value
-    File.read(@file_full_name)
-  end
-end
-
-class Controller
-  def initialize(event, url, collection, limit, path, client_host)
-    @mongo_accessor = MongoAccessor.new(url, collection, client_host)
-    @last_value_store = LastValueStore.new(path, collection)
-    @document_parser = DocumentParser.new(event)
-    @limit = limit
-  end
-
-  def get_next_events
-    last_date_value = @last_value_store.get_last_value
-
-    if last_date_value == ''
-      documents = @mongo_accessor.get_documents(@limit)
-    else
-      documents = @mongo_accessor.get_documents_by_ts(last_date_value, @limit)
-    end
-
-    profile_collection = ProfileCollection.new(documents, @document_parser)
-
-    @last_value_store.save_last_value(profile_collection.get_last_document_date)
-
-    profile_collection
-  end
-end
-
-lv = LastValueStore.new('/home/artem', 'mex.txt')
-
-lv.save_last_value("blasaldkfj")
-puts lv.get_last_value
-
-#mongo = MongoAccessor.new('mongodb://192.168.1.37/eleet-v2-dev', 'system.profile')
-
-#ProfileCollection.new(mongo.get_documents(10)).each do |document|
-  #puts document['_id']
-#end
-
-