oak-sql 0.0.11

SQL database query language parser with support for standard SQL syntax and database operations.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use crate::ast::statements::{
    ddl::{AlterStatement, CreateStatement, DropStatement},
    dml::{DeleteStatement, InsertStatement, UpdateStatement},
    query::SelectStatement,
};
use core::range::Range;
use oak_core::source::{SourceBuffer, ToSource};
use std::sync::Arc;

/// SQL root node representing a collection of SQL statements.
#[derive(Debug, Clone, Default)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct SqlRoot {
    /// The list of SQL statements in the script.
    pub statements: Vec<SqlStatement>,
    /// The full span of the SQL script.
    #[serde(with = "oak_core::serde_range")]
    pub span: Range<usize>,
}

impl ToSource for SqlRoot {
    fn to_source(&self, buffer: &mut SourceBuffer) {
        for (i, stmt) in self.statements.iter().enumerate() {
            if i > 0 {
                buffer.push(";");
            }
            stmt.to_source(buffer);
        }
        if !self.statements.is_empty() {
            buffer.push(";");
        }
    }
}

/// Represents a single SQL statement.
#[derive(Debug, Clone)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub enum SqlStatement {
    /// A SELECT statement.
    Select(SelectStatement),
    /// An INSERT statement.
    Insert(InsertStatement),
    /// An UPDATE statement.
    Update(UpdateStatement),
    /// A DELETE statement.
    Delete(DeleteStatement),
    /// A CREATE statement.
    Create(CreateStatement),
    /// A DROP statement.
    Drop(DropStatement),
    /// An ALTER statement.
    Alter(AlterStatement),
    /// An error occurred during parsing or building.
    Error {
        /// The error message.
        message: Arc<str>,
        /// The span where the error occurred.
        #[serde(with = "oak_core::serde_range")]
        span: Range<usize>,
    },
    /// An unknown or unparsed statement.
    Unknown {
        /// The span of the unknown statement.
        #[serde(with = "oak_core::serde_range")]
        span: Range<usize>,
    },
}

impl ToSource for SqlStatement {
    fn to_source(&self, buffer: &mut SourceBuffer) {
        match self {
            SqlStatement::Select(s) => s.to_source(buffer),
            SqlStatement::Insert(s) => s.to_source(buffer),
            SqlStatement::Update(s) => s.to_source(buffer),
            SqlStatement::Delete(s) => s.to_source(buffer),
            SqlStatement::Create(s) => s.to_source(buffer),
            SqlStatement::Drop(s) => s.to_source(buffer),
            SqlStatement::Alter(s) => s.to_source(buffer),
            SqlStatement::Error { message, .. } => {
                buffer.push("/* ERROR: ");
                buffer.push(message);
                buffer.push(" */");
            }
            SqlStatement::Unknown { .. } => {}
        }
    }
}