v2ray-core/common/protocol/user.go

45 lines
978 B
Go
Raw Normal View History

package protocol
2015-10-16 10:03:22 +00:00
2017-04-08 23:43:25 +00:00
import "time"
2015-10-30 23:38:31 +00:00
2016-09-17 22:41:21 +00:00
var (
2017-04-08 23:43:25 +00:00
ErrAccountMissing = newError("Account is not specified.")
ErrNonMessageType = newError("Not a protobuf message.")
ErrUnknownAccountType = newError("Unknown account type.")
2016-09-17 22:41:21 +00:00
)
2016-11-27 20:39:09 +00:00
func (v *User) GetTypedAccount() (Account, error) {
if v.GetAccount() == nil {
2016-09-17 22:41:21 +00:00
return nil, ErrAccountMissing
}
2016-10-16 12:22:21 +00:00
2016-11-27 20:39:09 +00:00
rawAccount, err := v.Account.GetInstance()
2016-09-17 22:41:21 +00:00
if err != nil {
return nil, err
}
2016-10-16 12:22:21 +00:00
if asAccount, ok := rawAccount.(AsAccount); ok {
return asAccount.AsAccount()
}
if account, ok := rawAccount.(Account); ok {
return account, nil
}
2017-04-08 23:43:25 +00:00
return nil, newError("Unknown account type: ", v.Account.Type)
2015-12-12 23:10:35 +00:00
}
2016-11-27 20:39:09 +00:00
func (v *User) GetSettings() UserSettings {
2017-01-31 16:46:39 +00:00
settings := UserSettings{}
switch v.Level {
case 0:
settings.PayloadTimeout = time.Second * 30
case 1:
settings.PayloadTimeout = time.Minute * 2
default:
settings.PayloadTimeout = time.Minute * 5
2015-10-31 13:08:13 +00:00
}
return settings
2015-10-16 10:03:22 +00:00
}
2016-09-17 22:41:21 +00:00
type UserSettings struct {
2017-01-31 16:46:39 +00:00
PayloadTimeout time.Duration
2016-09-17 22:41:21 +00:00
}