/
manager.go
99 lines (84 loc) · 2.09 KB
/
manager.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package server
import (
"context"
"strings"
"time"
"github.com/catpie/musdk-go"
v2raymanager "github.com/orvice/v2ray-manager"
"github.com/weeon/utils/task"
)
func getV2rayManager() ([]*v2raymanager.Manager, error) {
arr := strings.Split(cfg.V2rayClientAddr, ",")
var vms = make([]*v2raymanager.Manager, len(arr))
for k, v := range arr {
vm, err := v2raymanager.NewManager(v, cfg.V2rayTag, sdkLogger)
if err != nil {
return nil, err
}
vms[k] = vm
}
return vms, nil
}
func (u *UserManager) check() error {
ctx := context.Background()
logger.Info("check users from mu")
users, err := apiClient.GetUsers()
if err != nil {
logger.Errorw("get users fail ",
"error", err,
)
return err
}
logger.Infof("get %d users from mu", len(users))
for _, user := range users {
u.checkUser(ctx, user)
}
return nil
}
func (u *UserManager) checkUser(ctx context.Context, user musdk.User) error {
var err error
if user.IsEnable() && !u.Exist(user) {
// run user
exist, err := u.vm.AddUser(ctx, &user.V2rayUser)
if err != nil {
logger.Errorf("add user %s error %v", user.V2rayUser.UUID, err)
return err
}
if !exist {
logger.Errorf("add user %s success", user.V2rayUser.UUID)
}
u.AddUser(user)
return nil
}
if !user.IsEnable() && u.Exist(user) {
logger.Infof("stop user id %d uuid %s", user.Id, user.V2rayUser.UUID)
// stop user
err = u.vm.RemoveUser(ctx, &user.V2rayUser)
if err != nil {
logger.Errorf("remove user error %v", err)
time.Sleep(time.Second * 10)
return err
}
u.RemoveUser(user)
return nil
}
return nil
}
func (u *UserManager) restartUser() {}
func (u *UserManager) Run() error {
task.NewTaskAndRun("check_users", cfg.SyncTime, u.check, task.SetTaskLogger(sdkLogger))
task.NewTaskAndRun("save_traffic", cfg.SyncTime, u.saveTrafficDaemon, task.SetTaskLogger(sdkLogger))
return nil
}
func (u *UserManager) Down() {
u.cancel()
}
func (u *UserManager) saveTrafficDaemon() error {
ctx := context.Background()
u.usersMu.RLock()
defer u.usersMu.RUnlock()
for _, user := range u.users {
u.saveUserTraffic(ctx, user)
}
return nil
}