first commit

2026-08-05 17:37:32 +08:00
commit 38be479cef
72 changed files with 15670 additions and 0 deletions
@@ -0,0 +1,106 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"net"
+	"strconv"
+
+	"product/internal/config"
+	"product/internal/server"
+	"product/internal/svc"
+	"product/product"
+
+	"pkg.local/discovery"
+	"pkg.local/modelbase"
+	localmysql "pkg.local/mysql"
+	"pkg.local/validate"
+
+	"github.com/zeromicro/go-zero/core/conf"
+	"github.com/zeromicro/go-zero/core/service"
+	"github.com/zeromicro/go-zero/zrpc"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/reflection"
+)
+
+var configFile = flag.String("f", "etc/product.yaml", "the config file")
+
+func main() {
+	flag.Parse()
+
+	var c config.Config
+	conf.MustLoad(*configFile, &c)
+
+	if err := discovery.Init(discovery.Config{
+		Hosts:       c.Nacos.Hosts,
+		NamespaceId: c.Nacos.NamespaceId,
+		Group:       c.Nacos.Group,
+	}); err != nil {
+		log.Fatalf("nacos init: %v", err)
+	}
+
+	registerIP := c.Nacos.RegisterIP
+	port, err := listenPort(c.ListenOn)
+	if err != nil {
+		log.Fatalf("parse ListenOn: %v", err)
+	}
+	if err := discovery.Register(discovery.Instance{
+		ServiceName: c.Name,
+		IP:          registerIP,
+		Port:        port,
+		Group:       c.Nacos.Group,
+	}); err != nil {
+		log.Fatalf("nacos register: %v", err)
+	}
+	defer func() {
+		if err := discovery.Deregister(); err != nil {
+			log.Printf("nacos deregister: %v", err)
+		}
+	}()
+
+	db, err := localmysql.New(localmysql.Config{
+		Host:         c.Mysql.Host,
+		Port:         c.Mysql.Port,
+		User:         c.Mysql.User,
+		Password:     c.Mysql.Password,
+		Database:     c.Mysql.Database,
+		Charset:      c.Mysql.Charset,
+		Prefix:       c.Mysql.Prefix,
+		ReadHost:     c.Mysql.ReadHost,
+		ReadPort:     c.Mysql.ReadPort,
+		ReadUser:     c.Mysql.ReadUser,
+		ReadPassword: c.Mysql.ReadPassword,
+		ReadDatabase: c.Mysql.ReadDatabase,
+	})
+	if err != nil {
+		log.Fatalf("mysql init: %v", err)
+	}
+
+	// modelbase uses Table(prefix+name); keep gorm TablePrefix for Model(&T{}) paths.
+	modelbase.Init(db, modelbase.Config{Prefix: c.Mysql.Prefix})
+
+	ctx := svc.NewServiceContext(c, db)
+
+	s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
+		product.RegisterProductServer(grpcServer, server.NewProductServer(ctx))
+
+		if c.Mode == service.DevMode || c.Mode == service.TestMode {
+			reflection.Register(grpcServer)
+		}
+	})
+	defer s.Stop()
+
+	s.AddUnaryInterceptors(validate.UnaryServerInterceptor(validate.MustNew()))
+
+	fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
+	s.Start()
+}
+
+func listenPort(listenOn string) (uint64, error) {
+	_, portStr, err := net.SplitHostPort(listenOn)
+	if err != nil {
+		return 0, err
+	}
+	return strconv.ParseUint(portStr, 10, 64)
+}