feat provider添加AutoMigrate

2024-08-26 17:09:29 +08:00 · 2024-08-26 17:09:29 +08:00 · 742fc5250a
commit 742fc5250a
parent db0310a6a0
1 changed files with 5 additions and 0 deletions
--- a/structs/storage/provider.go
+++ b/structs/storage/provider.go
@ -17,6 +17,7 @@ type ProviderApi interface {
 	Create(provider v2.Provider) error
 	Update(provider v2.Provider) error
 	Delete(id uint) (provider v2.Provider, err error)
+	AutoMigrate() error
 }

 func NewProviderApi(db *gorm.DB) ProviderApi {
@ -100,3 +101,7 @@ func (p *providerApi) Delete(id uint) (provider v2.Provider, err error) {
 	})
 	return
 }
+
+func (p *providerApi) AutoMigrate() error {
+	return p.db.AutoMigrate(&v2.Provider{}, &v2.CalculateProcess{})
+}