29 lines
753 B
Go
29 lines
753 B
Go
package svc
|
|
|
|
import (
|
|
"godemo/category/category"
|
|
"godemo/file/file"
|
|
"godemo/gallery/internal/config"
|
|
"godemo/gallery/internal/model"
|
|
|
|
_ "github.com/lib/pq"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
"github.com/zeromicro/go-zero/zrpc"
|
|
)
|
|
|
|
type ServiceContext struct {
|
|
Config config.Config
|
|
FileRpc file.FileClient
|
|
CategoryRpc category.CategoryClient
|
|
ImagesModel model.ImagesModel
|
|
}
|
|
|
|
func NewServiceContext(c config.Config) *ServiceContext {
|
|
return &ServiceContext{
|
|
Config: c,
|
|
FileRpc: file.NewFileClient(zrpc.MustNewClient(c.FileRpc).Conn()),
|
|
CategoryRpc: category.NewCategoryClient(zrpc.MustNewClient(c.CategoryRpc).Conn()),
|
|
ImagesModel: model.NewImagesModel(sqlx.NewSqlConn("postgres", c.DB.DataSource)),
|
|
}
|
|
}
|