Apply suggestions from code review

Co-authored-by: dennis zhuang <killme2008@gmail.com>
Co-authored-by: LFC <990479+MichaelScofield@users.noreply.github.com>
Signed-off-by: Ruihang Xia <waynestxia@gmail.com>

Author: 3 people

src/operator/src/statement/comment.rs

@@ -27,6 +27,16 @@ use crate::error::{ExecuteDdlSnafu, ExternalSnafu, InvalidSqlSnafu, Result};
 use crate::statement::StatementExecutor;
 
 impl StatementExecutor {
+    /// Adds a comment to a database object (table, column, or flow).
+    ///
+    /// # Arguments
+    ///
+    /// * `stmt`: A `Comment` struct containing the object to comment on and the comment text.
+    /// * `query_ctx`: A `QueryContextRef` providing contextual information for the query.
+    ///
+    /// # Returns
+    ///
+    /// A `Result` containing the `Output` of the operation, or an error if the operation fails.
     pub async fn comment(&self, stmt: Comment, query_ctx: QueryContextRef) -> Result<Output> {
         let comment_on_task = self.create_comment_on_task_from_stmt(stmt, &query_ctx)?;
 

src/sql/src/parsers/comment_parser.rs

@@ -34,7 +34,7 @@ impl ParserContext<'_> {
         let target_token = self.parser.next_token();
         let comment = match target_token.token {
             Token::Word(word) if word.keyword == Keyword::TABLE => {
-                let raw_table = self.parse_object_name().context(error::UnexpectedSnafu {
+                let raw_table = self.parse_object_name().with_context(|_| error::UnexpectedSnafu {
                     expected: "a table name",
                     actual: self.peek_token_as_string(),
                 })?;
@@ -47,7 +47,7 @@ impl ParserContext<'_> {
             Token::Word(word)
                 if word.keyword == Keyword::NoKeyword && word.value.eq_ignore_ascii_case(FLOW) =>
             {
-                let raw_flow = self.parse_object_name().context(error::UnexpectedSnafu {
+                let raw_flow = self.parse_object_name().with_context(|_| error::UnexpectedSnafu {
                     expected: "a flow name",
                     actual: self.peek_token_as_string(),
                 })?;
@@ -78,7 +78,7 @@ impl ParserContext<'_> {
     }
 
     fn parse_column_comment_target(&mut self) -> Result<CommentObject> {
-        let raw = self.parse_object_name().context(error::UnexpectedSnafu {
+        let raw = self.parse_object_name().with_context(|_| error::UnexpectedSnafu {
             expected: "a column reference",
             actual: self.peek_token_as_string(),
         })?;