Skip to content

Commit 876d7d5

Browse files
ovrovr
committed
feat: Support FETCH (cursors)
1 parent 0fa812b commit 876d7d5

File tree

4 files changed

+180
-0
lines changed

4 files changed

+180
-0
lines changed

‎src/ast/mod.rs

Lines changed: 89 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -903,6 +903,17 @@ pub enum Statement {
903903
/// deleted along with the dropped table
904904
purge: bool,
905905
},
906+
/// FETCH - retrieve rows from a query using a cursor
907+
///
908+
/// Note: this is a PostgreSQL-specific statement,
909+
/// but may also compatible with other SQL.
910+
Fetch {
911+
/// Cursor name
912+
name: ObjectName,
913+
direction: FetchDirection,
914+
/// Optional, It's possible to fetch rows form cursor to the table
915+
into: Option<ObjectName>,
916+
},
906917
/// DISCARD [ ALL | PLANS | SEQUENCES | TEMPORARY | TEMP ]
907918
///
908919
/// Note: this is a PostgreSQL-specific statement,
@@ -1113,6 +1124,21 @@ impl fmt::Display for Statement {
11131124
write!(f, "{}", statement)
11141125
}
11151126
Statement::Query(s) => write!(f, "{}", s),
1127+
Statement::Fetch {
1128+
name,
1129+
direction,
1130+
into,
1131+
} => {
1132+
write!(f, "FETCH {} ", direction)?;
1133+
1134+
write!(f, "IN {}", name)?;
1135+
1136+
if let Some(into) = into {
1137+
write!(f, " INTO {}", into)?;
1138+
}
1139+
1140+
Ok(())
1141+
}
11161142
Statement::Directory {
11171143
overwrite,
11181144
local,
@@ -1842,6 +1868,69 @@ impl fmt::Display for Privileges {
18421868
}
18431869
}
18441870

1871+
/// Specific direction for FETCH statement
1872+
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
1873+
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
1874+
pub enum FetchDirection {
1875+
Count { limit: Value },
1876+
Next,
1877+
Prior,
1878+
First,
1879+
Last,
1880+
Absolute { limit: Value },
1881+
Relative { limit: Value },
1882+
All,
1883+
// FORWARD
1884+
// FORWARD count
1885+
Forward { limit: Option<Value> },
1886+
ForwardAll,
1887+
// BACKWARD
1888+
// BACKWARD count
1889+
Backward { limit: Option<Value> },
1890+
BackwardAll,
1891+
}
1892+
1893+
impl fmt::Display for FetchDirection {
1894+
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
1895+
match self {
1896+
FetchDirection::Count { limit } => f.write_str(&limit.to_string())?,
1897+
FetchDirection::Next => f.write_str("NEXT")?,
1898+
FetchDirection::Prior => f.write_str("PRIOR")?,
1899+
FetchDirection::First => f.write_str("FIRST")?,
1900+
FetchDirection::Last => f.write_str("LAST")?,
1901+
FetchDirection::Absolute { limit } => {
1902+
f.write_str("ABSOLUTE ")?;
1903+
f.write_str(&limit.to_string())?;
1904+
}
1905+
FetchDirection::Relative { limit } => {
1906+
f.write_str("RELATIVE ")?;
1907+
f.write_str(&limit.to_string())?;
1908+
}
1909+
FetchDirection::All => f.write_str("ALL")?,
1910+
FetchDirection::Forward { limit } => {
1911+
f.write_str("FORWARD")?;
1912+
1913+
if let Some(l) = limit {
1914+
f.write_str(" ")?;
1915+
f.write_str(&l.to_string())?;
1916+
}
1917+
}
1918+
FetchDirection::ForwardAll => f.write_str("FORWARD ALL")?,
1919+
FetchDirection::Backward { limit } => {
1920+
f.write_str("BACKWARD")?;
1921+
1922+
if let Some(l) = limit {
1923+
f.write_str(" ")?;
1924+
f.write_str(&l.to_string())?;
1925+
}
1926+
}
1927+
FetchDirection::BackwardAll => f.write_str("BACKWARD ALL")?,
1928+
};
1929+
1930+
Ok(())
1931+
}
1932+
}
1933+
18451934
/// A privilege on a database object (table, sequence, etc.).
18461935
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
18471936
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]

‎src/keywords.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ macro_rules! define_keywords {
6767
define_keywords!(
6868
ABORT,
6969
ABS,
70+
ABSOLUTE,
7071
ACTION,
7172
ADD,
7273
ALL,
@@ -92,6 +93,7 @@ define_keywords!(
9293
AUTO_INCREMENT,
9394
AVG,
9495
AVRO,
96+
BACKWARD,
9597
BEGIN,
9698
BEGIN_FRAME,
9799
BEGIN_PARTITION,
@@ -236,6 +238,7 @@ define_keywords!(
236238
FORCE_QUOTE,
237239
FOREIGN,
238240
FORMAT,
241+
FORWARD,
239242
FRAME_ROW,
240243
FREE,
241244
FREEZE,
@@ -383,6 +386,7 @@ define_keywords!(
383386
PREPARE,
384387
PRESERVE,
385388
PRIMARY,
389+
PRIOR,
386390
PRIVILEGES,
387391
PROCEDURE,
388392
PROGRAM,
@@ -411,6 +415,7 @@ define_keywords!(
411415
REGR_SXX,
412416
REGR_SXY,
413417
REGR_SYY,
418+
RELATIVE,
414419
RELEASE,
415420
RENAME,
416421
REPAIR,

‎src/parser.rs

Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,7 @@ impl<'a> Parser<'a> {
167167
Keyword::CREATE => Ok(self.parse_create()?),
168168
Keyword::DROP => Ok(self.parse_drop()?),
169169
Keyword::DISCARD => Ok(self.parse_discard()?),
170+
Keyword::FETCH => Ok(self.parse_fetch_statement()?),
170171
Keyword::DELETE => Ok(self.parse_delete()?),
171172
Keyword::INSERT => Ok(self.parse_insert()?),
172173
Keyword::UPDATE => Ok(self.parse_update()?),
@@ -1786,6 +1787,67 @@ impl<'a> Parser<'a> {
17861787
})
17871788
}
17881789

1790+
// FETCH [ direction { FROM | IN } ] cursor INTO target;
1791+
pub fn parse_fetch_statement(&mut self) -> Result<Statement, ParserError> {
1792+
let direction = if self.parse_keyword(Keyword::NEXT) {
1793+
FetchDirection::Next
1794+
} else if self.parse_keyword(Keyword::PRIOR) {
1795+
FetchDirection::Prior
1796+
} else if self.parse_keyword(Keyword::FIRST) {
1797+
FetchDirection::First
1798+
} else if self.parse_keyword(Keyword::LAST) {
1799+
FetchDirection::Last
1800+
} else if self.parse_keyword(Keyword::ABSOLUTE) {
1801+
FetchDirection::Absolute {
1802+
limit: self.parse_number_value()?,
1803+
}
1804+
} else if self.parse_keyword(Keyword::RELATIVE) {
1805+
FetchDirection::Relative {
1806+
limit: self.parse_number_value()?,
1807+
}
1808+
} else if self.parse_keyword(Keyword::FORWARD) {
1809+
if self.parse_keyword(Keyword::ALL) {
1810+
FetchDirection::ForwardAll
1811+
} else {
1812+
FetchDirection::Forward {
1813+
// TODO: Support optional
1814+
limit: Some(self.parse_number_value()?),
1815+
}
1816+
}
1817+
} else if self.parse_keyword(Keyword::BACKWARD) {
1818+
if self.parse_keyword(Keyword::ALL) {
1819+
FetchDirection::BackwardAll
1820+
} else {
1821+
FetchDirection::Backward {
1822+
// TODO: Support optional
1823+
limit: Some(self.parse_number_value()?),
1824+
}
1825+
}
1826+
} else if self.parse_keyword(Keyword::ALL) {
1827+
FetchDirection::All
1828+
} else {
1829+
FetchDirection::Count {
1830+
limit: self.parse_number_value()?,
1831+
}
1832+
};
1833+
1834+
self.expect_one_of_keywords(&[Keyword::FROM, Keyword::IN])?;
1835+
1836+
let name = self.parse_object_name()?;
1837+
1838+
let into = if self.parse_keyword(Keyword::INTO) {
1839+
Some(self.parse_object_name()?)
1840+
} else {
1841+
None
1842+
};
1843+
1844+
Ok(Statement::Fetch {
1845+
name,
1846+
direction,
1847+
into,
1848+
})
1849+
}
1850+
17891851
pub fn parse_discard(&mut self) -> Result<Statement, ParserError> {
17901852
let object_type = if self.parse_keyword(Keyword::ALL) {
17911853
DiscardObject::ALL

‎tests/sqlparser_postgres.rs

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1508,3 +1508,27 @@ fn parse_escaped_literal_string() {
15081508
"sql parser error: Unterminated encoded string literal at Line: 1, Column 8"
15091509
);
15101510
}
1511+
1512+
#[test]
1513+
fn parse_fetch() {
1514+
pg_and_generic().verified_stmt("FETCH 2048 IN \"SQL_CUR0x7fa44801bc00\"");
1515+
pg_and_generic().verified_stmt("FETCH 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1516+
pg_and_generic().verified_stmt("FETCH NEXT IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1517+
pg_and_generic().verified_stmt("FETCH PRIOR IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1518+
pg_and_generic().verified_stmt("FETCH FIRST IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1519+
pg_and_generic().verified_stmt("FETCH LAST IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1520+
pg_and_generic()
1521+
.verified_stmt("FETCH ABSOLUTE 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1522+
pg_and_generic()
1523+
.verified_stmt("FETCH RELATIVE 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1524+
pg_and_generic().verified_stmt("FETCH ALL IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1525+
pg_and_generic().verified_stmt("FETCH ALL IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1526+
pg_and_generic()
1527+
.verified_stmt("FETCH FORWARD 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1528+
pg_and_generic()
1529+
.verified_stmt("FETCH FORWARD ALL IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1530+
pg_and_generic()
1531+
.verified_stmt("FETCH BACKWARD 2048 IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1532+
pg_and_generic()
1533+
.verified_stmt("FETCH BACKWARD ALL IN \"SQL_CUR0x7fa44801bc00\" INTO \"new_table\"");
1534+
}

0 commit comments

Comments
 (0)