Class: GoodData::CloudResources::SnowflakeClient
- Inherits:
-
CloudResourceClient
- Object
- CloudResourceClient
- GoodData::CloudResources::SnowflakeClient
- Defined in:
- lib/gooddata/cloud_resources/snowflake/snowflake_client.rb
Constant Summary collapse
- SNOWFLAKE_GDC_APPLICATION_PARAMETER =
'application=GoodData_Platform'
- SNOWFLAKE_SEPARATOR_PARAM =
'?'
Class Method Summary collapse
Instance Method Summary collapse
- #build_url(url) ⇒ Object
- #connect ⇒ Object
-
#initialize(options = {}) ⇒ SnowflakeClient
constructor
A new instance of SnowflakeClient.
- #realize_query(query, _params) ⇒ Object
Methods inherited from CloudResourceClient
Constructor Details
#initialize(options = {}) ⇒ SnowflakeClient
Returns a new instance of SnowflakeClient.
30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
# File 'lib/gooddata/cloud_resources/snowflake/snowflake_client.rb', line 30 def initialize( = {}) raise("Data Source needs a client to Snowflake to be able to query the storage but 'snowflake_client' is empty.") unless ['snowflake_client'] if ['snowflake_client']['connection'].is_a?(Hash) @database = ['snowflake_client']['connection']['database'] @schema = ['snowflake_client']['connection']['schema'] || 'public' @warehouse = ['snowflake_client']['connection']['warehouse'] @url = build_url(['snowflake_client']['connection']['url']) @authentication = ['snowflake_client']['connection']['authentication'] else raise('Missing connection info for Snowflake client') end Java.net.snowflake.client.jdbc.SnowflakeDriver end |
Class Method Details
.accept?(type) ⇒ Boolean
25 26 27 |
# File 'lib/gooddata/cloud_resources/snowflake/snowflake_client.rb', line 25 def accept?(type) type == 'snowflake' end |
Instance Method Details
#build_url(url) ⇒ Object
86 87 88 89 90 91 92 93 94 95 96 97 98 |
# File 'lib/gooddata/cloud_resources/snowflake/snowflake_client.rb', line 86 def build_url(url) is_contain = url.include?(SNOWFLAKE_GDC_APPLICATION_PARAMETER) unless is_contain if url.include?(SNOWFLAKE_SEPARATOR_PARAM) url.concat("&") else url.concat(SNOWFLAKE_SEPARATOR_PARAM) end url.concat(SNOWFLAKE_GDC_APPLICATION_PARAMETER) end url end |
#connect ⇒ Object
73 74 75 76 77 78 79 80 81 82 83 84 |
# File 'lib/gooddata/cloud_resources/snowflake/snowflake_client.rb', line 73 def connect GoodData.logger.info "Setting up connection to Snowflake #{@url}" prop = java.util.Properties.new prop.setProperty('user', @authentication['basic']['userName']) prop.setProperty('password', @authentication['basic']['password']) prop.setProperty('schema', @schema) prop.setProperty('warehouse', @warehouse) prop.setProperty('db', @database) @connection = java.sql.DriverManager.getConnection(@url, prop) end |
#realize_query(query, _params) ⇒ Object
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
# File 'lib/gooddata/cloud_resources/snowflake/snowflake_client.rb', line 47 def realize_query(query, _params) GoodData.gd_logger.info("Realize SQL query: type=snowflake status=started") connect filename = "#{SecureRandom.urlsafe_base64(6)}_#{Time.now.to_i}.csv" measure = Benchmark.measure do statement = @connection.create_statement has_result = statement.execute(query) if has_result result = statement.get_result_set = result. col_count = .column_count CSV.open(filename, 'wb') do |csv| csv << Array(1..col_count).map { |i| .get_column_name(i) } # build the header csv << Array(1..col_count).map { |i| result.get_string(i)&.to_s } while result.next end end end GoodData.gd_logger.info("Realize SQL query: type=snowflake status=finished duration=#{measure.real}") filename ensure @connection&.close @connection = nil end |