feat: add pgdog.currval and pgdog.setval

pgdog/src/frontend/router/parser/rewrite/ee/nextval.rs

@@ -3,3 +3,11 @@ use super::Error;
 pub(crate) async fn nextval(_name: &str) -> Result<i64, Error> {
     Err(Error::EERequired)
 }
+
+pub(crate) async fn currval(_name: &str) -> Result<i64, Error> {
+    Err(Error::EERequired)
+}
+
+pub(crate) async fn setval(_name: &str, _value: i64, _is_called: bool) -> Result<i64, Error> {
+    Err(Error::EERequired)
+}

pgdog/src/frontend/router/parser/rewrite/statement/auto_id.rs

@@ -227,6 +227,7 @@ fn is_bigint_type(data_type: &str) -> bool {
 
 #[cfg(test)]
 mod tests {
+    use super::super::nextval::SequenceCall;
     use super::super::plan::GeneratedId;
     use crate::frontend::router::sharding::ShardedTable;
     use indexmap::IndexMap;
@@ -691,8 +692,14 @@ mod tests {
                 assert_eq!(
                     plan.generated_ids,
                     vec![
-                        (1, GeneratedId::Sequence(sequence.to_owned())),
-                        (2, GeneratedId::Sequence(sequence.to_owned())),
+                        (
+                            1,
+                            GeneratedId::Sequence(SequenceCall::Nextval(sequence.to_owned()))
+                        ),
+                        (
+                            2,
+                            GeneratedId::Sequence(SequenceCall::Nextval(sequence.to_owned()))
+                        ),
                     ]
                 );
             }

pgdog/src/frontend/router/parser/rewrite/statement/mod.rs

@@ -166,7 +166,7 @@ impl<'a> StatementRewrite<'a> {
                     }
                     Ok(None) => {
                         if let Some(replacement) =
-                            self.rewrite_nextval(node.as_ref(), mem, &mut next_param, &mut plan)
+                            self.rewrite_sequence(node.as_ref(), mem, &mut next_param, &mut plan)
                         {
                             node.replace(replacement);
                             None