grpc basic done
parent
496539df0f
commit
f7776c5782
GPG Key ID:
476C9A2B4B794EBB
@ -0,0 +1,18 @@
|
||||
package health
|
||||
|
||||
import (
|
||||
"github.com/ajikamaludin/go-grpc_basic/configs"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
config *configs.Configs
|
||||
logger *logrus.Logger
|
||||
}
|
||||
|
||||
func New(config *configs.Configs, logger *logrus.Logger) *Server {
|
||||
return &Server{
|
||||
config: config,
|
||||
logger: logger,
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,17 @@
|
||||
package health
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/ajikamaludin/go-grpc_basic/pkg/v1/utils/constants"
|
||||
hlpb "github.com/ajikamaludin/go-grpc_basic/proto/v1/health"
|
||||
"github.com/golang/protobuf/ptypes/empty"
|
||||
)
|
||||
|
||||
func (s *Server) Status(ctx context.Context, req *empty.Empty) (*hlpb.Response, error) {
|
||||
return &hlpb.Response{
|
||||
Success: true,
|
||||
Code: constants.SuccessCode,
|
||||
Desc: constants.SuccesDesc,
|
||||
}, nil
|
||||
}
|
||||
@ -0,0 +1,10 @@
|
||||
package constants
|
||||
|
||||
const (
|
||||
EnvProduction = "production"
|
||||
)
|
||||
|
||||
const (
|
||||
SuccessCode = `0000`
|
||||
SuccesDesc = `SUCCESS`
|
||||
)
|
||||
@ -0,0 +1,36 @@
|
||||
package router
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"net"
|
||||
|
||||
"github.com/ajikamaludin/go-grpc_basic/api/v1/health"
|
||||
"github.com/ajikamaludin/go-grpc_basic/configs"
|
||||
hlpb "github.com/ajikamaludin/go-grpc_basic/proto/v1/health"
|
||||
"github.com/sirupsen/logrus"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/reflection"
|
||||
)
|
||||
|
||||
func NewGRPCServer(configs *configs.Configs, logger *logrus.Logger) error {
|
||||
listen, err := net.Listen("tcp", fmt.Sprintf(":%v", configs.Config.Server.Grpc.Port))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// register grpc service server
|
||||
grpcServer := grpc.NewServer()
|
||||
|
||||
hlpb.RegisterHealthServiceServer(grpcServer, health.New(configs, logger))
|
||||
|
||||
// add reflection service
|
||||
reflection.Register(grpcServer)
|
||||
|
||||
// running gRPC server
|
||||
log.Println("[SERVER] GRPC server is running")
|
||||
|
||||
grpcServer.Serve(listen)
|
||||
|
||||
return nil
|
||||
}
|
||||
@ -0,0 +1,21 @@
|
||||
package router
|
||||
|
||||
import (
|
||||
"net"
|
||||
"net/http"
|
||||
|
||||
"github.com/soheilhy/cmux"
|
||||
)
|
||||
|
||||
func IgnoreErr(err error) bool {
|
||||
switch {
|
||||
case err == nil || err == http.ErrServerClosed || err == cmux.ErrListenerClosed:
|
||||
return true
|
||||
}
|
||||
|
||||
if opErr, ok := err.(*net.OpError); ok {
|
||||
return opErr.Err.Error() == "use of closed network connection"
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
Loading…
Reference in New Issue