mirror of
https://github.com/golang/go
synced 2024-10-01 05:28:33 -06:00
8f8fd1f239
The default timeout for urlfetch requests on App Engine Classic is 5 seconds. Sometimes the requests can take longer, so increase it to the maximum value of 60 seconds. The playground has its own timeout for running code so there's no need to impose a second level of protection. Also cleans up the code to remove old appenginevm code and use the correct imports. Change-Id: I15da96e5ba70fb008bf821f4609f431847662223 Reviewed-on: https://go-review.googlesource.com/129395 Reviewed-by: Brad Fitzpatrick <bradfitz@golang.org> Run-TryBot: Andrew Bonventre <andybons@golang.org> TryBot-Result: Gobot Gobot <gobot@golang.org>
68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
// Copyright 2013 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 playground registers HTTP handlers at "/compile" and "/share" that
|
|
// proxy requests to the golang.org playground service.
|
|
// This package may be used unaltered on App Engine.
|
|
package playground // import "golang.org/x/tools/playground"
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
const baseURL = "https://golang.org"
|
|
|
|
func init() {
|
|
http.HandleFunc("/compile", bounce)
|
|
http.HandleFunc("/share", bounce)
|
|
}
|
|
|
|
func bounce(w http.ResponseWriter, r *http.Request) {
|
|
b := new(bytes.Buffer)
|
|
if err := passThru(b, r); err != nil {
|
|
http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
|
|
report(r, err)
|
|
return
|
|
}
|
|
io.Copy(w, b)
|
|
}
|
|
|
|
func passThru(w io.Writer, req *http.Request) error {
|
|
if req.URL.Path == "/share" && !allowShare(req) {
|
|
return errors.New("Forbidden")
|
|
}
|
|
defer req.Body.Close()
|
|
url := baseURL + req.URL.Path
|
|
ctx, cancel := context.WithTimeout(contextFunc(req), 60*time.Second)
|
|
defer cancel()
|
|
r, err := post(ctx, url, req.Header.Get("Content-type"), req.Body)
|
|
if err != nil {
|
|
return fmt.Errorf("making POST request: %v", err)
|
|
}
|
|
defer r.Body.Close()
|
|
if _, err := io.Copy(w, r.Body); err != nil {
|
|
return fmt.Errorf("copying response Body: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var onAppengine = false // will be overridden by appengine.go
|
|
|
|
func allowShare(r *http.Request) bool {
|
|
if !onAppengine {
|
|
return true
|
|
}
|
|
switch r.Header.Get("X-AppEngine-Country") {
|
|
case "", "ZZ", "CN":
|
|
return false
|
|
}
|
|
return true
|
|
}
|