diff --git a/internal/database/db.go b/internal/database/db.go index 1769dc0..6f1e6fc 100644 --- a/internal/database/db.go +++ b/internal/database/db.go @@ -32,6 +32,15 @@ func ConnectDB(dbConfig *redis.Options) *redis.Client { return localDb } +// Ping test connection to Redis database +func Ping(db *redis.Client) error { + status := db.Ping(ctx) + if status.String() != "ping: PONG" { + return &PingError{} + } + return nil +} + func (config DBConfig) InsertPaste(key string, content string, secret string, ttl time.Duration) { type dbSchema struct { content string diff --git a/internal/database/error.go b/internal/database/error.go new file mode 100644 index 0000000..3985065 --- /dev/null +++ b/internal/database/error.go @@ -0,0 +1,7 @@ +package database + +type PingError struct{} + +func (m *PingError) Error() string { + return "Connection to redis not work" +} diff --git a/main.go b/main.go index df25d23..8af2291 100644 --- a/main.go +++ b/main.go @@ -2,6 +2,7 @@ package main import ( "embed" + "log" "git.gnous.eu/gnouseu/plakken/internal/config" "git.gnous.eu/gnouseu/plakken/internal/database" @@ -19,6 +20,10 @@ func main() { initConfig := config.GetConfig() dbConfig := database.InitDB(initConfig.RedisAddress, initConfig.RedisUser, initConfig.RedisPassword, initConfig.RedisDB) db := database.ConnectDB(dbConfig) + err := database.Ping(db) + if err != nil { + log.Fatal(err) + } serverConfig := httpServer.ServerConfig{ HTTPServer: httpServer.Config(initConfig.ListenAddress),