SQL in ruby (#91)

README.md

@@ -31,18 +31,18 @@ You will get output like so:
 ```bash
 $ invoke test python any
 
-docker run ...
-  ⏱  ./src/python/sort_builtin.py on ./data/sort_input_0.txt ran for 0.5 seconds
+docker run ... python ./src/python/sort_builtin.py
+  ⏱  ./src/python/sort_builtin.py on ./data/sort_input_0.txt ran for 0.52 seconds
   🟢 ./src/python/sort_builtin.py on ./data/sort_input_0.txt succeeded
-docker run ...
-  ⏱  ./src/python/sort_bubble_sort.py on ./data/sort_input_0.txt ran for 0.51 seconds
-  🟢 ./src/python/sort_bubble_sort.py on ./data/sort_input_0.txt succeeded
-docker run ...
-  ⏱  ./src/python/sort_selection_sort.py on ./data/sort_input_0.txt ran for 0.58 seconds
-  🟢 ./src/python/sort_selection_sort.py on ./data/sort_input_0.txt succeeded
-docker run ...
-  ⏱  ./src/python/sort_insertion_sort.py on ./data/sort_input_0.txt ran for 0.52 seconds
-  🟢 ./src/python/sort_insertion_sort.py on ./data/sort_input_0.txt succeeded
+docker run ... python ./src/python/sql_test.py
+  ⏱  ./src/python/sql_test.py on ./data/sql_input_2.sql ran for 0.75 seconds
+  🟢 ./src/python/sql_test.py on ./data/sql_input_2.sql succeeded
+docker run ... python ./src/python/sql_test.py
+  ⏱  ./src/python/sql_test.py on ./data/sql_input_3.sql ran for 0.75 seconds
+  🟢 ./src/python/sql_test.py on ./data/sql_input_3.sql succeeded
+docker run ... python ./src/python/sql_test.py
+  ⏱  ./src/python/sql_test.py on ./data/sql_input_1.sql ran for 0.74 seconds
+  🟢 ./src/python/sql_test.py on ./data/sql_input_1.sql succeeded
 
 ✨ script run success ✨
 ```

snippets/ruby/sql_test.rb

@@ -0,0 +1,144 @@
+class SQLState
+  attr_reader :state
+
+  def initialize(state = {})
+    @state = state
+  end
+
+  def read_table_meta(table_name)
+    @state[table_name] ? @state[table_name]['metadata'] || {} : {}
+  end
+
+  def read_table_rows(table_name)
+    @state[table_name] ? @state[table_name]['rows'] || [] : []
+  end
+
+  def read_information_schema
+    @state.values.map { |data| data['metadata'] }
+  end
+
+  def write_table_meta(table_name, data)
+    @state[table_name] ||= {}
+    @state[table_name]['metadata'] ||= {}
+    @state[table_name]['metadata'].merge!(data)
+    self.class.new(@state)
+  end
+
+  def write_table_rows(table_name, data)
+    @state[table_name] ||= {}
+    @state[table_name]['rows'] ||= []
+    @state[table_name]['rows'] << data
+    self.class.new(@state)
+  end
+end
+
+module SQLType
+  def self.varchar(data)
+    data.to_s.strip.sub(/^['"]/, '').sub(/['"]$/, '')
+  end
+
+  def self.int(data)
+    data.to_i
+  end
+end
+
+module SQLFunctions
+  def self.create_table(state, *args, table_schema: 'public')
+    output = []
+    table_name = args[2]
+
+    columns = {}
+    columns_str = args[3..].join(' ').gsub(/[()]/, '').strip
+    unless columns_str.empty?
+      columns = Hash[columns_str.split(',').map { |column| column.strip.split(' ') }]
+    end
+
+    if state.read_table_meta(table_name).empty?
+      metadata = {
+        'table_name' => table_name,
+        'table_schema' => table_schema,
+        'columns' => columns
+      }
+      state = state.write_table_meta(table_name, metadata)
+    end
+
+    [output, state]
+  end
+
+  def self.insert_into(state, *args)
+    output = []
+    table_name = args[2]
+
+    sql_type_map = {
+      'VARCHAR' => SQLType.method(:varchar),
+      'INT' => SQLType.method(:int)
+    }
+
+    values_index = args.index('VALUES')
+    raise 'VALUES not found' if values_index.nil?
+
+    keys = args[3...values_index].join(' ').gsub(/[()]/, '').split(',').map(&:strip)
+    values = args[values_index + 1..].join(' ').gsub(/[()]/, '').split(',').map(&:strip)
+    key_value_map = Hash[keys.zip(values)]
+
+    metadata = state.read_table_meta(table_name)
+    row = {}
+    key_value_map.each do |key, value|
+      row[key] = sql_type_map[metadata['columns'][key]].call(value)
+    end
+    state = state.write_table_rows(table_name, row)
+
+    [output, state]
+  end
+
+  def self.select(state, *args)
+    output = []
+
+    from_index = args.index('FROM')
+    # where_index = args.index('WHERE')
+
+    raise 'FROM not found' if from_index.nil?
+
+    select_keys = args[1...from_index].join(' ').split(',').map(&:strip)
+    from_value = args[from_index + 1]
+
+    data = if from_value == 'information_schema.tables'
+             state.read_information_schema
+           else
+             state.read_table_rows(from_value)
+           end
+
+    data.each do |datum|
+      output << select_keys.map { |key| [key, datum[key]] }.to_h
+    end
+
+    [output, state]
+  end
+end
+
+def run_sql(input_sql)
+  output = []
+  state = SQLState.new
+
+  input_sql = input_sql.map(&:strip).reject { |line| line.start_with?('--') }
+  input_sql = input_sql.join(' ').split(';')
+
+  sql_function_map = {
+    'CREATE TABLE' => SQLFunctions.method(:create_table),
+    'SELECT' => SQLFunctions.method(:select),
+    'INSERT INTO' => SQLFunctions.method(:insert_into)
+  }
+
+  input_sql.each do |line|
+    words = line.split(' ')
+    words.each_index do |i|
+      key = words[0..i].join(' ').strip
+      if sql_function_map.key?(key)
+        output, state = sql_function_map[key].call(state, *words.reject(&:empty?))
+        break
+      end
+    end
+  end
+
+  [output.to_json]
+end

src/ruby/sort_bubble_sort.rb

@@ -44,12 +44,12 @@ def do_sorting_round(input_list)
 # business logic end #
 ######################
 
-# 👇🏽 copy pasted helpers
+if __FILE__ == $PROGRAM_NAME
+  input_file_path = ENV['INPUT_PATH']
+  input_file = File.readlines(input_file_path)
 
-input_file_path = ENV['INPUT_PATH']
-input_file = File.readlines(input_file_path)
+  sorted_data = do_sorting(input_file)
 
-sorted_data = do_sorting(input_file)
-
-output_file_path = ENV['OUTPUT_PATH']
-File.open(output_file_path, 'wb') { |f| f.write(sorted_data.join('')) }
+  output_file_path = ENV['OUTPUT_PATH']
+  File.open(output_file_path, 'wb') { |f| f.write(sorted_data.join('')) }
+end

src/ruby/sort_builtin.rb

@@ -14,12 +14,12 @@ def do_sorting(input_list)
 # business logic end #
 ######################
 
-# 👇🏽 copy pasted helpers
+if __FILE__ == $PROGRAM_NAME
+  input_file_path = ENV['INPUT_PATH']
+  input_file = File.readlines(input_file_path)
 
-input_file_path = ENV['INPUT_PATH']
-input_file = File.readlines(input_file_path)
+  sorted_data = do_sorting(input_file)
 
-sorted_data = do_sorting(input_file)
-
-output_file_path = ENV['OUTPUT_PATH']
-File.open(output_file_path, 'wb') { |f| f.write(sorted_data.join('')) }
+  output_file_path = ENV['OUTPUT_PATH']
+  File.open(output_file_path, 'wb') { |f| f.write(sorted_data.join('')) }
+end

src/ruby/sort_insertion_sort.rb

@@ -55,12 +55,12 @@ def swap(idx)
 # business logic end #
 ######################
 
-# 👇🏽 copy pasted helpers
+if __FILE__ == $PROGRAM_NAME
+  input_file_path = ENV['INPUT_PATH']
+  input_file = File.readlines(input_file_path)
 
-input_file_path = ENV['INPUT_PATH']
-input_file = File.readlines(input_file_path)
+  sorted_data = do_sorting(input_file)
 
-sorted_data = do_sorting(input_file)
-
-output_file_path = ENV['OUTPUT_PATH']
-File.open(output_file_path, 'wb') { |f| f.write(sorted_data.join('')) }
+  output_file_path = ENV['OUTPUT_PATH']
+  File.open(output_file_path, 'wb') { |f| f.write(sorted_data.join('')) }
+end

src/ruby/sort_selection_sort.rb

@@ -56,12 +56,12 @@ def find_smallest_index(input_list)
 # business logic end #
 ######################
 
-# 👇🏽 copy pasted helpers
+if __FILE__ == $PROGRAM_NAME
+  input_file_path = ENV['INPUT_PATH']
+  input_file = File.readlines(input_file_path)
 
-input_file_path = ENV['INPUT_PATH']
-input_file = File.readlines(input_file_path)
+  sorted_data = do_sorting(input_file)
 
-sorted_data = do_sorting(input_file)
-
-output_file_path = ENV['OUTPUT_PATH']
-File.open(output_file_path, 'wb') { |f| f.write(sorted_data.join('')) }
+  output_file_path = ENV['OUTPUT_PATH']
+  File.open(output_file_path, 'wb') { |f| f.write(sorted_data.join('')) }
+end