You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
1.4 KiB
72 lines
1.4 KiB
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/signal"
|
|
"strconv"
|
|
"syscall"
|
|
"test3k/authDB/internal/config"
|
|
server "test3k/authDB/internal/transport/grpc"
|
|
api "test3k/authDB/pkg/api"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
func main() {
|
|
config := config.NewConfig()
|
|
ctx, _ := context.WithCancel(context.Background())
|
|
srv := server.NewServer(config)
|
|
s := grpc.NewServer()
|
|
|
|
//
|
|
signalChannel := make(chan os.Signal, 1)
|
|
signal.Notify(signalChannel, syscall.SIGINT)
|
|
signal.Notify(signalChannel, syscall.SIGTERM)
|
|
defer stop(signalChannel, s, srv)
|
|
|
|
// Запуск сервера
|
|
go start(config, s, srv)
|
|
|
|
//
|
|
for {
|
|
select {
|
|
case <-signalChannel:
|
|
return
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
// Остановка сервера
|
|
func stop(signalChannel chan os.Signal, s *grpc.Server, srv *server.AuthDBServer) {
|
|
defer s.GracefulStop()
|
|
defer srv.GracefulStop()
|
|
defer signal.Stop(signalChannel)
|
|
|
|
log.Println("authDBServer stopping ...")
|
|
}
|
|
|
|
// Запуск сервера
|
|
func start(config *config.Config, s *grpc.Server, srv *server.AuthDBServer) {
|
|
api.RegisterAuthDBServer(s, srv)
|
|
|
|
//
|
|
connStr := net.JoinHostPort("", strconv.Itoa(config.App.Port))
|
|
lis, era := net.Listen("tcp", connStr)
|
|
if era != nil {
|
|
log.Fatal(era)
|
|
}
|
|
|
|
//
|
|
log.Printf("authDBServer starting (%s)\n", connStr)
|
|
|
|
//
|
|
eru := s.Serve(lis)
|
|
if eru != nil {
|
|
log.Fatal("Failed starting server")
|
|
}
|
|
}
|
|
|