Refactor vmess config

This commit is contained in:
Claire Raymond
2015-10-16 10:03:22 +00:00
parent f10f08c87d
commit 771d0225c7
18 changed files with 263 additions and 184 deletions
+6 -6
View File
@@ -1,24 +1,24 @@
package mocks
import (
"github.com/v2ray/v2ray-core/proxy/vmess/protocol/user"
"github.com/v2ray/v2ray-core/proxy/vmess/config"
)
type MockUserSet struct {
UserIds []user.ID
UserIds []*config.ID
UserHashes map[string]int
Timestamps map[string]int64
}
func (us *MockUserSet) AddUser(user user.User) error {
us.UserIds = append(us.UserIds, user.Id)
func (us *MockUserSet) AddUser(user config.User) error {
us.UserIds = append(us.UserIds, user.ID())
return nil
}
func (us *MockUserSet) GetUser(userhash []byte) (*user.ID, int64, bool) {
func (us *MockUserSet) GetUser(userhash []byte) (*config.ID, int64, bool) {
idx, found := us.UserHashes[string(userhash)]
if found {
return &us.UserIds[idx], us.Timestamps[string(userhash)], true
return us.UserIds[idx], us.Timestamps[string(userhash)], true
}
return nil, 0, false
}