git add

Hunter0x7c7

2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece

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
package vless 
 
import ( 
    "strings" 
    "sync" 
 
    "github.com/v2fly/v2ray-core/v5/common/protocol" 
    "github.com/v2fly/v2ray-core/v5/common/uuid" 
) 
 
// Validator stores valid VLESS users. 
type Validator struct { 
    // Considering email's usage here, map + sync.Mutex/RWMutex may have better performance. 
    email sync.Map 
    users sync.Map 
} 
 
// Add a VLESS user, Email must be empty or unique. 
func (v *Validator) Add(u *protocol.MemoryUser) error { 
    if u.Email != "" { 
        _, loaded := v.email.LoadOrStore(strings.ToLower(u.Email), u) 
        if loaded { 
            return newError("User ", u.Email, " already exists.") 
        } 
    } 
    v.users.Store(u.Account.(*MemoryAccount).ID.UUID(), u) 
    return nil 
} 
 
// Del a VLESS user with a non-empty Email. 
func (v *Validator) Del(e string) error { 
    if e == "" { 
        return newError("Email must not be empty.") 
    } 
    le := strings.ToLower(e) 
    u, _ := v.email.Load(le) 
    if u == nil { 
        return newError("User ", e, " not found.") 
    } 
    v.email.Delete(le) 
    v.users.Delete(u.(*protocol.MemoryUser).Account.(*MemoryAccount).ID.UUID()) 
    return nil 
} 
 
// Get a VLESS user with UUID, nil if user doesn't exist. 
func (v *Validator) Get(id uuid.UUID) *protocol.MemoryUser { 
    u, _ := v.users.Load(id) 
    if u != nil { 
        return u.(*protocol.MemoryUser) 
    } 
    return nil 
}