From b8ff7fe65b1d844248e5aa2ee4c188e4d7eec002 Mon Sep 17 00:00:00 2001 From: eraden Date: Sun, 11 Jun 2023 23:13:09 +0200 Subject: [PATCH] Add using migration functions --- .../src/shipping/m20230603_120810_shipping.rs | 21 ++++++++++++------- 1 file changed, 13 insertions(+), 8 deletions(-) diff --git a/migration/src/shipping/m20230603_120810_shipping.rs b/migration/src/shipping/m20230603_120810_shipping.rs index b67d38a..8fc1786 100644 --- a/migration/src/shipping/m20230603_120810_shipping.rs +++ b/migration/src/shipping/m20230603_120810_shipping.rs @@ -1,7 +1,9 @@ use sea_orm_migration::prelude::*; use crate::constraint::Check; -use crate::{auto_uuid_not_null, ts_def_now_not_null, AsIden, CreateConstraint, IntoColumnDef}; +use crate::{ + auto_uuid_not_null, ts_def_now_not_null, AsIden, CreateConstraint, DropTable, IntoColumnDef, +}; #[derive(DeriveMigrationName)] pub struct Migration; @@ -11,18 +13,21 @@ impl MigrationTrait for Migration { async fn up(&self, m: &SchemaManager) -> Result<(), DbErr> { Self::create_tracking_links(m).await?; Self::create_custom_shipping_options(m).await?; + + Self::create_shipping_methods(m).await?; + Self::create_shipping_method_tax_lines(m).await?; + Self::create_shipping_options(m).await?; + Self::create_shipping_option_requirements(m).await?; + Self::create_shipping_profiles(m).await?; + Self::create_shipping_tax_rates(m).await?; Ok(()) } async fn down(&self, m: &SchemaManager) -> Result<(), DbErr> { - m.drop_table(Table::drop().table(TrackingLink::TrackingLinks).to_owned()) + self.drop_table(m, TrackingLink::TrackingLinks).await?; + self.drop_table(m, CustomShippingOption::CustomShippingOptions) .await?; - m.drop_table( - Table::drop() - .table(CustomShippingOption::CustomShippingOptions) - .to_owned(), - ) - .await?; + Ok(()) } }