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.
36 lines
742 B
36 lines
742 B
package main |
|
|
|
import ( |
|
clientv3 "go.etcd.io/etcd/client/v3" |
|
"google.golang.org/grpc/grpclog" |
|
"sonet/internal/postal/logic" |
|
"sonet/pkg/config" |
|
"sonet/pkg/grpc/discovery" |
|
"sonet/pkg/utils/logger" |
|
"sonet/pkg/utils/shutdown" |
|
) |
|
|
|
// websocket server with postalService |
|
func main() { |
|
grpclog.SetLoggerV2(logger.Logger) |
|
|
|
conf := config.LoadConfig(nil, "cmd/gateway_ws") |
|
postalServer := logic.NewPostalServer() |
|
|
|
// registry and run... |
|
etcdClient, err := clientv3.New(conf.Etcd) |
|
if err != nil { |
|
panic(err) |
|
} |
|
postalRegistry := discovery.NewRegister(etcdClient) |
|
shutdown.AddShutdownHook(postalRegistry.Stop) |
|
|
|
go func() { |
|
err = postalServer.Run(conf.Grpc, postalRegistry) |
|
if err != nil { |
|
panic(err) |
|
} |
|
}() |
|
|
|
shutdown.Await() |
|
}
|
|
|