mirror of
https://github.com/golang/go
synced 2024-11-18 09:44:50 -07:00
1f1f69e389
The original test would open a local port and then immediately close it and use the port number in subsequent tests. Between the port being closed and reused by the later process, it could be opened by some other program on the machine. Changed the test to run the server process directly and have it save the assigned port to a text file to be used by client processes. Fixes #5564. LGTM=adg R=golang-codereviews, gobot, adg CC=golang-codereviews https://golang.org/cl/72290043
112 lines
2.5 KiB
Go
112 lines
2.5 KiB
Go
// Copyright 2010 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"html/template"
|
|
"io/ioutil"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"regexp"
|
|
)
|
|
|
|
var (
|
|
addr = flag.Bool("addr", false, "find open address and print to final-port.txt")
|
|
)
|
|
|
|
type Page struct {
|
|
Title string
|
|
Body []byte
|
|
}
|
|
|
|
func (p *Page) save() error {
|
|
filename := p.Title + ".txt"
|
|
return ioutil.WriteFile(filename, p.Body, 0600)
|
|
}
|
|
|
|
func loadPage(title string) (*Page, error) {
|
|
filename := title + ".txt"
|
|
body, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Page{Title: title, Body: body}, nil
|
|
}
|
|
|
|
func viewHandler(w http.ResponseWriter, r *http.Request, title string) {
|
|
p, err := loadPage(title)
|
|
if err != nil {
|
|
http.Redirect(w, r, "/edit/"+title, http.StatusFound)
|
|
return
|
|
}
|
|
renderTemplate(w, "view", p)
|
|
}
|
|
|
|
func editHandler(w http.ResponseWriter, r *http.Request, title string) {
|
|
p, err := loadPage(title)
|
|
if err != nil {
|
|
p = &Page{Title: title}
|
|
}
|
|
renderTemplate(w, "edit", p)
|
|
}
|
|
|
|
func saveHandler(w http.ResponseWriter, r *http.Request, title string) {
|
|
body := r.FormValue("body")
|
|
p := &Page{Title: title, Body: []byte(body)}
|
|
err := p.save()
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
http.Redirect(w, r, "/view/"+title, http.StatusFound)
|
|
}
|
|
|
|
var templates = template.Must(template.ParseFiles("edit.html", "view.html"))
|
|
|
|
func renderTemplate(w http.ResponseWriter, tmpl string, p *Page) {
|
|
err := templates.ExecuteTemplate(w, tmpl+".html", p)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
}
|
|
}
|
|
|
|
var validPath = regexp.MustCompile("^/(edit|save|view)/([a-zA-Z0-9]+)$")
|
|
|
|
func makeHandler(fn func(http.ResponseWriter, *http.Request, string)) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
m := validPath.FindStringSubmatch(r.URL.Path)
|
|
if m == nil {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
fn(w, r, m[2])
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
http.HandleFunc("/view/", makeHandler(viewHandler))
|
|
http.HandleFunc("/edit/", makeHandler(editHandler))
|
|
http.HandleFunc("/save/", makeHandler(saveHandler))
|
|
|
|
if *addr {
|
|
l, err := net.Listen("tcp", "127.0.0.1:0")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
err = ioutil.WriteFile("final-port.txt", []byte(l.Addr().String()), 0644)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
s := &http.Server{}
|
|
s.Serve(l)
|
|
return
|
|
}
|
|
|
|
http.ListenAndServe(":8080", nil)
|
|
}
|