mirror of
https://github.com/foomo/gotsrpc.git
synced 2025-10-16 12:35:35 +00:00
102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
// Code generated by gotsrpc https://github.com/foomo/gotsrpc/v2 - DO NOT EDIT.
|
|
|
|
package service
|
|
|
|
import (
|
|
io "io"
|
|
ioutil "io/ioutil"
|
|
http "net/http"
|
|
time "time"
|
|
|
|
gotsrpc "github.com/foomo/gotsrpc/v2"
|
|
)
|
|
|
|
const (
|
|
ServiceGoTSRPCProxyTime = "Time"
|
|
ServiceGoTSRPCProxyTimeStruct = "TimeStruct"
|
|
)
|
|
|
|
type ServiceGoTSRPCProxy struct {
|
|
EndPoint string
|
|
service Service
|
|
}
|
|
|
|
func NewDefaultServiceGoTSRPCProxy(service Service) *ServiceGoTSRPCProxy {
|
|
return NewServiceGoTSRPCProxy(service, "/service")
|
|
}
|
|
|
|
func NewServiceGoTSRPCProxy(service Service, endpoint string) *ServiceGoTSRPCProxy {
|
|
return &ServiceGoTSRPCProxy{
|
|
EndPoint: endpoint,
|
|
service: service,
|
|
}
|
|
}
|
|
|
|
// ServeHTTP exposes your service
|
|
func (p *ServiceGoTSRPCProxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == http.MethodOptions {
|
|
return
|
|
} else if r.Method != http.MethodPost {
|
|
gotsrpc.ErrorMethodNotAllowed(w)
|
|
return
|
|
}
|
|
defer io.Copy(ioutil.Discard, r.Body) // Drain Request Body
|
|
|
|
funcName := gotsrpc.GetCalledFunc(r, p.EndPoint)
|
|
callStats, _ := gotsrpc.GetStatsForRequest(r)
|
|
callStats.Func = funcName
|
|
callStats.Package = "github.com/foomo/gotsrpc/v2/example/time/service"
|
|
callStats.Service = "Service"
|
|
switch funcName {
|
|
case ServiceGoTSRPCProxyTime:
|
|
var (
|
|
args []interface{}
|
|
rets []interface{}
|
|
)
|
|
var (
|
|
arg_v time.Time
|
|
)
|
|
args = []interface{}{&arg_v}
|
|
if err := gotsrpc.LoadArgs(&args, callStats, r); err != nil {
|
|
gotsrpc.ErrorCouldNotLoadArgs(w)
|
|
return
|
|
}
|
|
executionStart := time.Now()
|
|
timeRet := p.service.Time(arg_v)
|
|
callStats.Execution = time.Since(executionStart)
|
|
rets = []interface{}{timeRet}
|
|
if err := gotsrpc.Reply(rets, callStats, r, w); err != nil {
|
|
gotsrpc.ErrorCouldNotReply(w)
|
|
return
|
|
}
|
|
gotsrpc.Monitor(w, r, args, rets, callStats)
|
|
return
|
|
case ServiceGoTSRPCProxyTimeStruct:
|
|
var (
|
|
args []interface{}
|
|
rets []interface{}
|
|
)
|
|
var (
|
|
arg_v TimeStruct
|
|
)
|
|
args = []interface{}{&arg_v}
|
|
if err := gotsrpc.LoadArgs(&args, callStats, r); err != nil {
|
|
gotsrpc.ErrorCouldNotLoadArgs(w)
|
|
return
|
|
}
|
|
executionStart := time.Now()
|
|
timeStructRet := p.service.TimeStruct(arg_v)
|
|
callStats.Execution = time.Since(executionStart)
|
|
rets = []interface{}{timeStructRet}
|
|
if err := gotsrpc.Reply(rets, callStats, r, w); err != nil {
|
|
gotsrpc.ErrorCouldNotReply(w)
|
|
return
|
|
}
|
|
gotsrpc.Monitor(w, r, args, rets, callStats)
|
|
return
|
|
default:
|
|
gotsrpc.ClearStats(r)
|
|
gotsrpc.ErrorFuncNotFound(w)
|
|
}
|
|
}
|