mirror of
https://github.com/v2fly/v2ray-core.git
synced 2025-01-02 23:47:07 -05:00
24 lines
430 B
Go
24 lines
430 B
Go
|
package mocks
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/v2ray/v2ray-core"
|
||
|
|
)
|
||
|
|
|
||
|
|
type MockUserSet struct {
|
||
|
|
UserIds []core.ID
|
||
|
|
UserHashes map[string]int
|
||
|
|
}
|
||
|
|
|
||
|
|
func (us *MockUserSet) AddUser(user core.User) error {
|
||
|
|
us.UserIds = append(us.UserIds, user.Id)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (us *MockUserSet) GetUser(userhash []byte) (*core.ID, bool) {
|
||
|
|
idx, found := us.UserHashes[string(userhash)]
|
||
|
|
if found {
|
||
|
|
return &us.UserIds[idx], true
|
||
|
|
}
|
||
|
|
return nil, false
|
||
|
|
}
|