ai_scheduler/internal/domain/tools/registry.go

45 lines
1.7 KiB
Go

package tools
import (
"ai_scheduler/internal/config"
"ai_scheduler/internal/domain/tools/hyt/goods_add"
"ai_scheduler/internal/domain/tools/hyt/goods_brand_search"
"ai_scheduler/internal/domain/tools/hyt/goods_category_add"
"ai_scheduler/internal/domain/tools/hyt/goods_category_search"
"ai_scheduler/internal/domain/tools/hyt/goods_media_add"
"ai_scheduler/internal/domain/tools/hyt/product_upload"
"ai_scheduler/internal/domain/tools/hyt/supplier_search"
"ai_scheduler/internal/domain/tools/hyt/warehouse_search"
)
type Manager struct {
Hyt *HytTools
// Zltx *ZltxTools
}
type HytTools struct {
ProductUpload *product_upload.Client
SupplierSearch *supplier_search.Client
WarehouseSearch *warehouse_search.Client
GoodsAdd *goods_add.Client
GoodsMediaAdd *goods_media_add.Client
GoodsCategoryAdd *goods_category_add.Client
GoodsCategorySearch *goods_category_search.Client
GoodsBrandSearch *goods_brand_search.Client
}
func NewManager(cfg *config.Config) *Manager {
return &Manager{
Hyt: &HytTools{
ProductUpload: product_upload.New(cfg.EinoTools.HytProductUpload),
SupplierSearch: supplier_search.New(cfg.EinoTools.HytSupplierSearch),
WarehouseSearch: warehouse_search.New(cfg.EinoTools.HytWarehouseSearch),
GoodsAdd: goods_add.New(cfg.EinoTools.HytGoodsAdd),
GoodsMediaAdd: goods_media_add.New(cfg.EinoTools.HytGoodsMediaAdd),
GoodsCategoryAdd: goods_category_add.New(cfg.EinoTools.HytGoodsCategoryAdd),
GoodsCategorySearch: goods_category_search.New(cfg.EinoTools.HytGoodsCategorySearch),
GoodsBrandSearch: goods_brand_search.New(cfg.EinoTools.HytGoodsBrandSearch),
},
}
}