pgmt 0.4.9

PostgreSQL migration tool that keeps your schema files as the source of truth
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

//! Utility functions for handling comment operations in diff modules
//! Eliminates code duplication across different database object types

use crate::catalog::comments::{CommentAction, Commentable, diff_comments};
use crate::catalog::target::AttrTarget;
use crate::diff::operations::CommentOperation;

/// Handle comment creation for new objects.
/// Returns a CommentOperation if the object has a comment.
pub fn handle_comment_creation(
    object_comment: &Option<String>,
    target: AttrTarget,
) -> Option<CommentOperation> {
    object_comment
        .as_ref()
        .map(|comment| CommentOperation::Set {
            target,
            comment: comment.clone(),
        })
}

/// Process comment diff actions and convert them to comment operations.
pub fn handle_comment_diff<C: Commentable + Clone>(
    old: Option<&C>,
    new: Option<&C>,
    target_factory: impl Fn() -> AttrTarget,
) -> Vec<CommentOperation> {
    let comment_actions = diff_comments(old, new);
    comment_actions
        .into_iter()
        .map(|action| match action {
            CommentAction::SetComment { comment } => CommentOperation::Set {
                target: target_factory(),
                comment,
            },
            CommentAction::DropComment => CommentOperation::Drop {
                target: target_factory(),
            },
        })
        .collect()
}