kedaya_haitao/common
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

feat storage 添加db方法

Browse Source
This commit is contained in:
timerzz 2024-09-02 16:12:22 +08:00
parent 1290c1e1fd
commit 06e03d4f50
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
structs/storage/storage.go
View File

@ -3,6 +3,7 @@ package storage
import "gorm.io/gorm" import "gorm.io/gorm"
type Storage struct { type Storage struct {
db *gorm.DB
articleApi ArticleApi articleApi ArticleApi
providerApi ProviderApi providerApi ProviderApi
providerArticleApi ProviderArticleApi providerArticleApi ProviderArticleApi
@ -12,6 +13,7 @@ type Storage struct {
func NewStorage(db *gorm.DB) *Storage { func NewStorage(db *gorm.DB) *Storage {
return &Storage{ return &Storage{
db: db,
articleApi: NewArticleApi(db), articleApi: NewArticleApi(db),
providerApi: NewProviderApi(db), providerApi: NewProviderApi(db),
providerArticleApi: NewProviderArticleApi(db), providerArticleApi: NewProviderArticleApi(db),
@ -20,6 +22,9 @@ func NewStorage(db *gorm.DB) *Storage {
} }
} }
func (s *Storage) DB() *gorm.DB {
return s.db
}
func (s *Storage) Article() ArticleApi { func (s *Storage) Article() ArticleApi {
return s.articleApi return s.articleApi
} }