Cleaning/Updating k8petstore dev tooling and build fixing, moving webserver to
main.go. Replacing vagrant tooling with docker-machine.
This commit is contained in:
jayvyas
204
examples/k8petstore/web-server/src/main.go
Normal file
204
examples/k8petstore/web-server/src/main.go
Normal file
@@ -0,0 +1,204 @@
|
||||
/*
|
||||
Copyright 2014 The Kubernetes Authors All rights reserved.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"github.com/codegangsta/negroni"
|
||||
"github.com/gorilla/mux"
|
||||
"github.com/xyproto/simpleredis"
|
||||
)
|
||||
|
||||
//return the path to static assets (i.e. index.html)
|
||||
func pathToStaticContents() string {
|
||||
var static_content = os.Getenv("STATIC_FILES")
|
||||
// Take a wild guess. This will work in dev environment.
|
||||
if static_content == "" {
|
||||
println("*********** WARNING: DIDNT FIND ENV VAR 'STATIC_FILES', guessing your running in dev.")
|
||||
static_content = "../../static/"
|
||||
} else {
|
||||
println("=========== Read ENV 'STATIC_FILES', path to assets : " + static_content)
|
||||
}
|
||||
|
||||
//Die if no the static files are missing.
|
||||
_, err := os.Stat(static_content)
|
||||
if err != nil {
|
||||
println("*********** os.Stat failed on " + static_content + " This means no static files are available. Dying...")
|
||||
os.Exit(2)
|
||||
}
|
||||
return static_content
|
||||
}
|
||||
|
||||
func main() {
|
||||
|
||||
var connection = os.Getenv("REDISMASTER_SERVICE_HOST") + ":" + os.Getenv("REDISMASTER_SERVICE_PORT")
|
||||
|
||||
if connection == ":" {
|
||||
print("WARNING ::: If in kube, this is a failure: Missing env variable REDISMASTER_SERVICE_HOST")
|
||||
print("WARNING ::: Attempting to connect redis localhost.")
|
||||
connection = "127.0.0.1:6379"
|
||||
} else {
|
||||
print("Found redis master host " + os.Getenv("REDISMASTER_SERVICE_PORT"))
|
||||
connection = os.Getenv("REDISMASTER_SERVICE_HOST") + ":" + os.Getenv("REDISMASTER_SERVICE_PORT")
|
||||
}
|
||||
|
||||
println("Now connecting to : " + connection)
|
||||
/**
|
||||
* Create a connection pool. ?The pool pointer will otherwise
|
||||
* not be of any use.?https://gist.github.com/jayunit100/1d00e6d343056401ef00
|
||||
*/
|
||||
pool = simpleredis.NewConnectionPoolHost(connection)
|
||||
|
||||
println("Connection pool established : " + connection)
|
||||
|
||||
defer pool.Close()
|
||||
|
||||
r := mux.NewRouter()
|
||||
|
||||
println("Router created ")
|
||||
|
||||
/**
|
||||
* Define a REST path.
|
||||
* - The parameters (key) can be accessed via mux.Vars.
|
||||
* - The Methods (GET) will be bound to a handler function.
|
||||
*/
|
||||
r.Path("/info").Methods("GET").HandlerFunc(InfoHandler)
|
||||
r.Path("/lrange/{key}").Methods("GET").HandlerFunc(ListRangeHandler)
|
||||
r.Path("/rpush/{key}/{value}").Methods("GET").HandlerFunc(ListPushHandler)
|
||||
r.Path("/llen").Methods("GET").HandlerFunc(LLENHandler)
|
||||
|
||||
//for dev environment, the site is one level up...
|
||||
|
||||
r.PathPrefix("/").Handler(http.FileServer(http.Dir(pathToStaticContents())))
|
||||
|
||||
r.Path("/env").Methods("GET").HandlerFunc(EnvHandler)
|
||||
|
||||
list := simpleredis.NewList(pool, "k8petstore")
|
||||
HandleError(nil, list.Add("jayunit100"))
|
||||
HandleError(nil, list.Add("tstclaire"))
|
||||
HandleError(nil, list.Add("rsquared"))
|
||||
|
||||
// Verify that this is 3 on startup.
|
||||
infoL := HandleError(pool.Get(0).Do("LLEN", "k8petstore")).(int64)
|
||||
fmt.Printf("\n=========== Starting DB has %d elements \n", infoL)
|
||||
if infoL < 3 {
|
||||
print("Not enough entries in DB. something is wrong w/ redis querying")
|
||||
print(infoL)
|
||||
panic("Failed ... ")
|
||||
}
|
||||
|
||||
println("=========== Now launching negroni...this might take a second...")
|
||||
n := negroni.Classic()
|
||||
n.UseHandler(r)
|
||||
n.Run(":3000")
|
||||
println("Done ! Web app is now running.")
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* the Pool will be populated on startup,
|
||||
* it will be an instance of a connection pool.
|
||||
* Hence, we reference its address rather than copying.
|
||||
*/
|
||||
var pool *simpleredis.ConnectionPool
|
||||
|
||||
/**
|
||||
* REST
|
||||
* input: key
|
||||
*
|
||||
* Writes all members to JSON.
|
||||
*/
|
||||
func ListRangeHandler(rw http.ResponseWriter, req *http.Request) {
|
||||
println("ListRangeHandler")
|
||||
|
||||
key := mux.Vars(req)["key"]
|
||||
|
||||
list := simpleredis.NewList(pool, key)
|
||||
|
||||
//members := HandleError(list.GetAll()).([]string)
|
||||
members := HandleError(list.GetLastN(4)).([]string)
|
||||
|
||||
print(members)
|
||||
membersJSON := HandleError(json.MarshalIndent(members, "", " ")).([]byte)
|
||||
|
||||
print("RETURN MEMBERS = " + string(membersJSON))
|
||||
rw.Write(membersJSON)
|
||||
}
|
||||
|
||||
func LLENHandler(rw http.ResponseWriter, req *http.Request) {
|
||||
println("=========== LLEN HANDLER")
|
||||
|
||||
infoL := HandleError(pool.Get(0).Do("LLEN", "k8petstore")).(int64)
|
||||
fmt.Printf("=========== LLEN is %d ", infoL)
|
||||
lengthJSON := HandleError(json.MarshalIndent(infoL, "", " ")).([]byte)
|
||||
fmt.Printf("================ LLEN json is %d", infoL)
|
||||
|
||||
print("RETURN LEN = " + string(lengthJSON))
|
||||
rw.Write(lengthJSON)
|
||||
|
||||
}
|
||||
|
||||
func ListPushHandler(rw http.ResponseWriter, req *http.Request) {
|
||||
println("ListPushHandler")
|
||||
|
||||
/**
|
||||
* Expect a key and value as input.
|
||||
*
|
||||
*/
|
||||
key := mux.Vars(req)["key"]
|
||||
value := mux.Vars(req)["value"]
|
||||
|
||||
println("New list " + key + " " + value)
|
||||
list := simpleredis.NewList(pool, key)
|
||||
HandleError(nil, list.Add(value))
|
||||
ListRangeHandler(rw, req)
|
||||
}
|
||||
|
||||
func InfoHandler(rw http.ResponseWriter, req *http.Request) {
|
||||
println("InfoHandler")
|
||||
|
||||
info := HandleError(pool.Get(0).Do("INFO")).([]byte)
|
||||
rw.Write(info)
|
||||
}
|
||||
|
||||
func EnvHandler(rw http.ResponseWriter, req *http.Request) {
|
||||
println("EnvHandler")
|
||||
|
||||
environment := make(map[string]string)
|
||||
for _, item := range os.Environ() {
|
||||
splits := strings.Split(item, "=")
|
||||
key := splits[0]
|
||||
val := strings.Join(splits[1:], "=")
|
||||
environment[key] = val
|
||||
}
|
||||
|
||||
envJSON := HandleError(json.MarshalIndent(environment, "", " ")).([]byte)
|
||||
rw.Write(envJSON)
|
||||
}
|
||||
|
||||
func HandleError(result interface{}, err error) (r interface{}) {
|
||||
if err != nil {
|
||||
print("ERROR : " + err.Error())
|
||||
//panic(err)
|
||||
}
|
||||
return result
|
||||
}
|
||||
Reference in New Issue
Block a user