Method acquireFileDB

server/server.go:777–809

acquireFileDB returns a shared *sql.DB for the given user, creating one if needed. The caller must call releaseFileDB when the connection is no longer needed.

(username string, passthrough bool)

}

// initConnsMap initializes the connection registry map.
// This is a separate method to work around cases where a local variable
// named "clientConn" shadows the type name (e.g., in worker.go).
func (s *Server) initConnsMap() {
	s.conns = make(map[int32]*clientConn)
	if s.recentErrors == nil {
		s.recentErrors = newRecentErrorRing(0)
	}
}

// registerConn adds a client connection to the registry for pg_stat_activity.
func (s *Server) registerConn(c *clientConn) {
	s.connsMu.Lock()
	s.conns[c.pid] = c
	s.connsMu.Unlock()
}

// unregisterConn removes a client connection from the registry.
func (s *Server) unregisterConn(pid int32) {
	s.connsMu.Lock()
	delete(s.conns, pid)
	s.connsMu.Unlock()
}

// DrainOrgConnections requests a clean close of every PostgreSQL connection
// for orgID at its next idle protocol boundary. Connections already blocked
// waiting for client input are woken immediately; executing queries are not
// cancelled and close after their next ReadyForQuery.
func (s *Server) DrainOrgConnections(orgID string) int {
	return s.drainConnections(func(c *clientConn) bool {
		return c.orgID == orgID
	})
}

// DrainUserConnections requests a clean close of every PostgreSQL connection
// for one org user. It is used when a project reader's credentials or access