Prefix with “Must” all functions that panic

Just following what the standard library does.

pkg/db.go

@@ -29,9 +29,9 @@ func NewDatabase(ctx context.Context, connString string) (*Db, error) {
 			cookie = value
 		}
 		if _, err := conn.Exec(ctx, "select set_cookie($1)", cookie); err != nil {
-                       log.Printf("ERROR - Failed to set role: %v", err)
-                       return false
-	       }
+			log.Printf("ERROR - Failed to set role: %v", err)
+			return false
+		}
 		return true
 	}
 
@@ -58,11 +58,19 @@ func (db *Db) Acquire(ctx context.Context) (*Conn, error) {
 	return &Conn{conn}, nil
 }
 
+func (db *Db) MustAcquire(ctx context.Context) *Conn {
+	conn, err := db.Acquire(ctx)
+	if err != nil {
+		panic(err)
+	}
+	return conn
+}
+
 type Conn struct {
 	*pgxpool.Conn
 }
 
-func (c *Conn) Text(ctx context.Context, def string, sql string, args ...interface{}) string {
+func (c *Conn) MustGetText(ctx context.Context, def string, sql string, args ...interface{}) string {
 	var result string
 	if err := c.Conn.QueryRow(ctx, sql, args...).Scan(&result); err != nil {
 		if err == pgx.ErrNoRows {
@@ -74,7 +82,7 @@ func (c *Conn) Text(ctx context.Context, def string, sql string, args ...interfa
 	return result
 }
 
-func (c *Conn) Exec(ctx context.Context, sql string, args ...interface{}) {
+func (c *Conn) MustExec(ctx context.Context, sql string, args ...interface{}) {
 	if _, err := c.Conn.Exec(ctx, sql, args...); err != nil {
 		panic(err)
 	}

pkg/locale.go

@@ -11,7 +11,7 @@ import (
 const contextLocaleKey = "numerus-locale"
 
 func Locale(db *Db, next http.Handler) http.Handler {
-	availableLanguages := getAvailableLanguages(db)
+	availableLanguages := mustGetAvailableLanguages(db)
 	var matcher = language.NewMatcher(availableLanguages)
 
 	locales := map[language.Tag]*gotext.Locale{}
@@ -46,7 +46,7 @@ func getLocale(r *http.Request) *gotext.Locale {
 	return r.Context().Value(contextLocaleKey).(*gotext.Locale)
 }
 
-func getAvailableLanguages(db *Db) []language.Tag {
+func mustGetAvailableLanguages(db *Db) []language.Tag {
 	rows, err := db.Query(context.Background(), "select lang_tag from language where selectable")
 	if err != nil {
 		panic(err)

pkg/login.go

@@ -41,7 +41,7 @@ func LoginHandler() http.Handler {
 		}
 		if r.Method == "POST" {
 			conn := getConn(r)
-			cookie := conn.Text(r.Context(), "", "select login($1, $2, $3)", page.Email, page.Password, remoteAddr(r))
+			cookie := conn.MustGetText(r.Context(), "", "select login($1, $2, $3)", page.Email, page.Password, remoteAddr(r))
 			if cookie != "" {
 				http.SetCookie(w, createSessionCookie(cookie, 8766*24*time.Hour))
 				http.Redirect(w, r, "/", http.StatusSeeOther)
@@ -52,7 +52,7 @@ func LoginHandler() http.Handler {
 		} else {
 			w.WriteHeader(http.StatusOK)
 		}
-		renderTemplate(w, r, "login.html", page)
+		mustRenderTemplate(w, r, "login.html", page)
 	})
 }
 
@@ -61,7 +61,7 @@ func LogoutHandler() http.Handler {
 		user := getUser(r)
 		if user.LoggedIn {
 			conn := getConn(r)
-			conn.Exec(r.Context(), "select logout()")
+			conn.MustExec(r.Context(), "select logout()")
 			http.SetCookie(w, createSessionCookie("", -24*time.Hour))
 		}
 		http.Redirect(w, r, "/login", http.StatusSeeOther)
@@ -94,10 +94,7 @@ func CheckLogin(db *Db, next http.Handler) http.Handler {
 			ctx = context.WithValue(ctx, ContextCookieKey, cookie.Value)
 		}
 
-		conn, err := db.Acquire(ctx)
-		if err != nil {
-			panic(err)
-		}
+		conn := db.MustAcquire(ctx)
 		defer conn.Release()
 		ctx = context.WithValue(ctx, ContextConnKey, conn)
 

pkg/profile.go

@@ -29,7 +29,7 @@ func ProfileHandler() http.Handler {
 		conn := getConn(r)
 		page := ProfilePage{
 			Email:     user.Email,
-			Languages: getLanguageOptions(r.Context(), conn),
+			Languages: mustGetLanguageOptions(r.Context(), conn),
 		}
 		if r.Method == "POST" {
 			r.ParseForm()
@@ -38,17 +38,17 @@ func ProfileHandler() http.Handler {
 			page.Password = r.FormValue("password")
 			page.PasswordConfirm = r.FormValue("password_confirm")
 			page.Language = r.FormValue("language")
-			conn.Exec(r.Context(), "update user_profile set name = $1, email = $2, lang_tag = $3", page.Name, page.Email, page.Language);
+			conn.MustExec(r.Context(), "update user_profile set name = $1, email = $2, lang_tag = $3", page.Name, page.Email, page.Language)
 		} else {
 			if err := conn.QueryRow(r.Context(), "select name, lang_tag from user_profile").Scan(&page.Name, &page.Language); err != nil {
 				panic(nil)
 			}
 		}
-		renderTemplate(w, r, "profile.html", page)
+		mustRenderTemplate(w, r, "profile.html", page)
 	})
 }
 
-func getLanguageOptions(ctx context.Context, conn *Conn) []LanguageOption {
+func mustGetLanguageOptions(ctx context.Context, conn *Conn) []LanguageOption {
 	rows, err := conn.Query(ctx, "select lang_tag, endonym from language where selectable")
 	if err != nil {
 		panic(err)

pkg/router.go

@@ -13,7 +13,7 @@ func NewRouter(db *Db) http.Handler {
 	router.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		user := getUser(r)
 		if user.LoggedIn {
-			renderTemplate(w, r, "index.html", nil)
+			mustRenderTemplate(w, r, "index.html", nil)
 		} else {
 			http.Redirect(w, r, "/login", http.StatusSeeOther)
 		}

pkg/template.go

@@ -6,7 +6,7 @@ import (
 	"net/http"
 )
 
-func renderTemplate(wr io.Writer, r *http.Request, filename string, data interface{}) {
+func mustRenderTemplate(wr io.Writer, r *http.Request, filename string, data interface{}) {
 	locale := getLocale(r)
 	t := template.New(filename)
 	t.Funcs(template.FuncMap{