Initial checkin
This commit is contained in:
commit
5faf40c2ab
14
README.md
Normal file
14
README.md
Normal file
@ -0,0 +1,14 @@
|
||||
## path values
|
||||
|
||||
The three examples to go with the blog post at https://hawkins.id.au/posts/go-mux/path-values
|
||||
|
||||
These should all work identically, each just shows a different way of achieving
|
||||
the same thing - extracing a user id from the path and making the user struct
|
||||
(instantiated from some sort of database, faked in common/db/db.go for this
|
||||
example) available to the route handler.
|
||||
|
||||
go run go-mux/pathvalues/1-basic
|
||||
|
||||
go run go-mux/pathvalues/2-slightly-less-basic
|
||||
|
||||
go run go-mux/pathvalues/3-middleware
|
22
common/db/db.go
Normal file
22
common/db/db.go
Normal file
@ -0,0 +1,22 @@
|
||||
package db
|
||||
|
||||
import "fmt"
|
||||
|
||||
type User struct {
|
||||
ID int
|
||||
}
|
||||
|
||||
func LoadUser(id int) (User, error) {
|
||||
if id < 0 {
|
||||
return User{}, fmt.Errorf("invalid negative user id '%d'", id)
|
||||
}
|
||||
if id > 100 {
|
||||
return User{}, fmt.Errorf("invalid user id '%d'", id)
|
||||
}
|
||||
|
||||
return User{ID: id}, nil
|
||||
}
|
||||
|
||||
func (u User) String() string {
|
||||
return fmt.Sprintf("this user is id: %d", u.ID)
|
||||
}
|
36
pathvalues/1-basic/main.go
Normal file
36
pathvalues/1-basic/main.go
Normal file
@ -0,0 +1,36 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"go-mux/common/db"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/user/{userId}/view", viewUser)
|
||||
|
||||
slog.Info("starting web service on :8080")
|
||||
panic(http.ListenAndServe(":8080", nil))
|
||||
}
|
||||
|
||||
func viewUser(w http.ResponseWriter, r *http.Request) {
|
||||
userID := r.PathValue("userId")
|
||||
userIDint, err := strconv.Atoi(userID)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
w.Write([]byte("bad user id"))
|
||||
return
|
||||
}
|
||||
user, err := db.LoadUser(userIDint)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
w.Write([]byte(fmt.Sprintf("could not load user: %s", err.Error())))
|
||||
return
|
||||
}
|
||||
|
||||
slog.Info("loaded user")
|
||||
// show the user id
|
||||
w.Write([]byte(fmt.Sprintf("loaded user %s", user)))
|
||||
}
|
42
pathvalues/2-slightly-less-basic/main.go
Normal file
42
pathvalues/2-slightly-less-basic/main.go
Normal file
@ -0,0 +1,42 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"go-mux/common/db"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/user/{userId}/view", viewUser)
|
||||
|
||||
slog.Info("starting web service on :8080")
|
||||
panic(http.ListenAndServe(":8080", nil))
|
||||
}
|
||||
|
||||
func viewUser(w http.ResponseWriter, r *http.Request) {
|
||||
user, err := loadUserFromRequest(r)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
w.Write([]byte(fmt.Sprintf("could not load user: %s", err.Error())))
|
||||
return
|
||||
}
|
||||
|
||||
slog.Info("loaded user")
|
||||
// show the user id
|
||||
w.Write([]byte(fmt.Sprintf("loaded user %s", user)))
|
||||
}
|
||||
|
||||
func loadUserFromRequest(r *http.Request) (db.User, error) {
|
||||
userID := r.PathValue("userId")
|
||||
userIDint, err := strconv.Atoi(userID)
|
||||
if err != nil {
|
||||
return db.User{}, err
|
||||
}
|
||||
user, err := db.LoadUser(userIDint)
|
||||
if err != nil {
|
||||
return db.User{}, err
|
||||
}
|
||||
return user, nil
|
||||
}
|
48
pathvalues/3-middleware/main.go
Normal file
48
pathvalues/3-middleware/main.go
Normal file
@ -0,0 +1,48 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"go-mux/common/db"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
var userContextKey = "user"
|
||||
|
||||
func main() {
|
||||
http.Handle("/user/{userId}/view", getUserFromPath(http.HandlerFunc(viewUser)))
|
||||
|
||||
slog.Info("starting web service on :8080")
|
||||
panic(http.ListenAndServe(":8080", nil))
|
||||
}
|
||||
|
||||
func viewUser(w http.ResponseWriter, r *http.Request) {
|
||||
user := r.Context().Value(userContextKey).(db.User)
|
||||
|
||||
slog.Info("loaded user")
|
||||
// show the user id
|
||||
w.Write([]byte(fmt.Sprintf("loaded user %s", user.String())))
|
||||
}
|
||||
|
||||
func getUserFromPath(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
userID := r.PathValue("userId")
|
||||
userIDint, err := strconv.Atoi(userID)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
w.Write([]byte("bad user id"))
|
||||
return
|
||||
}
|
||||
user, err := db.LoadUser(userIDint)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
w.Write([]byte(fmt.Sprintf("could not load user: %s", err.Error())))
|
||||
return
|
||||
}
|
||||
newContext := context.WithValue(r.Context(), userContextKey, user)
|
||||
r = r.Clone(newContext)
|
||||
next.ServeHTTP(w, r)
|
||||
})
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user