From f45e4cb8045fe60f96f467dc4ea1e2ae57445712 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期三, 03 七月 2024 23:13:40 +0800 Subject: [PATCH] 默认位置parent_id为视图位置id --- main.go | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 54 insertions(+), 0 deletions(-) diff --git a/main.go b/main.go index 3ee1279..a2ea7c2 100644 --- a/main.go +++ b/main.go @@ -2,6 +2,9 @@ import ( "context" + "fmt" + "google.golang.org/grpc" + "net" "net/http" "os" "os/signal" @@ -10,7 +13,14 @@ "wms/conf" "wms/models" "wms/pkg/logx" + "wms/proto/init_client" + "wms/proto/inventory_order" + "wms/proto/product_inventory" + "wms/proto/purchase_wms" "wms/router" + "wms/service" + "wms/task" + "wms/utils/dingtalkrobot" ) func main() { @@ -38,6 +48,44 @@ WriteTimeout: 5 * time.Second, } go shutdown(server) + //鍚姩grpc瀹㈡埛绔� + //go controllers.InitInventoryOrderServiceConn() + //go controllers.InitProductInventoryServiceConn() + //go controllers.InitCodeServiceConn() + //go supplier.InitConn() + //go client.InitConn() + go init_client.InitClient() + //鍚姩grpc鏈嶅姟 + go func() { + ln, err := net.Listen("tcp", ":"+conf.WebConf.GrpcPort) + if err != nil { + logx.Errorf("grpc server init error: %v", err.Error()) + panic(fmt.Sprintf("grpc server init error: %v", err.Error())) + } + s := grpc.NewServer() + //todo 娣诲姞鍏蜂綋鏈嶅姟 + product_inventory.RegisterProductInventoryServiceServer(s, &product_inventory.Server{}) + purchase_wms.RegisterPurchaseServiceServer(s, &purchase_wms.Server{}) + inventory_order.RegisterInventoryOrderServiceServer(s, &inventory_order.Server{}) + err = s.Serve(ln) + if err != nil { + logx.Errorf("grpc server init error: %v", err.Error()) + panic(fmt.Sprintf("grpc server init error: %v", err.Error())) + } + }() + + go service.InitLocationReportData() + go service.InitHistoryReportData() + + //瀹氭椂浠诲姟鍒濆鍖� + if err := task.Init(); err != nil { + logx.Errorf("task init error: %v", err.Error()) + return + } + + //閽夐拤鏈哄櫒浜哄垵濮嬪寲 + dingtalkrobot.Init(conf.DingTalkConf.AlarmKey, conf.DingTalkConf.AlarmUrl) + logx.Error(server.ListenAndServe().Error()) } @@ -51,6 +99,12 @@ ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) defer cancel() + //controllers.CloseInventoryOrderServiceConn() + //controllers.CloseProductInventoryServiceConn() + //controllers.CloseCodeServiceConn() + //supplier.CloseConn() + //client.CloseConn() + init_client.CloseClient() // 鍏抽棴HTTP鏈嶅姟鍣� if err := server.Shutdown(ctx); err != nil { logx.Infof("鏈嶅姟浼橀泤閫�鍑哄け璐�: %v", err) -- Gitblit v1.8.0