go fmt
This commit is contained in:
166
context.go
166
context.go
@@ -5,152 +5,148 @@
|
||||
package dsrpc
|
||||
|
||||
import (
|
||||
"io"
|
||||
"net"
|
||||
"time"
|
||||
"io"
|
||||
"net"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Context struct {
|
||||
start time.Time
|
||||
remoteHost string
|
||||
sockReader io.Reader
|
||||
sockWriter io.Writer
|
||||
start time.Time
|
||||
remoteHost string
|
||||
sockReader io.Reader
|
||||
sockWriter io.Writer
|
||||
|
||||
reqHeader *Header
|
||||
reqRPC *Request
|
||||
reqPacket *Packet
|
||||
reqHeader *Header
|
||||
reqRPC *Request
|
||||
reqPacket *Packet
|
||||
|
||||
resPacket *Packet
|
||||
resHeader *Header
|
||||
resRPC *Response
|
||||
resPacket *Packet
|
||||
resHeader *Header
|
||||
resRPC *Response
|
||||
|
||||
binReader io.Reader
|
||||
binWriter io.Writer
|
||||
binReader io.Reader
|
||||
binWriter io.Writer
|
||||
}
|
||||
|
||||
|
||||
func NewContext() *Context {
|
||||
context := &Context{}
|
||||
context.start = time.Now()
|
||||
return context
|
||||
context := &Context{}
|
||||
context.start = time.Now()
|
||||
return context
|
||||
}
|
||||
|
||||
func CreateContext(conn net.Conn) *Context {
|
||||
context := &Context{}
|
||||
context.start = time.Now()
|
||||
context.sockReader = conn
|
||||
context.sockWriter = conn
|
||||
context := &Context{}
|
||||
context.start = time.Now()
|
||||
context.sockReader = conn
|
||||
context.sockWriter = conn
|
||||
|
||||
context.reqPacket = NewPacket()
|
||||
context.resPacket = NewPacket()
|
||||
context.reqPacket = NewPacket()
|
||||
context.resPacket = NewPacket()
|
||||
|
||||
context.reqHeader = NewHeader()
|
||||
context.reqRPC = NewRequest()
|
||||
context.reqHeader = NewHeader()
|
||||
context.reqRPC = NewRequest()
|
||||
|
||||
context.resHeader = NewHeader()
|
||||
context.resRPC = NewResponse()
|
||||
context.resRPC = NewResponse()
|
||||
context.resHeader = NewHeader()
|
||||
context.resRPC = NewResponse()
|
||||
context.resRPC = NewResponse()
|
||||
|
||||
return context
|
||||
return context
|
||||
}
|
||||
|
||||
func (context *Context) Request() *Request {
|
||||
return context.reqRPC
|
||||
func (context *Context) Request() *Request {
|
||||
return context.reqRPC
|
||||
}
|
||||
|
||||
func (context *Context) RemoteHost() string {
|
||||
return context.remoteHost
|
||||
return context.remoteHost
|
||||
}
|
||||
|
||||
func (context *Context) Start() time.Time {
|
||||
return context.start
|
||||
return context.start
|
||||
}
|
||||
|
||||
func (context *Context) Method() string {
|
||||
var method string
|
||||
if context.reqRPC != nil {
|
||||
method = context.reqRPC.Method
|
||||
}
|
||||
return method
|
||||
var method string
|
||||
if context.reqRPC != nil {
|
||||
method = context.reqRPC.Method
|
||||
}
|
||||
return method
|
||||
}
|
||||
|
||||
func (context *Context) ReqRpcSize() int64 {
|
||||
var size int64
|
||||
if context.reqHeader != nil {
|
||||
size = context.reqHeader.rpcSize
|
||||
}
|
||||
return size
|
||||
var size int64
|
||||
if context.reqHeader != nil {
|
||||
size = context.reqHeader.rpcSize
|
||||
}
|
||||
return size
|
||||
}
|
||||
|
||||
|
||||
func (context *Context) ReqBinSize() int64 {
|
||||
var size int64
|
||||
if context.reqHeader != nil {
|
||||
size = context.reqHeader.binSize
|
||||
}
|
||||
return size
|
||||
var size int64
|
||||
if context.reqHeader != nil {
|
||||
size = context.reqHeader.binSize
|
||||
}
|
||||
return size
|
||||
}
|
||||
|
||||
func (context *Context) ResBinSize() int64 {
|
||||
var size int64
|
||||
if context.resHeader != nil {
|
||||
size = context.resHeader.binSize
|
||||
}
|
||||
return size
|
||||
var size int64
|
||||
if context.resHeader != nil {
|
||||
size = context.resHeader.binSize
|
||||
}
|
||||
return size
|
||||
}
|
||||
|
||||
func (context *Context) ResRpcSize() int64 {
|
||||
var size int64
|
||||
if context.resHeader != nil {
|
||||
size = context.resHeader.rpcSize
|
||||
}
|
||||
return size
|
||||
var size int64
|
||||
if context.resHeader != nil {
|
||||
size = context.resHeader.rpcSize
|
||||
}
|
||||
return size
|
||||
}
|
||||
|
||||
func (context *Context) ReqSize() int64 {
|
||||
var size int64
|
||||
if context.reqHeader != nil {
|
||||
size += context.reqHeader.binSize
|
||||
size += context.reqHeader.rpcSize
|
||||
}
|
||||
return size
|
||||
var size int64
|
||||
if context.reqHeader != nil {
|
||||
size += context.reqHeader.binSize
|
||||
size += context.reqHeader.rpcSize
|
||||
}
|
||||
return size
|
||||
}
|
||||
|
||||
func (context *Context) ResSize() int64 {
|
||||
var size int64
|
||||
if context.resHeader != nil {
|
||||
size += context.resHeader.binSize
|
||||
size += context.resHeader.rpcSize
|
||||
}
|
||||
return size
|
||||
var size int64
|
||||
if context.resHeader != nil {
|
||||
size += context.resHeader.binSize
|
||||
size += context.resHeader.rpcSize
|
||||
}
|
||||
return size
|
||||
}
|
||||
|
||||
|
||||
|
||||
func (context *Context) SetAuthIdent(ident []byte) {
|
||||
context.reqRPC.Auth.Ident = ident
|
||||
func (context *Context) SetAuthIdent(ident []byte) {
|
||||
context.reqRPC.Auth.Ident = ident
|
||||
}
|
||||
|
||||
func (context *Context) SetAuthSalt(salt []byte) {
|
||||
context.reqRPC.Auth.Salt = salt
|
||||
func (context *Context) SetAuthSalt(salt []byte) {
|
||||
context.reqRPC.Auth.Salt = salt
|
||||
}
|
||||
|
||||
func (context *Context) SetAuthHash(hash []byte) {
|
||||
context.reqRPC.Auth.Hash = hash
|
||||
func (context *Context) SetAuthHash(hash []byte) {
|
||||
context.reqRPC.Auth.Hash = hash
|
||||
}
|
||||
|
||||
func (context *Context) AuthIdent() []byte {
|
||||
return context.reqRPC.Auth.Ident
|
||||
return context.reqRPC.Auth.Ident
|
||||
}
|
||||
|
||||
func (context *Context) AuthSalt() []byte {
|
||||
return context.reqRPC.Auth.Salt
|
||||
return context.reqRPC.Auth.Salt
|
||||
}
|
||||
|
||||
func (context *Context) AuthHash() []byte {
|
||||
return context.reqRPC.Auth.Hash
|
||||
return context.reqRPC.Auth.Hash
|
||||
}
|
||||
|
||||
func (context *Context) Auth() *Auth {
|
||||
return context.reqRPC.Auth
|
||||
return context.reqRPC.Auth
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user