Add crontab

config.go

@@ -40,6 +40,11 @@ func (c *UserConfig) UnmarshalJSON(data []byte) error {
 		return err
 	}
 
+	// set default value
+	if cfg.SecurityStr == "" {
+		cfg.SecurityStr = "AUTO"
+	}
+
 	cfg.securityConfig = &protocol.SecurityConfig{
 		Type: protocol.SecurityType(protocol.SecurityType_value[strings.ToUpper(cfg.SecurityStr)]),
 	}
@@ -54,6 +59,7 @@ type myPluginConfig struct {
 	TrafficRate float64      `json:"trafficRate"`
 	MySQL       *MySQLConfig `json:"mysql"`
 	UserConfig  *UserConfig  `json:"user"`
+	GRPCAddr    string       `json:"gRPCAddr"`
 }
 
 type Config struct {

panel.go

@@ -2,8 +2,8 @@ package v2ray_ssrpanel_plugin
 
 import (
 	"fmt"
+	"github.com/robfig/cron"
 	"google.golang.org/grpc"
-	"time"
 	"v2ray.com/core/common/protocol"
 	"v2ray.com/core/common/serial"
 	"v2ray.com/core/proxy/vmess"
@@ -25,17 +25,28 @@ func NewPanel(gRPCConn *grpc.ClientConn, db *DB, globalConfig *Config) *Panel {
 }
 
 func (p *Panel) Start() {
-	for {
+	doFunc := func() {
 		if err := p.do(); err != nil {
 			newError("panel#do").Base(err).AtError().WriteToLog()
 		}
-		time.Sleep(10 * time.Second)
 	}
+	doFunc()
+
+	c := cron.New()
+	c.AddFunc(fmt.Sprintf("@every %ds", p.globalConfig.myPluginConfig.CheckRate), doFunc)
+	c.Start()
+	c.Run()
 }
 
 func (p *Panel) do() error {
-	newError("start doing ssr panel jobs").AtWarning().WriteToLog()
-	defer newError("finished doing ssr panel jobs").AtWarning().WriteToLog()
+	var addedUserCount, deletedUserCount int
+	var uplinkTraffic, downlinkTraffic int64
+	newError("start jobs").AtDebug().WriteToLog()
+	defer func() {
+		// todo
+		newError(fmt.Sprintf("jobs info: addded %d users, deleteted %d users, downlink traffic %d KB, uplink traffic %d KB",
+			addedUserCount, deletedUserCount, downlinkTraffic, uplinkTraffic)).AtInfo().WriteToLog()
+	}()
 
 	userModels, err := p.db.GetAllUsers()
 	if err != nil {
@@ -69,14 +80,14 @@ func (p *Panel) do() error {
 			p.handlerServiceClient.DelUser(userModel.Email)
 		}
 	}
+	deletedUserCount = len(delUserModels)
 
-	fmt.Println(addUserModels)
-	fmt.Println(delUserModels)
 	// Add
 	p.userModels = append(p.userModels, addUserModels...)
 	for _, userModel := range addUserModels {
 		p.handlerServiceClient.AddUser(p.convertUser(userModel))
 	}
+	addedUserCount = len(addUserModels)
 
 	return nil
 }

plugin.go

@@ -1,6 +1,7 @@
 package v2ray_ssrpanel_plugin
 
 import (
+	"fmt"
 	"os"
 	"time"
 	"v2ray.com/core/common/errors"
@@ -30,11 +31,12 @@ func run() error {
 	}
 
 	go func() {
-		grpcAddr := "localhost:10086"
-		gRPCConn, err := connectGRPC(grpcAddr, 45*time.Second)
+		gRPCAddr := globalCfg.myPluginConfig.GRPCAddr
+		gRPCConn, err := connectGRPC(gRPCAddr, 45*time.Second)
 		if err != nil {
-			fatal("connect to gRPC server: ", err)
+			fatal(fmt.Sprintf("connect to gRPC server \"%s\"", gRPCAddr), err)
 		}
+		newError(fmt.Sprintf("Connected gRPC server \"%s\"", gRPCAddr)).AtWarning().WriteToLog()
 
 		p := NewPanel(gRPCConn, db, globalCfg)
 		p.Start()