refactor: make common function

Author: ctron

common/src/db/create.rs

@@ -0,0 +1,37 @@
+use sea_orm::{ConnectionTrait, DbErr};
+use sea_orm_migration::SchemaManager;
+use sea_query::{IntoIden, extension::postgres::Type};
+
+/// create a type, if it not already exists
+///
+/// This is required as Postgres doesn't support `CREATE TYPE IF NOT EXISTS`
+pub async fn create_enum_if_not_exists<T, I>(
+    manager: &SchemaManager<'_>,
+    name: impl IntoIden + Clone,
+    values: I,
+) -> Result<(), DbErr>
+where
+    T: IntoIden,
+    I: IntoIterator<Item = T>,
+{
+    let builder = manager.get_connection().get_database_backend();
+    let r#type = name.clone().into_iden();
+    let stmt = builder.build(Type::create().as_enum(name).values(values));
+    let stmt = format!(
+        r#"
+DO $$
+BEGIN
+  IF NOT EXISTS (
+    SELECT 1 FROM pg_type WHERE typname = '{name}'
+  ) THEN
+    {stmt};
+  END IF;
+END$$;
+"#,
+        name = r#type.to_string()
+    );
+
+    manager.get_connection().execute_unprepared(&stmt).await?;
+
+    Ok(())
+}

common/src/db/mod.rs

@@ -3,7 +3,10 @@ pub mod limiter;
 pub mod multi_model;
 pub mod query;
 
+mod create;
 mod func;
+
+pub use create::*;
 pub use func::*;
 
 use anyhow::Context;

migration/src/m0002010_add_advisory_scores.rs

@@ -5,6 +5,7 @@ use crate::{
 use sea_orm::sea_query::extension::postgres::*;
 use sea_orm_migration::prelude::*;
 use strum::VariantNames;
+use trustify_common::db::create_enum_if_not_exists;
 use trustify_module_ingestor::{
     graph::cvss::ScoreCreator,
     service::advisory::{csaf, cve, osv},
@@ -13,40 +14,6 @@ use trustify_module_ingestor::{
 #[derive(DeriveMigrationName)]
 pub struct Migration;
 
-/// create a type, if it not already exists
-///
-/// This is required as Postgres doesn't support `CREATE TYPE IF NOT EXISTS`
-pub async fn create_enum_if_not_exists<T, I>(
-    manager: &SchemaManager<'_>,
-    name: impl IntoIden + Clone,
-    values: I,
-) -> Result<(), DbErr>
-where
-    T: IntoIden,
-    I: IntoIterator<Item = T>,
-{
-    let builder = manager.get_connection().get_database_backend();
-    let r#type = name.clone().into_iden();
-    let stmt = builder.build(Type::create().as_enum(name).values(values));
-    let stmt = format!(
-        r#"
-DO $$
-BEGIN
-  IF NOT EXISTS (
-    SELECT 1 FROM pg_type WHERE typname = '{name}'
-  ) THEN
-    {stmt};
-  END IF;
-END$$;
-"#,
-        name = r#type.to_string()
-    );
-
-    manager.get_connection().execute_unprepared(&stmt).await?;
-
-    Ok(())
-}
-
 #[async_trait::async_trait]
 impl MigrationTraitWithData for Migration {
     async fn up(&self, manager: &SchemaDataManager) -> Result<(), DbErr> {