mirror of https://github.com/tidwall/tile38.git
84 lines
1.4 KiB
Go
84 lines
1.4 KiB
Go
|
package endpoint
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"golang.org/x/net/context"
|
||
|
|
|
||
|
|
"github.com/tidwall/tile38/hservice"
|
||
|
|
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
grpcExpiresAfter = time.Second * 30
|
||
|
|
)
|
||
|
|
|
||
|
|
type GRPCEndpointConn struct {
|
||
|
|
mu sync.Mutex
|
||
|
|
ep Endpoint
|
||
|
|
ex bool
|
||
|
|
t time.Time
|
||
|
|
conn *grpc.ClientConn
|
||
|
|
sconn hservice.HookServiceClient
|
||
|
|
}
|
||
|
|
|
||
|
|
func newGRPCEndpointConn(ep Endpoint) *GRPCEndpointConn {
|
||
|
|
return &GRPCEndpointConn{
|
||
|
|
ep: ep,
|
||
|
|
t: time.Now(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (conn *GRPCEndpointConn) Expired() bool {
|
||
|
|
conn.mu.Lock()
|
||
|
|
defer conn.mu.Unlock()
|
||
|
|
if !conn.ex {
|
||
|
|
if time.Now().Sub(conn.t) > grpcExpiresAfter {
|
||
|
|
if conn.conn != nil {
|
||
|
|
conn.close()
|
||
|
|
}
|
||
|
|
conn.ex = true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return conn.ex
|
||
|
|
}
|
||
|
|
func (conn *GRPCEndpointConn) close() {
|
||
|
|
if conn.conn != nil {
|
||
|
|
conn.conn.Close()
|
||
|
|
conn.conn = nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (conn *GRPCEndpointConn) Send(msg string) error {
|
||
|
|
conn.mu.Lock()
|
||
|
|
defer conn.mu.Unlock()
|
||
|
|
if conn.ex {
|
||
|
|
return errors.New("expired")
|
||
|
|
}
|
||
|
|
conn.t = time.Now()
|
||
|
|
if conn.conn == nil {
|
||
|
|
addr := fmt.Sprintf("%s:%d", conn.ep.GRPC.Host, conn.ep.GRPC.Port)
|
||
|
|
var err error
|
||
|
|
conn.conn, err = grpc.Dial(addr, grpc.WithInsecure())
|
||
|
|
if err != nil {
|
||
|
|
conn.close()
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
conn.sconn = hservice.NewHookServiceClient(conn.conn)
|
||
|
|
}
|
||
|
|
r, err := conn.sconn.Send(context.Background(), &hservice.MessageRequest{msg})
|
||
|
|
if err != nil {
|
||
|
|
conn.close()
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if !r.Ok {
|
||
|
|
conn.close()
|
||
|
|
return errors.New("invalid grpc reply")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|