feat: en redis and logs

product/product.go

@@ -3,8 +3,8 @@ package main
 import (
 	"flag"
 	"fmt"
-	"log"
 	"net"
+	"os"
 	"strconv"
 
 	"product/internal/config"
@@ -13,11 +13,14 @@ import (
 	"product/product"
 
 	"pkg.local/discovery"
+	"pkg.local/log"
 	"pkg.local/modelbase"
-	localmysql "pkg.local/mysql"
+	"pkg.local/mysql"
+	"pkg.local/redis"
 	"pkg.local/validate"
 
 	"github.com/zeromicro/go-zero/core/conf"
+	"github.com/zeromicro/go-zero/core/logx"
 	"github.com/zeromicro/go-zero/core/service"
 	"github.com/zeromicro/go-zero/zrpc"
 	"google.golang.org/grpc"
@@ -32,6 +35,19 @@ func main() {
 	var c config.Config
 	conf.MustLoad(*configFile, &c)
 
+	if err := log.Init(log.Config{
+		Path:       c.AppLog.Path,
+		InfoFile:   c.AppLog.InfoFile,
+		ErrorFile:  c.AppLog.ErrorFile,
+		FatalFile:  c.AppLog.FatalFile,
+		MaxSize:    c.AppLog.MaxSize,
+		MaxBackups: c.AppLog.MaxBackups,
+		MaxAge:     c.AppLog.MaxAge,
+	}); err != nil {
+		fmt.Fprintf(os.Stderr, "log init: %v\n", err)
+		os.Exit(1)
+	}
+
 	if err := discovery.Init(discovery.Config{
 		Hosts:       c.Nacos.Hosts,
 		NamespaceId: c.Nacos.NamespaceId,
@@ -55,11 +71,11 @@ func main() {
 	}
 	defer func() {
 		if err := discovery.Deregister(); err != nil {
-			log.Printf("nacos deregister: %v", err)
+			log.Errorf("nacos deregister: %v", err)
 		}
 	}()
 
-	db, err := localmysql.New(localmysql.Config{
+	db, err := mysql.New(mysql.Config{
 		Host:         c.Mysql.Host,
 		Port:         c.Mysql.Port,
 		User:         c.Mysql.User,
@@ -77,6 +93,15 @@ func main() {
 		log.Fatalf("mysql init: %v", err)
 	}
 
+	if err := redis.Init(redis.Config{
+		Host:     c.BizRedis.Host,
+		Port:     c.BizRedis.Port,
+		Password: c.BizRedis.Password,
+		DB:       c.BizRedis.DB,
+	}); err != nil {
+		log.Fatalf("redis init: %v", err)
+	}
+
 	// modelbase uses Table(prefix+name); keep gorm TablePrefix for Model(&T{}) paths.
 	modelbase.Init(db, modelbase.Config{Prefix: c.Mysql.Prefix})
 
@@ -91,9 +116,11 @@ func main() {
 	})
 	defer s.Stop()
 
+	logx.AddWriter(logx.NewWriter(os.Stdout))
+
 	s.AddUnaryInterceptors(validate.UnaryServerInterceptor(validate.MustNew()))
 
-	fmt.Printf("Starting rpc server at %s...\n", c.ListenOn)
+	log.Infof("Starting rpc server at %s...", c.ListenOn)
 	s.Start()
 }