risingwave_meta_model_migration/
utils.rs

1use sea_orm::DatabaseBackend;
2use sea_orm_migration::prelude::*;
3
4#[easy_ext::ext(ColumnDefExt)]
5impl ColumnDef {
6    /// Set column type as `longblob` for MySQL, `bytea` for Postgres, and `blob` for Sqlite.
7    ///
8    /// Should be preferred over [`binary`](ColumnDef::binary) or [`blob`](ColumnDef::blob) for large binary fields,
9    /// typically the fields wrapping protobuf or other serialized data. Otherwise, MySQL will return an error
10    /// when the length exceeds 65535 bytes.
11    pub fn rw_binary(&mut self, manager: &SchemaManager) -> &mut Self {
12        match manager.get_database_backend() {
13            DatabaseBackend::MySql => self.custom(extension::mysql::MySqlType::LongBlob),
14            #[expect(clippy::disallowed_methods)]
15            DatabaseBackend::Postgres | DatabaseBackend::Sqlite => self.blob(),
16        }
17    }
18
19    /// Set column type as `longtext` for MySQL, and `text` for Postgres and Sqlite.
20    ///
21    /// Should be preferred over [`text`](ColumnDef::text) or [`string`](ColumnDef::string) for large text fields,
22    /// typically user-specified contents like UDF body or SQL definition. Otherwise, MySQL will return an error
23    /// when the length exceeds 65535 bytes.
24    pub fn rw_long_text(&mut self, manager: &SchemaManager) -> &mut Self {
25        match manager.get_database_backend() {
26            DatabaseBackend::MySql => self.custom(Alias::new("longtext")),
27            DatabaseBackend::Postgres | DatabaseBackend::Sqlite => self.text(),
28        }
29    }
30}
risingwave_meta_model_migration/utils.rs

risingwave_meta_model_migration/
utils.rs
