Refactor seen service to use database

This commit is contained in:
Ben Toogood
2021-02-04 11:18:33 +00:00
parent b02d4dacb5
commit 6ca55ba18d
6 changed files with 64 additions and 358 deletions

View File

@@ -1,15 +1,18 @@
package main
import (
"github.com/micro/services/seen/domain"
"github.com/micro/services/seen/handler"
pb "github.com/micro/services/seen/proto"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"github.com/micro/micro/v3/service"
"github.com/micro/micro/v3/service/config"
"github.com/micro/micro/v3/service/logger"
"github.com/micro/micro/v3/service/store"
)
var dbAddress = "postgresql://postgres@localhost:5432/seen?sslmode=disable"
func main() {
// Create service
srv := service.New(
@@ -17,10 +20,22 @@ func main() {
service.Version("latest"),
)
// Connect to the database
cfg, err := config.Get("seen.database")
if err != nil {
logger.Fatalf("Error loading config: %v", err)
}
addr := cfg.String(dbAddress)
db, err := gorm.Open(postgres.Open(addr), &gorm.Config{})
if err != nil {
logger.Fatalf("Error connecting to database: %v", err)
}
if err := db.AutoMigrate(&handler.SeenInstance{}); err != nil {
logger.Fatalf("Error migrating database: %v", err)
}
// Register handler
pb.RegisterSeenHandler(srv.Server(), &handler.Seen{
Domain: domain.New(store.DefaultStore),
})
pb.RegisterSeenHandler(srv.Server(), &handler.Seen{DB: db})
// Run service
if err := srv.Run(); err != nil {