Function split_sql_queries

src/parser/sql_parser.rs:69–149

Split SQL content into individual queries Handles semicolons inside strings and comments properly

(content: &str)

/// Split SQL content into individual queries
/// Handles semicolons inside strings and comments properly
fn split_sql_queries(content: &str) -> Vec<String> {
  let mut queries = Vec::new();
  let mut current_query = String::new();
  let mut in_string = false;
  let mut in_comment = false;
  let mut string_delimiter = '\0';
  let mut chars = content.chars().peekable();

  while let Some(ch) = chars.next() {
    match ch {
      // Handle string literals
      '\'' | '"' if !in_comment => {
        if !in_string {
          in_string = true;
          string_delimiter = ch;
        } else if ch == string_delimiter {
          // Check for escaped quotes
          if chars.peek() == Some(&ch) {
            current_query.push(ch);
            current_query.push(chars.next().unwrap()); // consume the escaped quote
            continue;
          } else {
            in_string = false;
          }
        }
        current_query.push(ch);
      }

      // Handle single-line comments
      '-' if !in_string && !in_comment && chars.peek() == Some(&'-') => {
        in_comment = true;
        current_query.push(ch);
        current_query.push(chars.next().unwrap()); // consume second dash
      }

      // Handle multi-line comments
      '/' if !in_string && !in_comment && chars.peek() == Some(&'*') => {
        in_comment = true;
        current_query.push(ch);
        current_query.push(chars.next().unwrap()); // consume asterisk
      }

      // Non-comment dash or slash
      '-' | '/' if !in_string && !in_comment => {
        current_query.push(ch);
      }

      '*' if in_comment && !in_string => {
        current_query.push(ch);
        if chars.peek() == Some(&'/') {
          current_query.push(chars.next().unwrap()); // consume slash
          in_comment = false;
        }
      }

      // Handle newlines (end single-line comments)
      '\n' | '\r' if in_comment => {
        in_comment = false;