class Multiwoven::Integrations::Source::Oracle::Client

def reformat_query(sql_query)
  offset = nil
  limit = nil
  sql_query = sql_query.gsub(";", "")
  if sql_query.match?(/LIMIT (\d+)/i)
    limit = sql_query.match(/LIMIT (\d+)/i)[1].to_i
    sql_query.sub!(/LIMIT \d+/i, "")
  end
  if sql_query.match?(/OFFSET (\d+)/i)
    offset = sql_query.match(/OFFSET (\d+)/i)[1].to_i
    sql_query.sub!(/OFFSET \d+/i, "")
  end
  sql_query.strip!
  if offset && limit
    "#{sql_query} OFFSET #{offset} ROWS FETCH NEXT #{limit} ROWS ONLY"
  elsif offset
    "#{sql_query} OFFSET #{offset} ROWS"
  elsif limit
    "#{sql_query} FETCH NEXT #{limit} ROWS ONLY"
  else
    sql_query
  end
end