Dockerfile

2026-03-18 10:17:50 +08:00
parent 8a028ff412
commit 5c5dbc7420
6 changed files with 987 additions and 0 deletions
--- a/main.go
+++ b/main.go
@@ -0,0 +1,94 @@
+package main
+
+import (
+	assetController "assets/controller/asset"
+	enumController "assets/controller/enum"
+	procurementController "assets/controller/procurement"
+	stockController "assets/controller/stock"
+	syncController "assets/controller/sync"
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"gitea.com/red-future/common/http"
+	"gitea.com/red-future/common/jaeger"
+
+	_ "gitea.com/red-future/common/swagger"
+	_ "github.com/gogf/gf/contrib/drivers/pgsql/v2"
+
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+func main() {
+	ctx := context.Background()
+	defer jaeger.ShutDown(ctx)
+	// 注册路由
+	// http.Httpserver.BindMiddleware("/*", middleware.ModuleTenantCheck)
+	http.RouteRegister([]interface{}{
+		assetController.Asset,
+		assetController.AssetSku,
+		assetController.Category,
+		assetController.PrivateSku,
+		assetController.PrivateCategory,
+		enumController.Enum,
+		stockController.StockManage,
+		stockController.StockDetails,
+		stockController.StockBatch,
+		stockController.PrivateStock,
+		stockController.InventoryCount,
+		stockController.InventoryCountDetail,
+		stockController.InventoryWarning,
+		stockController.InventoryWarningHistory,
+		stockController.Location,
+		stockController.Warehouse,
+		stockController.Zone,
+		stockController.UnitConversion,
+		procurementController.Supplier,
+		procurementController.PurchaseOrder,
+		procurementController.PurchaseOrderItem,
+		syncController.Sync,
+	})
+	// 注册RPC服务
+	//if err := message.AutoRegisterServices(ctx, map[string]interface{}{
+	//	"assetSkuService": assetService.AssetSku,
+	//}); err != nil {
+	//	g.Log().Warningf(ctx, "RPC 服务注册失败，服务将无法通过 NATS 调用: %v", err)
+	//} else {
+	//	g.Log().Info(ctx, "RPC 服务注册成功")
+	//}
+	// 注册消息队列消费者
+	//msgPlugin, err := message.GetMsgPlugin(context.TODO(), message.MessageNATS)
+	//if err != nil {
+	//	g.Log().Warningf(ctx, "消息队列插件获取失败，服务将无法消费消息: %v", err)
+	//} else {
+	//	msgPlugin.Subscribe(ctx, &message.NatsSubscribeMsgConfig{
+	//		QueueName:     public.StockDetailGroupName,
+	//		Durable:       true,
+	//		DelayTime:     10,
+	//		ConsumerName:  public.StockDetailConsumerName,
+	//		AutoAck:       public.StockDetailAutoAck,
+	//		PrefetchCount: public.StockDetailPrefetchCount,
+	//		HandleFunc:    stockService.StockManage.AddStock,
+	//	})
+	//}
+
+	//plugin, err := message.GetMsgPlugin(message.MessageRedis)
+	//if err != nil {
+	//	g.Log().Warningf(ctx, "消息队列插件获取失败，服务将无法消费消息: %v", err)
+	//} else {
+	//	plugin.Subscribe(ctx, &message.RedisSubscribeMsgConfig{
+	//		QueueName:     public.StockDetailQueueName,
+	//		ConsumerName:  public.StockDetailConsumerName,
+	//		PrefetchCount: public.StockDetailPrefetchCount,
+	//		AutoAck:       public.StockDetailAutoAck,
+	//		HandleFunc:    stockService.StockManage.AddStock,
+	//	})
+	//}
+
+	// 监听退出信号
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+	g.Log().Info(ctx, "收到退出信号，正在关闭服务...")
+}