#55 Add Plugin Subscription::StatDb

id774 · id774 · commit f70873117b5b · 2013-07-12T11:39:27.000+09:00
diff --git a/plugins/subscription/stat_db.rb b/plugins/subscription/stat_db.rb
@@ -0,0 +1,66 @@
+# -*- coding: utf-8 -*-
+# Name::      Automatic::Plugin::Subscription::StatDb
+# Author::    774 <http://id774.net>
+# Created::   Jul 12, 2013
+# Updated::   Jul 12, 2013
+# Copyright:: Copyright (c) 2012-2013 Automatic Ruby Developers.
+# License::   Licensed under the GNU GENERAL PUBLIC LICENSE, Version 3.0.
+
+module Automatic::Plugin
+  class SubscriptionStatDb
+    require 'open-uri'
+    require 'active_support'
+    require 'active_support/core_ext'
+    require 'rss'
+
+    def initialize(config, pipeline=[])
+      @config = config
+      @pipeline = pipeline
+    end
+
+    def run
+      @return_feeds = []
+      @config['params'].each {|param|
+        url = "http://statdb.nstac.go.jp/api/1.0b/app/getStatsData?appId=#{@config['appid']}&#{param}"
+        retries = 0
+        begin
+          create_rss(URI.encode(url))
+        rescue
+          retries += 1
+          Automatic::Log.puts("error", "ErrorCount: #{retries}, Fault in parsing: #{url}")
+          sleep @config['interval'].to_i unless @config['interval'].nil?
+          retry if retries <= @config['retry'].to_i unless @config['retry'].nil?
+        end
+      }
+      @return_feeds
+    end
+
+    private
+    def create_rss(url)
+      Automatic::Log.puts("info", "Parsing: #{url}")
+      hash = Hash.from_xml open(url).read
+      json = hash.to_json
+      data = ActiveSupport::JSON.decode(json)
+      unless data.nil?
+        rss = create(data, url)
+        sleep @config['interval'].to_i unless @config['interval'].nil?
+        @return_feeds << rss
+      end
+    end
+
+    def create(data, url)
+      RSS::Maker.make("2.0") {|maker|
+        xss = maker.xml_stylesheets.new_xml_stylesheet
+        maker.channel.title = "Automatic Ruby"
+        maker.channel.description = "Automatic Ruby"
+        maker.channel.link = "https://github.com/automaticruby/automaticruby"
+        maker.items.do_sort = true
+        item = maker.items.new_item
+        item.title = "StatDB Feed by Automatic Ruby"
+        item.link = url
+        item.content_encoded = data
+        item.date = Time.now
+      }
+    end
+  end
+end
diff --git a/test/integration/test_statdb.yml b/test/integration/test_statdb.yml
@@ -0,0 +1,15 @@
+global:
+  timezone: Asia/Tokyo
+  cache:
+    base: /tmp
+  log:
+    level: info
+
+plugins:
+  - module: SubscriptionStatDb
+    config:
+      appid: your_app_id
+      params:
+        - cdArea=09003,22004&cdCat01=010920070&cdTimeFrom=2012000101&cdTimeTo=2013000303
+
+  - module: PublishConsole
