package main
import (
"easyCache"
"flag"
"fmt"
"log"
"net/http"
)
var db = map[string]string{
"Tom": "630",
"Jack": "589",
"Sam": "567",
}
func createGroup() *easyCache.Group {
return easyCache.NewGroup("score", 0, easyCache.GetterFunc(func(key string) ([]byte, error) {
log.Println("[SlowDB] search key", key)
if v, ok := db[key]; ok {
return []byte(v), nil
}
return nil, fmt.Errorf("%s not exist", key)
}))
}
func startCacheServer(addr string, addrs []string, easy *easyCache.Group) {
peers := easyCache.NewHttpPool(addr)
peers.Set(addrs...)
easy.RegisterPeers(peers)
log.Println("easyCache is running at", addr)
log.Fatal(http.ListenAndServe(addr[7:], peers))
}
func startAPIServer(apiAddr string, gee *easyCache.Group) {
http.Handle("/api", http.HandlerFunc(
func(w http.ResponseWriter, r *http.Request) {
key := r.URL.Query().Get("key")
view, err := gee.Get(key)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/octet-stream")
w.Write([]byte(view.String()))
}))
log.Println("fontend server is running at", apiAddr)
log.Fatal(http.ListenAndServe(apiAddr[7:], nil))
}
func main() {
var port int
var api bool
flag.IntVar(&port, "port", 9010, "Geecache server port")
flag.BoolVar(&api, "api", false, "Start a api server?")
flag.Parse()
apiAddr := "http://127.0.0.1:9999"
addrMap := map[int]string{
9011: "http://127.0.0.1:9011",
9012: "http://127.0.0.1:9012",
9010: "http://127.0.0.1:9010",
}
var addrs []string
for _, v := range addrMap {
addrs = append(addrs, v)
}
gee := createGroup()
if api {
go startAPIServer(apiAddr, gee)
}
startCacheServer(addrMap[port], []string(addrs), gee)
}
warechar / easycache Goto Github PK
View Code? Open in Web Editor NEWLicense: Apache License 2.0