foomo/gotsrpc
1
0
Fork 0
mirror of https://github.com/foomo/gotsrpc.git synced 2025-10-16 12:35:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

made gorpc client public

Browse Source
This commit is contained in:
franklin 2017-03-20 13:39:26 +01:00
parent c78f00ba6c
commit 99830b9e50
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
go.go
View File

@ -632,7 +632,7 @@ func renderGoRPCServiceClients(services map[string]*Service, fullPackageName str
// Client type
g.l(`
type ` + clientName + ` struct {
client *gorpc.Client
Client *gorpc.Client
}
`)
// Constructor
@ -640,20 +640,19 @@ func renderGoRPCServiceClients(services map[string]*Service, fullPackageName str
func New` + clientName + `(addr string, tlsConfig *tls.Config) *` + clientName + ` {
client := &` + clientName + `{}
if tlsConfig == nil {
client.client = gorpc.NewTCPClient(addr)
client.Client = gorpc.NewTCPClient(addr)
} else {
client.client = gorpc.NewTLSClient(addr, tlsConfig)
client.Client = gorpc.NewTLSClient(addr, tlsConfig)
}
client.Start()
return client
}
func (c *` + clientName + `) Start() {
c.client.Start()
c.Client.Start()
}
func (c *` + clientName + `) Stop() {
c.client.Stop()
c.Client.Stop()
}
`)
g.nl()
@ -679,9 +678,9 @@ func renderGoRPCServiceClients(services map[string]*Service, fullPackageName str
g.l(`func (c *` + clientName + `) ` + method.Name + `(` + strings.Join(params, ", ") + `) (` + strings.Join(returns, ", ") + `) {`)
g.l(`req := ` + service.Name + method.Name + `Request{` + strings.Join(args, ", ") + `}`)
if len(rets) > 0 {
g.l(`rpcCallRes, rpcCallErr := c.client.Call(req)`)
g.l(`rpcCallRes, rpcCallErr := c.Client.Call(req)`)
} else {
g.l(`_, rpcCallErr := c.client.Call(req)`)
g.l(`_, rpcCallErr := c.Client.Call(req)`)
}
g.l(`if rpcCallErr != nil {`)
g.l(`clientErr = rpcCallErr`)