multiwoven-integrations 0.1.68 → 0.1.69

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 77816f54f13495e303382ef4c1e2e932472475d31e0f486d3dcc0535e561f36d
-  data.tar.gz: 854a6871e65a3ad4764548ea14793a04c02d42cde0ed31712c252aff960134ab
+  metadata.gz: 504054c37b59618cb9743ef91bb97069a89346c37ca31f70aca89947542f8261
+  data.tar.gz: 978b116bd88a9a313840f94ea6010bc0e07dcfab31d3f9f7ed6ab3c751a3f159
 SHA512:
-  metadata.gz: 0de8475fa0b27194c1a61b58d56f11fe3a5df8058c6065664a2c67c7af5c5d4c709aab352c0be4c7f53b9f62fda1031a47f1590844755b54897b38a652173306
-  data.tar.gz: dce5ca510e72bccf871075a1a5dd523db56961454b8d4ba987e6037b219f07683a79b1dabc03e51e9b533b18af52ce048ad60684fb2f9ee6e619c9fac272316b
+  metadata.gz: 1add5f808f5ba6774635129869a4c2470f2f7c0f1ec71bf43e09187ed482423d32b41d31f436159240f0ee751b785ac2a9504bbaf83c616eaf0146b4a0974094
+  data.tar.gz: e97ea09b8de2d4170102549a965f5606a3563b8b621b4be8114acdfa5aca05c6899a0d79925faf3578f952057febc16764534a48cf08cd59d81d6f561ec6602c

data/lib/multiwoven/integrations/rollout.rb

@@ -2,7 +2,7 @@
 
 module Multiwoven
   module Integrations
-    VERSION = "0.1.68"
+    VERSION = "0.1.69"
 
     ENABLED_SOURCES = %w[
       Snowflake
@@ -12,6 +12,7 @@ module Multiwoven
       Databricks
       SalesforceConsumerGoodsCloud
       AwsAthena
+      Clickhouse
     ].freeze
 
     ENABLED_DESTINATIONS = %w[

data/lib/multiwoven/integrations/source/clickhouse/client.rb

@@ -0,0 +1,102 @@
+# frozen_string_literal: true
+
+module Multiwoven::Integrations::Source
+  module Clickhouse
+    include Multiwoven::Integrations::Core
+    class Client < SourceConnector
+      def check_connection(connection_config)
+        connection_config = connection_config.with_indifferent_access
+        create_connection(connection_config)
+        ConnectionStatus.new(
+          status: ConnectionStatusType["succeeded"]
+        ).to_multiwoven_message
+      rescue StandardError => e
+        ConnectionStatus.new(
+          status: ConnectionStatusType["failed"], message: e.message
+        ).to_multiwoven_message
+      end
+
+      def discover(connection_config)
+        connection_config = connection_config.with_indifferent_access
+        query = "SELECT table_name, column_name, data_type, is_nullable FROM information_schema.columns WHERE table_schema = '#{connection_config[:database]}' ORDER BY table_name, ordinal_position;"
+        db = create_connection(connection_config)
+        records = query_execution(db, query)
+        catalog = Catalog.new(streams: create_streams(records))
+        catalog.to_multiwoven_message
+      rescue StandardError => e
+        handle_exception(
+          "CLICKHOUSE:DISCOVER:EXCEPTION",
+          "error",
+          e
+        )
+      end
+
+      def read(sync_config)
+        connection_config = sync_config.source.connection_specification
+        connection_config = connection_config.with_indifferent_access
+        query = sync_config.model.query
+        query = batched_query(query, sync_config.limit, sync_config.offset) unless sync_config.limit.nil? && sync_config.offset.nil?
+        db = create_connection(connection_config)
+        query(db, query)
+      rescue StandardError => e
+        handle_exception(
+          "CLICKHOUSE:READ:EXCEPTION",
+          "error",
+          e
+        )
+      end
+
+      private
+
+      def query(connection, query)
+        query_execution(connection, query).map do |row|
+          RecordMessage.new(data: row, emitted_at: Time.now.to_i).to_multiwoven_message
+        end
+      end
+
+      def create_connection(connection_config)
+        @auth_token = Base64.strict_encode64("#{connection_config[:username]}:#{connection_config[:password]}")
+        Faraday.new(connection_config[:host]) do |faraday|
+          faraday.request :url_encoded
+          faraday.adapter Faraday.default_adapter
+        end
+      end
+
+      def query_execution(connection, query)
+        response = connection.post do |req|
+          req.url "/"
+          req.headers["Authorization"] = "Basic #{@auth_token}"
+          req.headers["Content-Type"] = "text/plain"
+          req.body = query
+        end
+        column_names = query[/SELECT (.*?) FROM/i, 1].split(",").map(&:strip)
+        response.body.strip.split("\n").map do |row|
+          columns = row.split("\t")
+          column_names.zip(columns).to_h
+        end
+      end
+
+      def create_streams(records)
+        group_by_table(records).map do |_, r|
+          Multiwoven::Integrations::Protocol::Stream.new(name: r[:tablename], action: StreamAction["fetch"], json_schema: convert_to_json_schema(r[:columns]))
+        end
+      end
+
+      def group_by_table(records)
+        result = {}
+        records.each_with_index do |entry, index|
+          table_name = entry["table_name"]
+          column_data = {
+            column_name: entry["column_name"],
+            data_type: entry["data_type"].gsub(/Nullable\((\w+)\)/, '\1').downcase.gsub!(/\d+/, ""),
+            is_nullable: entry["is_nullable"] == "1"
+          }
+          result[index] ||= {}
+          result[index][:tablename] = table_name
+          result[index][:columns] = [column_data]
+        end
+        result
+      end
+    end
+  end
+end

data/lib/multiwoven/integrations/source/clickhouse/config/meta.json

@@ -0,0 +1,15 @@
+{
+  "data": {
+    "name": "Clickhouse",
+    "title": "ClickHouse",
+    "connector_type": "source",
+    "category": "Data Warehouse",
+    "documentation_url": "https://docs.multiwoven.com/integrations/sources/clickhouse",
+    "github_issue_label": "source-clickhouse",
+    "icon": "icon.svg",
+    "license": "MIT",
+    "release_stage": "alpha",
+    "support_level": "community",
+    "tags": ["language:ruby", "multiwoven"]
+  }
+}

data/lib/multiwoven/integrations/source/clickhouse/config/spec.json

@@ -0,0 +1,42 @@
+{
+  "documentation_url": "https://docs.multiwoven.com/integrations/sources/clickhouse",
+  "stream_type": "dynamic",
+  "connector_query_type": "raw_sql",
+  "connection_specification": {
+    "$schema": "http://json-schema.org/draft-07/schema#",
+    "title": "ClickHouse",
+    "type": "object",
+    "required": ["url", "username", "password", "database"],
+    "properties": {
+      "url": {
+        "description": "The ClickHouse host url to connect.",
+        "examples": ["tu61szglca.us-west-2.aws.clickhouse.cloud"],
+        "type": "string",
+        "title": "Personal URL",
+        "order": 0
+      },
+      "username": {
+        "description": "The username for ClickHouse.",
+        "examples": ["Default"],
+        "type": "string",
+        "title": "Username",
+        "order": 1
+      },
+      "password": {
+        "description": "The password for ClickHouse.",
+        "examples": ["Default"],
+        "type": "string",
+        "multiwoven_secret": true,
+        "title": "Password",
+        "order": 2
+      },
+      "database": {
+        "description": "The ClickHouse database.",
+        "examples": ["default"],
+        "type": "string",
+        "title": "Database",
+        "order": 3
+      }
+    }
+  }
+}

data/lib/multiwoven/integrations/source/clickhouse/icon.svg

@@ -0,0 +1,4 @@
+<svg height="2222" viewBox="0 0 9 8" width="2500" xmlns="http://www.w3.org/2000/svg">
+    <path d="m0 7h1v1h-1z" fill="#f00"/>
+    <path d="m0 0h1v7h-1zm2 0h1v8h-1zm2 0h1v8h-1zm2 0h1v8h-1zm2 3.25h1v1.5h-1z" fill="#fc0"/>
+</svg>

data/lib/multiwoven/integrations.rb

@@ -25,6 +25,8 @@ require "securerandom"
 require "aws-sdk-athena"
 require "zip"
 require "zendesk_api"
+require "faraday"
+require "base64"
 
 # Service
 require_relative "integrations/config"
@@ -51,6 +53,7 @@ require_relative "integrations/source/postgresql/client"
 require_relative "integrations/source/databricks/client"
 require_relative "integrations/source/salesforce_consumer_goods_cloud/client"
 require_relative "integrations/source/aws_athena/client"
+require_relative "integrations/source/clickhouse/client"
 
 # Destination
 require_relative "integrations/destination/klaviyo/client"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: multiwoven-integrations
 version: !ruby/object:Gem::Version
-  version: 0.1.68
+  version: 0.1.69
 platform: ruby
 authors:
 - Subin T P
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-05-21 00:00:00.000000000 Z
+date: 2024-05-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -508,6 +508,10 @@ files:
 - lib/multiwoven/integrations/source/bigquery/config/meta.json
 - lib/multiwoven/integrations/source/bigquery/config/spec.json
 - lib/multiwoven/integrations/source/bigquery/icon.svg
+- lib/multiwoven/integrations/source/clickhouse/client.rb
+- lib/multiwoven/integrations/source/clickhouse/config/meta.json
+- lib/multiwoven/integrations/source/clickhouse/config/spec.json
+- lib/multiwoven/integrations/source/clickhouse/icon.svg
 - lib/multiwoven/integrations/source/databricks/client.rb
 - lib/multiwoven/integrations/source/databricks/config/meta.json
 - lib/multiwoven/integrations/source/databricks/config/spec.json