mirror of
https://github.com/unknwon/the-way-to-go_ZH_CN.git
synced 2025-08-12 05:11:49 +08:00
update book code
This commit is contained in:
41
eBook/examples/chapter_19/goto_v1/main.go
Normal file
41
eBook/examples/chapter_19/goto_v1/main.go
Normal file
@@ -0,0 +1,41 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
const AddForm = `
|
||||
<form method="POST" action="/add">
|
||||
URL: <input type="text" name="url">
|
||||
<input type="submit" value="Add">
|
||||
</form>
|
||||
`
|
||||
var store = NewURLStore()
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/", Redirect)
|
||||
http.HandleFunc("/add", Add)
|
||||
http.ListenAndServe(":8080", nil)
|
||||
}
|
||||
|
||||
func Redirect(w http.ResponseWriter, r *http.Request) {
|
||||
key := r.URL.Path[1:]
|
||||
url := store.Get(key)
|
||||
if url == "" {
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
http.Redirect(w, r, url, http.StatusFound)
|
||||
}
|
||||
|
||||
|
||||
func Add(w http.ResponseWriter, r *http.Request) {
|
||||
url := r.FormValue("url")
|
||||
if url == "" {
|
||||
fmt.Fprint(w, AddForm)
|
||||
return
|
||||
}
|
||||
key := store.Put(url)
|
||||
fmt.Fprintf(w, "http://localhost:8080/%s", key)
|
||||
}
|
Reference in New Issue
Block a user