rockset 0.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 4964022460f62f6c4382f0109975dc0dbe6826cd
+  data.tar.gz: e8f3c63413429207d44ea4ff861c590d207c9acd
+SHA512:
+  metadata.gz: 86852877c71e767bdc2187ac2565f4f43776c9878675bf374c8af08e8fd3393c0ab726782be85f82c2de7e6a9b69518b325a5dc7fac0b3c85371a1fae6c64b94
+  data.tar.gz: 52062e28ffc1b9b0d5acf52ae277a02dda807f8fe58622547ca59b09ee958dab63143675f2ae4983938d0f8ee988ac80f444de7c3b39ed99cb36325da4673c88

data/lib/rockset.rb

@@ -0,0 +1,199 @@
+module Rockset
+    require "json"
+    require "uri"
+    require "net/http"
+
+    class InvalidQuery < StandardError
+    end
+
+    def Rockset.auth(api_key, server)
+        $key = api_key
+        $server = "https://#{server}/v1/orgs/self"
+    end
+
+    def Rockset.query(query)
+        uri = URI("#{$server}/queries")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Post.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            req.body = "{\"sql\": {\"query\": \"#{query}\"}}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.validate_query(query)
+        uri = URI("#{$server}/queries/validations")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Post.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            req.body = "{\"sql\": {\"query\": \"#{query}\"}}"
+            http.request(req)
+        end
+        if res.code == "200"
+            return("Valid")
+        else
+            raise(InvalidQuery.new(res.body))
+        end
+    end
+
+    def Rockset.add_docs(docs, collection, workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/collections/#{collection}/docs")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Post.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            req.body = "{\"data\": #{docs.to_json}}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.del_docs(docs, collection, workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/collections/#{collection}/docs")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Delete.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            req.body = "{\"data\": #{docs.to_json}}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.patch_docs(docs, collection, workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/collections/#{collection}/docs")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Patch.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            req.body = "{\"data\": #{docs.to_json}}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.create_collection(collection_metadata, workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/collections")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Post.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            req.body = "#{collection_metadata.to_json}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.del_collection(collection, workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/collections/#{collection}")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Delete.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.get_collection(collection, workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/collections/#{collection}")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Get.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.get_collection_qlambdas(collection, workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/collections/#{collection}/lambdas")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Get.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.get_collections
+        uri = URI("#{$server}/collections")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Get.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.get_workspace_collections(workspace="commons")
+        uri = URI("#{server}/ws/#{workspace}/collections")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Get.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.get_org
+        uri = URI("#{$server}")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Get.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.add_qlambda(name, query, description="", default_params=[], workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/lambdas")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Post.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            req.body = "{
+                \"name\": \"#{name}\",
+                \"description\": \"#{description}\",
+                \"sql\": {
+                    \"query\": \"#{query}\",
+                    \"default_parameters\": #{default_params.to_json}
+                }
+            }"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.del_qlambda(qlambda, workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/lambdas/#{qlambda}")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Delete.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+
+    def Rockset.exec_qlambda(qlambda, version, parameters=[], workspace="commons")
+        uri = URI("#{$server}/ws/#{workspace}/lambdas/#{qlambda}/versions/#{version}")
+        res = Net::HTTP.start(uri.host, uri.port, use_ssl: true) do |http|
+            req = Net::HTTP::Post.new(uri)
+            req["Content-Type"] = "application/json"
+            req["Authorization"] = "ApiKey #{$key}"
+            req.body = "{
+                \"parameters\": #{parameters.to_json}
+            }"
+            http.request(req)
+        end
+        return(JSON.parse(res.body))
+    end
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: rockset
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- Aarav Borthakur
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-12-06 00:00:00.000000000 Z
+dependencies: []
+description: 
+email: 
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/rockset.rb
+homepage: 
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2.3
+signing_key: 
+specification_version: 4
+summary: Rockset Ruby Client
+test_files: []