PMM-11341 Include permissions in database query. (#162)

JiriCtvrtka · web-flow · commit cc475843a13a · 2023-11-16T11:33:37.000+01:00
diff --git a/cmd/postgres_exporter/queries.go b/cmd/postgres_exporter/queries.go
@@ -277,7 +277,7 @@ func addQueries(content []byte, pgVersion semver.Version, server *Server) error
 }
 
 func queryDatabases(server *Server) ([]string, error) {
-	rows, err := server.db.Query("SELECT datname FROM pg_database WHERE datallowconn = true AND datistemplate = false AND datname != current_database()")
+	rows, err := server.db.Query("SELECT datname FROM pg_database WHERE datallowconn = true AND datistemplate = false AND datname != current_database() AND has_database_privilege(current_user, datname, 'connect')")
 	if err != nil {
 		return nil, fmt.Errorf("Error retrieving databases: %v", err)
 	}

Original file line number	Diff line number	Diff line change
`@@ -277,7 +277,7 @@ func addQueries(content []byte, pgVersion semver.Version, server *Server) error`
`277`	`277`	`}`
`278`	`278`
`279`	`279`	`func queryDatabases(server *Server) ([]string, error) {`
`280`		`- rows, err := server.db.Query("SELECT datname FROM pg_database WHERE datallowconn = true AND datistemplate = false AND datname != current_database()")`
	`280`	`+ rows, err := server.db.Query("SELECT datname FROM pg_database WHERE datallowconn = true AND datistemplate = false AND datname != current_database() AND has_database_privilege(current_user, datname, 'connect')")`
`281`	`281`	`if err != nil {`
`282`	`282`	`return nil, fmt.Errorf("Error retrieving databases: %v", err)`
`283`	`283`	`}`