Top Level Namespace

Defined Under Namespace

Classes: EachSQL

Instance Method Summary collapse

Instance Method Details

#EachSQL(input, type = :default) ⇒ Object

Shortcut method for creating a Enumerable EachSQL object for the given input. @param input Input script. @param The type of the input SQL script. :default, :mysql, and :oracle (or :plsql) @yield Executable SQL statement or block. @return Array of executable SQL statements and blocks.



13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/each_sql.rb', line 13

def EachSQL input, type = :default
  esql   = EachSQL.new(type)
  ret    = []
  result = {}

  process = lambda {
    return if esql.empty?
    result = esql.shift
    sqls   = result[:sqls]
    sqls.each do |sql|
      if block_given?
        yield sql
      else
        ret << sql
      end
    end
  }

  input.to_s.each_line do |line|
    case line
    when /^\s*delimiter\s+(\S+)/i
      process.call
      if esql.empty?
        esql.delimiter = $1
      else
        esql << line
      end
    when /#{Regexp.escape esql.delimiter}/
      esql << line
      process.call
    else
      esql << line
    end
  end

  if !esql.empty?
    process.call
  end

  if sql = result[:leftover]
    if block_given?
      yield sql
    else
      ret << sql
    end
  end

  ret
end