wip(api): use postgres

Author: heiytor

api/server.go

@@ -8,8 +8,8 @@ import (
 	"github.com/labstack/echo/v4"
 	"github.com/shellhub-io/shellhub/api/routes"
 	"github.com/shellhub-io/shellhub/api/services"
-	"github.com/shellhub-io/shellhub/api/store/mongo"
-	"github.com/shellhub-io/shellhub/api/store/mongo/options"
+	"github.com/shellhub-io/shellhub/api/store/pg"
+	"github.com/shellhub-io/shellhub/api/store/pg/options"
 	"github.com/shellhub-io/shellhub/pkg/api/internalclient"
 	"github.com/shellhub-io/shellhub/pkg/cache"
 	"github.com/shellhub-io/shellhub/pkg/geoip/geolite2"
@@ -83,14 +83,13 @@ func (s *Server) Setup(ctx context.Context) error {
 
 	log.Debug("Redis cache initialized successfully")
 
-	store, err := mongo.NewStore(ctx, s.env.MongoURI, cache, options.RunMigatrions)
+	uri := pg.URI(s.env.PostgresHost, s.env.PostgresPort, s.env.PostgresUser, s.env.PostgresPassword, s.env.PostgresDB)
+	store, err := pg.New(ctx, uri, options.Migrate())
 	if err != nil {
-		log.
-			WithError(err).
-			Fatal("failed to create the store")
+		return err
 	}
 
-	log.Debug("MongoDB store connected successfully")
+	log.Debug("PostgreSQL store connected successfully")
 
 	apiClient, err := internalclient.NewClient(internalclient.WithAsynqWorker(s.env.RedisURI))
 	if err != nil {

api/store/pg/pg.go

@@ -14,7 +14,7 @@ import (
 )
 
 type pg struct {
-	db *bun.DB
+	driver *bun.DB
 }
 
 func URI(host, port, user, password, db string) string {
@@ -34,10 +34,16 @@ func New(ctx context.Context, uri string, opts ...options.Option) (store.Store,
 		return nil, err
 	}
 
-	pg := &pg{db: bun.NewDB(stdlib.OpenDBFromPool(pool), pgdialect.New())}
-	if err := pg.db.Ping(); err != nil {
+	pg := &pg{driver: bun.NewDB(stdlib.OpenDBFromPool(pool), pgdialect.New())}
+	if err := pg.driver.Ping(); err != nil {
 		return nil, err
 	}
 
+	for _, opt := range opts {
+		if err := opt(ctx, pg.driver.DB); err != nil {
+			return nil, err
+		}
+	}
+
 	return pg, nil
 }