mirror of
https://github.com/kevin-DL/services.git
synced 2026-01-11 19:04:35 +00:00
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
pb "github.com/micro/services/db/proto"
|
|
"github.com/micro/services/pkg/tracing"
|
|
|
|
"github.com/micro/services/db/handler"
|
|
|
|
"github.com/micro/micro/v3/service"
|
|
"github.com/micro/micro/v3/service/logger"
|
|
|
|
"database/sql"
|
|
|
|
"github.com/micro/micro/v3/service/config"
|
|
|
|
_ "github.com/jackc/pgx/v4/stdlib"
|
|
)
|
|
|
|
var dbAddress = "postgresql://postgres:postgres@localhost:5432/db?sslmode=disable"
|
|
|
|
func main() {
|
|
// Create service
|
|
srv := service.New(
|
|
service.Name("db"),
|
|
service.Version("latest"),
|
|
)
|
|
|
|
// Connect to the database
|
|
cfg, err := config.Get("micro.db.database")
|
|
if err != nil {
|
|
logger.Fatalf("Error loading config: %v", err)
|
|
}
|
|
addr := cfg.String(dbAddress)
|
|
sqlDB, err := sql.Open("pgx", addr)
|
|
if err != nil {
|
|
logger.Fatalf("Failed to open connection to DB %s", err)
|
|
}
|
|
h := &handler.Db{}
|
|
h.DBConn(sqlDB)
|
|
|
|
// Register handler
|
|
pb.RegisterDbHandler(srv.Server(), h)
|
|
|
|
// Register handler
|
|
pb.RegisterDbHandler(srv.Server(), &handler.Db{})
|
|
|
|
traceCloser := tracing.SetupOpentracing("db")
|
|
defer traceCloser.Close()
|
|
// Run service
|
|
if err := srv.Run(); err != nil {
|
|
logger.Fatal(err)
|
|
}
|
|
}
|