embulk-input-google_analytics 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 45cf305036e0b44cbd5f8dc0327280406b9f541b
-  data.tar.gz: 357797127e74378167f8ef27cf0e10d2a5c4223e
+  metadata.gz: f199cc26d3fdc13973dcc95596b26f900b579ddf
+  data.tar.gz: 740cf1a32904b1644eb7049c927ce511df950dc0
 SHA512:
-  metadata.gz: b5417ea74d3e6a700a0f42edc4558ca7a4fe5e56024aa79f9c49c8122c96acbb7519697805fed62d8fde13989a176dd00dc35bee3337fd9549b3b746bf42ae30
-  data.tar.gz: 6fec758e1ee78ca9e4055060bca8caef42ac3be47d97ef0623d7790135c40bf945627618ac92507a16da029a4760043db7cdec41a91507ec390a9dbec7ddf8a9
+  metadata.gz: fc38e8e8e8521ed7aa92a9bf2e641662b841277ac3629f2f8c1dabbd2c87040b1b7fee5a9fd5c87333e06d4f19a548c7328e7d9d45670bccd0def96e9ed00ed6
+  data.tar.gz: 946942eb013dcdeb528711b5e1919e20847ba713f9a835840de03d8dbe2fd8a611874bec7b171a001aa4ee51cb2571853ac6a665c818aea3e6486706af708096

data/CHANGELOG.md

@@ -1,3 +1,6 @@
+## 0.1.5 - 2016-08-23
+* Enable custom dimensions [#10](https://github.com/treasure-data/embulk-input-google_analytics/pull/10)
+
 ## 0.1.4 - 2016-08-19
 * Always add `view_id` into records [#9](https://github.com/treasure-data/embulk-input-google_analytics/pull/9)
 

data/embulk-input-google_analytics.gemspec

@@ -1,7 +1,7 @@
 
 Gem::Specification.new do |spec|
   spec.name          = "embulk-input-google_analytics"
-  spec.version       = "0.1.4"
+  spec.version       = "0.1.5"
   spec.authors       = ["uu59"]
   spec.summary       = "Google Analytics input plugin for Embulk"
   spec.description   = "Loads records from Google Analytics."

data/lib/embulk/input/google_analytics/client.rb

@@ -122,6 +122,10 @@ module Embulk
         end
 
         def get_columns_list
+          get_custom_dimensions + get_metadata_columns
+        end
+
+        def get_metadata_columns
           # https://developers.google.com/analytics/devguides/reporting/metadata/v3/reference/metadata/columns/list
           service = Google::Apis::AnalyticsV3::AnalyticsService.new
           service.authorization = auth
@@ -130,6 +134,15 @@ module Embulk
           end
         end
 
+        def get_custom_dimensions
+          # https://developers.google.com/analytics/devguides/config/mgmt/v3/mgmtReference/management/customDimensions/list
+          service = Google::Apis::AnalyticsV3::AnalyticsService.new
+          service.authorization = auth
+          retryer.with_retry do
+            service.list_custom_dimensions(get_profile[:account_id], get_profile[:web_property_id]).to_h[:items]
+          end
+        end
+
         def build_report_request(page_token = nil)
           query = {
             view_id: view_id,

data/lib/embulk/input/google_analytics/plugin.rb

@@ -17,16 +17,22 @@ module Embulk
             col_info = columns_list.find{|col| col[:id] == col_name}
             raise ConfigError.new("Unknown metric/dimension '#{col_name}'") unless col_info
 
-            col_type = 
-              case col_info[:attributes][:dataType]
-              when "STRING"
+            col_type =
+              if col_info[:attributes]
+                # standard dimension
+                case col_info[:attributes][:dataType]
+                when "STRING"
+                  :string
+                when "INTEGER"
+                  :long
+                when "PERCENT", "FLOAT", "CURRENCY"
+                  :double
+                when "TIME"
+                  :timestamp
+                end
+              else
+                # custom dimension
                 :string
-              when "INTEGER"
-                :long
-              when "PERCENT", "FLOAT", "CURRENCY"
-                :double
-              when "TIME"
-                :timestamp
               end
 
             # time_series column should be timestamp

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: embulk-input-google_analytics
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - uu59
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2016-08-19 00:00:00.000000000 Z
+date: 2016-08-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   requirement: !ruby/object:Gem::Requirement