62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"git.usbharu.dev/usbharu/unos/illust/data"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/reflection"
|
|
"gopkg.in/ini.v1"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/signal"
|
|
)
|
|
|
|
func main() {
|
|
load, err := ini.Load("config.ini")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
port := load.Section("gRPC").Key("port").MustInt(50051)
|
|
mongoUrl := load.Section("mongodb").Key("url").MustString("mongodb://localhost:20071")
|
|
|
|
listen, err := net.Listen("tcp", fmt.Sprintf(":%d", port))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
server := grpc.NewServer()
|
|
|
|
connect, err := mongo.Connect(context.Background(), options.Client().ApplyURI(mongoUrl))
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
database := connect.Database("")
|
|
|
|
data.RegisterIllustDataServiceServer(server, data.NewIllustDataServiceImpl(*database))
|
|
|
|
reflection.Register(server)
|
|
|
|
go func() {
|
|
log.Printf("start gRPC server port: %v", port)
|
|
server.Serve(listen)
|
|
}()
|
|
|
|
defer func() {
|
|
if err := connect.Disconnect(context.Background()); err != nil {
|
|
panic(err)
|
|
}
|
|
}()
|
|
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, os.Interrupt)
|
|
<-quit
|
|
|
|
server.GracefulStop()
|
|
}
|