mirror of
https://github.com/1f349/lavender.git
synced 2024-12-22 23:54:10 +00:00
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package server
|
|
|
|
import (
|
|
"github.com/1f349/lavender/database"
|
|
"github.com/1f349/lavender/logger"
|
|
"net/http"
|
|
)
|
|
|
|
var _ error = (*ErrDatabaseActionFailed)(nil)
|
|
|
|
type ErrDatabaseActionFailed struct {
|
|
err error
|
|
}
|
|
|
|
func (e ErrDatabaseActionFailed) Error() string {
|
|
return "database action failed: " + e.err.Error()
|
|
}
|
|
|
|
func (e ErrDatabaseActionFailed) Unwrap() error {
|
|
return e.err
|
|
}
|
|
|
|
// 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.Queries) error) bool {
|
|
logger.Logger.Helper()
|
|
if h.DbTxError(action) != nil {
|
|
http.Error(rw, "Database error", http.StatusInternalServerError)
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (h *httpServer) DbTxError(action func(tx *database.Queries) error) error {
|
|
logger.Logger.Helper()
|
|
err := action(h.db)
|
|
if err != nil {
|
|
logger.Logger.Warn("Database action error", "err", err)
|
|
return ErrDatabaseActionFailed{err: err}
|
|
}
|
|
return nil
|
|
}
|