Sha256: 69f48b27f509854bd4a1454396e783be392753eea3e92e17cd4716d13510cd96

Contents?: true

Size: 1.49 KB

Versions: 7

Compression:

Stored size: 1.49 KB

Contents

class Baza::SqlQueries::GenericInsert
  def initialize(args)
    @db = args.fetch(:db)
    @table_name = args.fetch(:table_name)
    @data = args.fetch(:data)
    @buffer = args[:buffer]
    @return_sql = args[:return_sql]
    @return_id = args[:return_id]
  end

  def execute
    if @return_sql
      to_sql
    elsif @buffer
      @buffer.query(to_sql)
    else
      @db.query(to_sql)
      @db.last_id if @return_id
    end
  end

  def to_sql
    sql = "INSERT INTO #{@db.sep_table}#{@db.escape_table(@table_name)}#{@db.sep_table}"

    if !@data || @data.empty?
      sql << " #{sql_default_values}"
    else
      sql << " #{sql_columns} VALUES #{sql_values}"
    end

    sql
  end

private

  def sql_default_values
    if @db.opts.fetch(:type).to_s.include?("mysql")
      "VALUES ()" # This is the correct syntax for inserting a blank row in MySQL.
    elsif @db.opts.fetch(:type).to_s.include?("sqlite3")
      "DEFAULT VALUES"
    else
      raise "Unknown database-type: '#{@db.opts.fetch(:type)}'."
    end
  end

  def sql_columns
    sql = "("

    first = true
    @data.each_key do |key|
      if first
        first = false
      else
        sql << ", "
      end

      sql << "#{@db.sep_col}#{@db.escape_column(key)}#{@db.sep_col}"
    end

    sql << ")"
    sql
  end

  def sql_values
    sql = "("

    first = true
    @data.each_value do |value|
      if first
        first = false
      else
        sql << ", "
      end

      sql << @db.sqlval(value)
    end

    sql << ")"
    sql
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version Path
baza-0.0.27 lib/baza/sql_queries/generic_insert.rb
baza-0.0.26 lib/baza/sql_queries/generic_insert.rb
baza-0.0.25 lib/baza/sql_queries/generic_insert.rb
baza-0.0.24 lib/baza/sql_queries/generic_insert.rb
baza-0.0.23 lib/baza/sql_queries/generic_insert.rb
baza-0.0.22 lib/baza/sql_queries/generic_insert.rb
baza-0.0.21 lib/baza/sql_queries/generic_insert.rb