mirror of https://github.com/kianby/stacosys
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
7 years ago
|
package main
|
||
|
|
||
|
import (
|
||
|
"github.com/patrickmn/go-cache"
|
||
|
"io"
|
||
|
"io/ioutil"
|
||
|
"log"
|
||
|
"net/http"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
const stacoURL string = "http://127.0.0.1:8100"
|
||
|
|
||
|
var countCache = cache.New(5*time.Minute, 10*time.Minute)
|
||
|
|
||
|
func commentsCount(w http.ResponseWriter, r *http.Request) {
|
||
|
|
||
|
// only GET method is supported
|
||
|
if r.Method != "GET" {
|
||
|
http.NotFound(w, r)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// set header
|
||
|
w.Header().Add("Content-Type", "application/json")
|
||
|
w.Header().Add("Access-Control-Allow-Origin", "*")
|
||
|
|
||
|
// get cached value
|
||
|
cachedBody, found := countCache.Get(r.URL.String())
|
||
|
if found {
|
||
|
//log.Printf("return cached value")
|
||
|
w.Write([]byte(cachedBody.(string)))
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// relay request to stacosys
|
||
|
response, err := http.Get(stacoURL + r.URL.String())
|
||
|
if err != nil {
|
||
|
http.NotFound(w, r)
|
||
|
return
|
||
|
}
|
||
|
defer response.Body.Close()
|
||
|
body, err := ioutil.ReadAll(response.Body)
|
||
|
if err != nil {
|
||
|
http.NotFound(w, r)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// cache body and return response
|
||
|
countCache.Set(r.URL.String(), string(body), cache.DefaultExpiration)
|
||
|
log.Printf(string(body))
|
||
|
w.Write(body)
|
||
|
}
|
||
|
|
||
|
func comments(w http.ResponseWriter, r *http.Request) {
|
||
|
io.WriteString(w, "Hello world!")
|
||
|
}
|
||
|
|
||
|
func main() {
|
||
|
http.HandleFunc("/comments/count", commentsCount)
|
||
|
http.HandleFunc("/comments", comments)
|
||
|
http.ListenAndServe(":8200", nil)
|
||
|
}
|