Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 10 additions & 10 deletions main.go
Original file line number Diff line number Diff line change
Expand Up @@ -386,16 +386,6 @@ func main() {
// Set DB reference in S3Manager for lazy client initialization
GetS3Manager().SetDB(db)

// Initialize the schema
if err = initializeSchema(db); err != nil {
log.Fatal().Err(err).Msg("Failed to initialize schema")
// Perform cleanup before exiting
if err := db.Close(); err != nil {
log.Error().Err(err).Msg("Failed to close database connection during cleanup")
}
os.Exit(1)
}

var dbLog waLog.Logger
if *waDebug != "" {
dbLog = waLog.Stdout("Database", *waDebug, *colorOutput)
Expand All @@ -420,6 +410,16 @@ func main() {
os.Exit(1)
}

// Initialize the schema
if err = initializeSchema(db); err != nil {
log.Fatal().Err(err).Msg("Failed to initialize schema")
// Perform cleanup before exiting
if err := db.Close(); err != nil {
log.Error().Err(err).Msg("Failed to close database connection during cleanup")
}
os.Exit(1)
}

serverMode := HTTP
if *mode == "stdio" {
serverMode = Stdio
Expand Down
9 changes: 6 additions & 3 deletions migrations.go
Original file line number Diff line number Diff line change
Expand Up @@ -220,9 +220,12 @@ END $$;
`

const addWhatsmeowMessageSecretsMessageIDIndexSQL = `
CREATE INDEX CONCURRENTLY IF NOT EXISTS whatsmeow_message_secrets_message_id_idx
ON public.whatsmeow_message_secrets (message_id);
-- PostgreSQL version
DO $$
BEGIN
CREATE INDEX IF NOT EXISTS whatsmeow_message_secrets_message_id_idx
ON whatsmeow_message_secrets (message_id);
END $$;
-- SQLite version (handled in code)
`

Expand Down
Loading