sosedoff/pgweb
1
1
Fork 0
mirror of https://github.com/sosedoff/pgweb.git synced 2024-12-14 19:21:46 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use regular expression for checking some of the connectio errors

Browse Source
This commit is contained in:
Dan Sosedoff 2020-10-18 09:32:34 -05:00
parent e1fc9ddcef
commit f61c1c45d5
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/cli/cli.go
View File

@ -5,6 +5,7 @@ import (
"os"
"os/exec"
"os/signal"
"regexp"
"strings"
"github.com/gin-gonic/gin"
@ -28,6 +29,9 @@ SECURITY WARNING: You are running pgweb in read-only mode.
This mode is designed for environments where users could potentially delete / change data.
For proper read-only access please follow postgresql role management documentation.
------------------------------------------------------`
regexErrConnectionRefused = regexp.MustCompile(`(connection|actively) refused`)
regexErrAuthFailed = regexp.MustCompile(`authentication failed`)
)
func exitWithMessage(message string) {
@ -95,13 +99,13 @@ func initClient() {
}
// Do not bail if local server is not running.
if strings.Contains(msg, "connection refused") {
if regexErrConnectionRefused.MatchString(msg) {
fmt.Println("Error:", msg)
return
}
// Do not bail if local auth is invalid
if strings.Contains(msg, "authentication failed") {
if regexErrAuthFailed.MatchString(msg) {
fmt.Println("Error:", msg)
return
}