From 827905820b34b1851ba5e0956e67e892d51c8d21 Mon Sep 17 00:00:00 2001 From: haohongfan <haohongfan@ling.ai> Date: Mon, 14 Sep 2020 23:29:34 +0800 Subject: [PATCH] feat: add some comment --- remoting/getty/config.go | 18 ++++++++---------- remoting/getty/readwriter.go | 12 +++++++++--- 2 files changed, 17 insertions(+), 13 deletions(-) diff --git a/remoting/getty/config.go b/remoting/getty/config.go index 4213a8705..dcf59d082 100644 --- a/remoting/getty/config.go +++ b/remoting/getty/config.go @@ -30,7 +30,7 @@ import ( ) type ( - // GettySessionParam ... + // GettySessionParam is session configuration for getty GettySessionParam struct { CompressEncoding bool `default:"false" yaml:"compress_encoding" json:"compress_encoding,omitempty"` TcpNoDelay bool `default:"true" yaml:"tcp_no_delay" json:"tcp_no_delay,omitempty"` @@ -50,8 +50,7 @@ type ( SessionName string `default:"rpc" yaml:"session_name" json:"session_name,omitempty"` } - // ServerConfig - //Config holds supported types by the multiconfig package + // ServerConfig holds supported types by the multiconfig package ServerConfig struct { SSLEnabled bool @@ -69,8 +68,7 @@ type ( GettySessionParam GettySessionParam `required:"true" yaml:"getty_session_param" json:"getty_session_param,omitempty"` } - // ClientConfig - //Config holds supported types by the multiconfig package + // ClientConfig holds supported types by the multiconfig package ClientConfig struct { ReconnectInterval int `default:"0" yaml:"reconnect_interval" json:"reconnect_interval,omitempty"` @@ -99,7 +97,7 @@ type ( } ) -// GetDefaultClientConfig ... +// GetDefaultClientConfig gets client default configuration func GetDefaultClientConfig() ClientConfig { return ClientConfig{ ReconnectInterval: 0, @@ -127,7 +125,7 @@ func GetDefaultClientConfig() ClientConfig { }} } -// GetDefaultServerConfig ... +// GetDefaultServerConfig gets server default configuration func GetDefaultServerConfig() ServerConfig { return ServerConfig{ SessionTimeout: "180s", @@ -152,7 +150,7 @@ func GetDefaultServerConfig() ServerConfig { } } -// CheckValidity ... +// CheckValidity confirm getty sessian params func (c *GettySessionParam) CheckValidity() error { var err error @@ -175,7 +173,7 @@ func (c *GettySessionParam) CheckValidity() error { return nil } -// CheckValidity ... +// CheckValidity confirm client params. func (c *ClientConfig) CheckValidity() error { var err error @@ -197,7 +195,7 @@ func (c *ClientConfig) CheckValidity() error { return perrors.WithStack(c.GettySessionParam.CheckValidity()) } -// CheckValidity ... +// CheckValidity confirm server params func (c *ServerConfig) CheckValidity() error { var err error diff --git a/remoting/getty/readwriter.go b/remoting/getty/readwriter.go index 26623edd0..c6585c2dc 100644 --- a/remoting/getty/readwriter.go +++ b/remoting/getty/readwriter.go @@ -36,16 +36,18 @@ import ( // RpcClientPackageHandler //////////////////////////////////////////// -// RpcClientPackageHandler ... +// RpcClientPackageHandler Read data from server and Write data to server type RpcClientPackageHandler struct { client *Client } -// NewRpcClientPackageHandler ... +// NewRpcClientPackageHandler create a RpcClientPackageHandler func NewRpcClientPackageHandler(client *Client) *RpcClientPackageHandler { return &RpcClientPackageHandler{client: client} } +// Read data from server. if the package size from server is larger than 4096 byte, server will read 4096 byte +// and send to client each time. the Read can assemble it. func (p *RpcClientPackageHandler) Read(ss getty.Session, data []byte) (interface{}, int, error) { resp, length, err := (p.client.codec).Decode(data) //err := pkg.Unmarshal(buf, p.client) @@ -63,6 +65,7 @@ func (p *RpcClientPackageHandler) Read(ss getty.Session, data []byte) (interface return resp, length, nil } +// Write send the data to server func (p *RpcClientPackageHandler) Write(ss getty.Session, pkg interface{}) ([]byte, error) { req, ok := pkg.(*remoting.Request) if !ok { @@ -87,7 +90,7 @@ func (p *RpcClientPackageHandler) Write(ss getty.Session, pkg interface{}) ([]by // rpcServerPkgHandler = &RpcServerPackageHandler{} //) -// RpcServerPackageHandler ... +// RpcServerPackageHandler Read data from client and Write data to client type RpcServerPackageHandler struct { server *Server } @@ -96,6 +99,8 @@ func NewRpcServerPackageHandler(server *Server) *RpcServerPackageHandler { return &RpcServerPackageHandler{server: server} } +// Read data from client. if the package size from client is larger than 4096 byte, client will read 4096 byte +// and send to client each time. the Read can assemble it. func (p *RpcServerPackageHandler) Read(ss getty.Session, data []byte) (interface{}, int, error) { req, length, err := (p.server.codec).Decode(data) //resp,len, err := (*p.).DecodeResponse(buf) @@ -113,6 +118,7 @@ func (p *RpcServerPackageHandler) Read(ss getty.Session, data []byte) (interface return req, length, err } +// Write send the data to client func (p *RpcServerPackageHandler) Write(ss getty.Session, pkg interface{}) ([]byte, error) { res, ok := pkg.(*remoting.Response) if !ok { -- GitLab