Method HandlerFuncs

internal/server/cluster/gateway.go:156–345

HandlerFuncs returns the HTTP handlers that should be added to the REST API endpoint in order to handle database-related requests. There are two handlers, one for the /internal/raft endpoint and the other for /internal/db, which handle respectively raft and gRPC-SQL requests. These handlers might

(heartbeatHandler HeartbeatHandler, trustedCerts func() (map[certificate.Type]map[string]x509.Certificate, error))

// non-clustered member not available over the network or because it is not a
// database node part of the cowsql cluster.
func (g *Gateway) HandlerFuncs(heartbeatHandler HeartbeatHandler, trustedCerts func() (map[certificate.Type]map[string]x509.Certificate, error)) map[string]http.HandlerFunc {
	database := func(w http.ResponseWriter, r *http.Request) {
		g.lock.RLock()

		certs, err := trustedCerts()
		if err != nil {
			g.lock.RUnlock()
			http.Error(w, "403 invalid client certificate", http.StatusForbidden)
			return
		}

		if !tlsCheckCert(r, g.networkCert, g.state().ServerCert(), certs) {
			g.lock.RUnlock()
			http.Error(w, "403 invalid client certificate", http.StatusForbidden)
			return
		}

		g.lock.RUnlock()

		// Compare the cowsql version of the connecting client
		// with our own one.
		versionHeader := r.Header.Get("X-Dqlite-Version")
		if versionHeader == "" {
			// No version header means an old pre cowsql 1.0 client.
			versionHeader = "0"
		}

		version, err := strconv.Atoi(versionHeader)
		if err != nil {
			http.Error(w, "400 invalid cowsql version", http.StatusBadRequest)
			return
		}

		if version != cowsqlVersion {
			if version > cowsqlVersion {
				g.lock.Lock()
				if !g.upgradeTriggered {
					err = triggerUpdate(g.state())
					if err == nil {
						g.upgradeTriggered = true
					}
				}
				g.lock.Unlock()
				http.Error(w, "503 unsupported cowsql version", http.StatusServiceUnavailable)
			} else {
				http.Error(w, "426 cowsql version too old ", http.StatusUpgradeRequired)
			}

			return
		}

		// Handle heartbeats (these normally come from leader, but can come from joining nodes too).
		if r.Method == "PUT" {
			if g.shutdownCtx.Err() != nil {
				logger.Warn("Rejecting heartbeat request as shutting down")
				http.Error(w, "503 Shutting down", http.StatusServiceUnavailable)
				return
			}