grafbase-sql-ast 0.1.9

An AST to generate SQL queries
Documentation
use super::Function;
use crate::ast::Expression;

/// The encode format.
#[derive(Debug, Clone, Copy, PartialEq)]
pub enum EncodeFormat {
    Base64,
    Escape,
    Hex,
}

/// A representation of the `encode` function in PostgreSQL.
#[derive(Debug, Clone, PartialEq)]
pub struct Encode<'a> {
    pub(crate) expression: Expression<'a>,
    pub(crate) format: EncodeFormat,
}

/// Return the given table as JSONB collection.
pub fn encode<'a>(expression: impl Into<Expression<'a>>, format: EncodeFormat) -> Function<'a> {
    let fun = Encode {
        expression: expression.into(),
        format,
    };

    fun.into()
}

impl<'a> From<Encode<'a>> for Function<'a> {
    fn from(value: Encode<'a>) -> Self {
        Self {
            typ_: super::FunctionType::Encode(value),
            alias: None,
        }
    }
}