rookie-ninja / rk-db Goto Github PK

postgres:
  - name: user
    enabled: true
    domain: "*"
    addr: "192.168.8.104:5432"
    user: postgres
    pass: 12345678
    database:
      - name: webchat
        autoCreate: true

package main

import (
    "context"
    "fmt"
    "log"
    "os"
    "time"

    greeter "github.com/chater-co/web-chat/api/gen/v1"
    rkboot "github.com/rookie-ninja/rk-boot"
    rkpostgres "github.com/rookie-ninja/rk-db/postgres"
    rkgrpc "github.com/rookie-ninja/rk-grpc/boot"
    "google.golang.org/grpc"
    "google.golang.org/grpc/grpclog"
    "gorm.io/gorm"
)

var userDb *gorm.DB

func init() {
    grpclog.SetLoggerV2(grpclog.NewLoggerV2(os.Stdout, os.Stdout, os.Stdout))
}

func main() {
    boot := rkboot.NewBoot()
    boot.Bootstrap(context.TODO())

    pgEntry := rkpostgres.GetPostgresEntry("user")
    if pgEntry == nil {
        log.Fatalf("failed to get Postgres entry")
    }
    userDb = pgEntry.GetDB("webchat")
    if !userDb.DryRun {
        userDb.AutoMigrate(&User{})
    }

    grpcEntry := rkgrpc.GetGrpcEntry("chat-app")
    grpcEntry.AddRegFuncGrpc(registerGreeter)
    grpcEntry.AddRegFuncGw(greeter.RegisterGreeterHandlerFromEndpoint)
    boot.Bootstrap(context.Background())
    boot.WaitForShutdownSig(context.TODO())
}

func registerGreeter(server *grpc.Server) {
    greeter.RegisterGreeterServer(server, &GreeterServer{})
}

type GreeterServer struct{}

type Base struct {
    CreatedAt time.Time      `yaml:"-" json:"-"`
    UpdatedAt time.Time      `yaml:"-" json:"-"`
    DeletedAt gorm.DeletedAt `yaml:"-" json:"-" gorm:"index"`
}

type User struct {
    Base
    Id   int    `yaml:"id" json:"id" gorm:"primaryKey"`
    Name string `yaml:"name" json:"name"`
}

func (server *GreeterServer) Greeter(ctx context.Context, request *greeter.GreeterRequest) (*greeter.GreeterResponse, error) {
    return &greeter.GreeterResponse{
        Message: fmt.Sprintf("Hello %s!", request.Name),
    }, nil
}