github
/
v2ray-core
mirror of https://github.com/v2ray/v2ray-core
1
0
Fork 0
Code Issues Releases Wiki Activity

refactor

pull/651/head
Darien Raymond 2017-08-24 22:56:31 +02:00
parent 1c2b6d9536
commit 7392bb495b
1 changed files with 7 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
common/protocol/user.go
View File

@ -2,18 +2,12 @@ package protocol
import "time" import "time"
var ( func (u *User) GetTypedAccount() (Account, error) {
ErrAccountMissing = newError("Account is not specified.") if u.GetAccount() == nil {
ErrNonMessageType = newError("Not a protobuf message.") return nil, newError("Account missing").AtWarning()
ErrUnknownAccountType = newError("Unknown account type.")
)
func (v *User) GetTypedAccount() (Account, error) {
if v.GetAccount() == nil {
return nil, ErrAccountMissing
} }
rawAccount, err := v.Account.GetInstance() rawAccount, err := u.Account.GetInstance()
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -23,12 +17,12 @@ func (v *User) GetTypedAccount() (Account, error) {
if account, ok := rawAccount.(Account); ok { if account, ok := rawAccount.(Account); ok {
return account, nil return account, nil
} }
return nil, newError("Unknown account type: ", v.Account.Type) return nil, newError("Unknown account type: ", u.Account.Type)
} }
func (v *User) GetSettings() UserSettings { func (u *User) GetSettings() UserSettings {
settings := UserSettings{} settings := UserSettings{}
switch v.Level { switch u.Level {
case 0: case 0:
settings.PayloadTimeout = time.Second * 30 settings.PayloadTimeout = time.Second * 30
case 1: case 1: