mirror of
https://github.com/1f349/lavender.git
synced 2024-12-22 07:34:06 +00:00
34 lines
911 B
Go
34 lines
911 B
Go
package server
|
|
|
|
import (
|
|
"github.com/1f349/lavender/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
|
|
}
|