tulip/cmd/red-tulip/server/db.go

package server

import (
	"github.com/1f349/tulip/database"
	"log"
	"net/http"
)

// DbTx wraps a database transaction with http error messages and a simple action
// function. If the action function returns an error the transaction will be
// rolled back. If there is no error then the transaction is committed.
func (h *HttpServer) DbTx(rw http.ResponseWriter, action func(tx *database.Tx) error) bool {
	tx, err := h.db.Begin()
	if err != nil {
		http.Error(rw, "Failed to begin database transaction", http.StatusInternalServerError)
		return true
	}
	defer tx.Rollback()

	err = action(tx)
	if err != nil {
		http.Error(rw, "Database error", http.StatusInternalServerError)
		log.Println("Database action error:", err)
		return true
	}
	err = tx.Commit()
	if err != nil {
		http.Error(rw, "Database error", http.StatusInternalServerError)
		log.Println("Database commit error:", err)
	}

	return false
}