package store import ( "github.com/rs/zerolog" "github.com/rs/zerolog/log" "go.uber.org/fx" "gorm.io/gorm" "sneak.berlin/go/directory/internal/config" "sneak.berlin/go/directory/internal/database" "sneak.berlin/go/directory/internal/logger" ) const ( UsersTableName string = "users" UserAttributesTableName string = "userattributes" ) type StoreParams struct { fx.In Logger *logger.Logger Config *config.Config Database *database.Database } type Store struct { db *gorm.DB logger *logger.Logger log *zerolog.Logger params *StoreParams } func New(lc fx.Lifecycle, params *StoreParams) (*Store, error) { s := new(Store) s.logger = params.Logger s.log = params.Logger.Get() var err error s.db, err = params.Database.Get() if err != nil { log.Error().Err(err).Msg("failed to get database object") return nil, err } // Auto migrate the schema err = s.db.AutoMigrate(&User{}, &UserAttribute{}) if err != nil { log.Error().Err(err).Msg("failed to migrate database schema") return nil, err } return s, nil } func (s *Store) Close() { log.Info().Msg("closing store db") db, err := s.db.DB() if err != nil { log.Error().Err(err).Msg("failed to get generic database object from gorm") return } db.Close() }