Merge pull request #1 from ant-in-giant/v004

V004

Author: sogaoh

CHANGELOG.md

@@ -0,0 +1,23 @@
+# Change Log
+
+## v0.0.4
+2020.10.31
+- 複数サイトを１つの Heroku アプリでサービスメトリック採取できるように対応
+
+
+## v0.0.3
+2020.07.27
+- 複数サイトのPVを同一グラフに表示する調整
+    - refs Pubs/Mackerel Ambassador (sogaoh) blogs/
+        - [GoogleAnalyticsで取得したアクティブユーザー数を定期的にMackerelサービスメトリックに投稿する (1)](https://esa-pages.io/p/sharing/6641/posts/976/ebc3295116a84b02c966.html)
+        - [GoogleAnalyticsで取得したアクティブユーザー数を定期的にMackerelサービスメトリックに投稿する (2)](https://esa-pages.io/p/sharing/6641/posts/979/7ba0cdf6e32419970274.html)
+
+## v0.0.2
+2020.07.27
+- 単一サイト毎のサービスメトリック送信
+
+## v0.0.1
+2016.09.18
+- [a-know さんのオリジナル版公開](https://github.com/a-know/currentvisitor2mackerel)
+    - refs [Google Analyticsの「現在の訪問人数」を Mackerel に投稿する heroku app を作ったよ - えいのうにっき](http://blog.a-know.me/entry/2016/09/19/162242)
+

README.md

@@ -4,7 +4,10 @@ Post current visitor number of Google Analytics to [mackerel.io](https://mackere
 ## heroku app deploy button
 [![Deploy](https://www.herokucdn.com/deploy/button.svg)](https://heroku.com/deploy)
 
-## How to use
+
+## References
 In Japanese.
 
-[Google Analyticsの「現在の訪問人数」を Mackerel に投稿する heroku app を作ったよ](http://blog.a-know.me/entry/2016/09/19/162242)
+[GoogleAnalyticsで取得したアクティブユーザー数を定期的にMackerelサービスメトリックに投稿する の関連コード](https://github.com/sogaoh/mackerel-practice/tree/master/google-01)
+[Google Analyticsの「現在の訪問人数」を Mackerel に投稿する heroku app を作ったよ - えいのうにっき](http://blog.a-know.me/entry/2016/09/19/162242)
+

app.json

@@ -6,17 +6,13 @@
       "description": "Google Cloud Platform Service Account Email Address.",
       "value": "[email protected]"
     },
-    "PROFILE_ID": {
-      "description": "Google Analytics Profile ID.",
-      "value": "12345678"
-    },
     "GOOGLE_API_KEY": {
       "description": "Google Cloud Platform Service Account's RSA private key.",
       "value": "-----BEGIN RSA PRIVATE KEY-----\nxxxx\n-----END RSA PRIVATE KEY-----"
     },
-    "WEBSITE_NAME": {
-      "description": "Website name Google Analytics embedded.",
-      "value": "hoge-blog"
+    "SITE_AND_VIEW_ID_JSON": {
+      "description": "Website names (Google Analytics embedded) & Google Analytics View IDs.",
+      "value": "[{\"hoge-blog\":12345678},{\"fuga-blog\":23456789}]"
     },
     "MACKEREL_API_KEY": {
       "description": "Your organization's api-key of mackerel.io.",

app.rb

@@ -2,6 +2,7 @@
 require 'net/http'
 require 'uri'
 require 'google/api_client'
+require 'json'
 
 
 get '/' do
@@ -10,53 +11,86 @@
 get '/post' do
   # Update these to match your own apps credentials
   service_account_email = ENV['SERVICE_ACCOUNT_EMAIL'] # Email of service account
-  profile_id = ENV['PROFILE_ID'] # Analytics profile ID.
 
   # Get the Google API client
   client = Google::APIClient.new(
-    :application_name => 'current visitor of Google Analytics post to mackerel.io',
-    :application_version => '0.0.3'
+      :application_name => 'current visitor of Google Analytics post to mackerel.io',
+      :application_version => '0.0.4'
   )
 
   key = OpenSSL::PKey::RSA.new(ENV['GOOGLE_API_KEY'].gsub("\\n", "\n"))
   client.authorization = Signet::OAuth2::Client.new(
-    :token_credential_uri => 'https://accounts.google.com/o/oauth2/token',
-    :audience             => 'https://accounts.google.com/o/oauth2/token',
-    :scope                => 'https://www.googleapis.com/auth/analytics.readonly',
-    :issuer               => service_account_email,
-    :signing_key          => key,
+      :token_credential_uri => 'https://accounts.google.com/o/oauth2/token',
+      :audience => 'https://accounts.google.com/o/oauth2/token',
+      :scope => 'https://www.googleapis.com/auth/analytics.readonly',
+      :issuer => service_account_email,
+      :signing_key => key,
   )
 
   # Request a token for our service account
   client.authorization.fetch_access_token!
 
   # Get the analytics API
-  analytics = client.discovered_api('analytics','v3')
-
-  # Execute the query, get the value like `[["1"]]`
-  response = client.execute(:api_method => analytics.data.realtime.get, :parameters => {
-    'ids' => "ga:" + profile_id,
-    'metrics' => "ga:activeVisitors",
-  }).data.rows
-
-  number = response.empty? ? 0 : response.first.first.to_i
-  payload = [ {
-                 name: "current_visitors.#{ENV['WEBSITE_NAME']}",
-                 time: Time.now.to_i,
-                 value: number,
-            } ].to_json
-
-  uri = URI.parse("https://mackerel.io/api/v0/services/#{ENV['MACKEREL_SERVICE_NAME']}/tsdb")
-  Net::HTTP.new(uri.host, uri.port).tap do |https|
-    https.use_ssl = true
-    req = Net::HTTP::Post.new(uri.request_uri).tap do |q|
-      q['Content-Type'] = 'application/json'
-      q['X-Api-Key'] = ENV['MACKEREL_API_KEY']
-      q.body = payload
+  analytics = client.discovered_api('analytics', 'v3')
+
+  # Get visitors of plural sites
+  site_and_view_id_json = JSON.parse(ENV['SITE_AND_VIEW_ID_JSON'])
+
+  responses = {}
+  result_status = 0
+  site_and_view_id_json.each do |site_and_view_id|
+    site_and_view_id.each do |site, view_id|
+      #p "site=#{site}、view_id=#{view_id}"
+
+      # Execute the query, get the value like `[["1"]]`
+      response = client.execute(:api_method => analytics.data.realtime.get, :parameters => {
+          'ids' => "ga:" + "#{view_id}",   # Analytics view ID.
+          'metrics' => "ga:activeVisitors",
+      }).data.rows
+
+      number = response.empty? ? 0 : response.first.first.to_i
+      payload = [{
+                     name: "current_visitors.#{site}",
+                     time: Time.now.to_i,
+                     value: number,
+                 }].to_json
+
+      uri = URI.parse("https://mackerel.io/api/v0/services/#{ENV['MACKEREL_SERVICE_NAME']}/tsdb")
+      Net::HTTP.new(uri.host, uri.port).tap do |https|
+        https.use_ssl = true
+        req = Net::HTTP::Post.new(uri.request_uri).tap do |q|
+          q['Content-Type'] = 'application/json'
+          q['X-Api-Key'] = ENV['MACKEREL_API_KEY']
+          q.body = payload
+        end
+        res = https.request(req)
+
+        responses.store("#{site}", :result => {:status => res.code, :body => "#{res.body}"})
+        result_status = [result_status, res.code.to_i].max
+      end
     end
-    res = https.request(req)
-    status res.code
-    headers 'Content-Type' => 'application/json'
-    body "#{res.body}"
   end
+
+  headers 'Content-Type' => 'application/json'
+  status result_status
+  resp = {
+      body: responses,
+  }
+  resp.to_json
+end
+
+get '/sites' do
+  site_and_view_id_json = JSON.parse(ENV['SITE_AND_VIEW_ID_JSON'])
+
+  site_and_view_id_json.each do |site_and_view_id|
+    site_and_view_id.each do |site,view_id|
+      p "site=#{site}, viewId=#{view_id}"
+    end
+  end
+
+  content_type :json
+  response = {
+      body: site_and_view_id_json,
+  }
+  response.to_json
 end