func PushSocketHandler(ws *websocket.Conn) {
timer := time.Now()
// can we pass this in somehow?
config := util.MzGetConfig("config.ini")
// Convert the token_key from base64 (if present)
if k, ok := config["token_key"]; ok {
key, _ := base64.URLEncoding.DecodeString(k.(string))
config["token_key"] = key
}
logger := util.NewHekaLogger(config)
store := storage.New(config, logger)
sock := PushWS{Uaid: "",
Socket: ws,
Scmd: make(chan PushCommand),
Ccmd: make(chan PushCommand),
Store: store,
Logger: logger,
Born: timer}
sock.Logger.Info("main", "New socket connection detected", nil)
defer func(log *util.HekaLogger) {
if r := recover(); r != nil {
log.Error("main", r.(error).Error(), nil)
}
}(sock.Logger)
go NewWorker(config).Run(sock)
for {
select {
case serv_cmd := <-sock.Scmd:
result, args := HandleServerCommand(serv_cmd, &sock)
sock.Logger.Debug("main",
fmt.Sprintf("Returning Result %s", result),
nil)
sock.Scmd <- PushCommand{result, args}
}
}
}
// -- REST
func UpdateHandler(resp http.ResponseWriter, req *http.Request, config util.JsMap, logger *util.HekaLogger) {
// Handle the version updates.
var err error
timer := time.Now()
logger.Debug("main", fmt.Sprintf("Handling Update %s", req.URL.Path), nil)
if req.Method != "PUT" {
http.Error(resp, "", http.StatusMethodNotAllowed)
return
}
vers := fmt.Sprintf("%d", time.Now().UTC().Unix())
elements := strings.Split(req.URL.Path, "/")
pk := elements[len(elements)-1]
if len(pk) == 0 {
logger.Error("main", "No token, rejecting request", nil)
http.Error(resp, "Token not found", http.StatusNotFound)
return
}
store := storage.New(config, logger)
if token, ok := config["token_key"]; ok {
var err error
bpk, err := Decode(token.([]byte),
pk)
if err != nil {
logger.Error("main",
fmt.Sprintf("Could not decode token %s, %s", pk, err),
nil)
http.Error(resp, "", http.StatusNotFound)
return
}
pk = strings.TrimSpace(string(bpk))
}
uaid, appid, err := storage.ResolvePK(pk)
if err != nil {
logger.Error("main",
fmt.Sprintf("Could not resolve PK %s, %s", pk, err), nil)
return
}
currentHost := "localhost"
if val, ok := config["shard.currentHost"]; ok {
currentHost = val.(string)
}
host, err := store.GetUAIDHost(uaid)
if err != nil {
logger.Error("main",
fmt.Sprintf("Could not discover host for %s, %s (using default)",
uaid, err), nil)
if val, ok := config["shard.defaultHost"]; ok {
host = val.(string)
} else {
val = "localhost"
}
}
if host != currentHost || host != "localhost" {
logger.Info("main", fmt.Sprintf("Proxying request to %s", host), nil)
err = proxyNotification(host, req.URL.Path)
if err != nil {
logger.Error("main",
fmt.Sprintf("Proxy to %s failed: %s", host, err),
nil)
}
return
}
defer func(uaid, appid, path string, timer time.Time) {
logger.Info("timer", "Client Update complete",
util.JsMap{
"uaid": uaid,
"path": req.URL.Path,
"channelID": appid,
"duration": time.Now().Sub(timer).Nanoseconds()})
}(uaid, appid, req.URL.Path, timer)
logger.Info("main",
fmt.Sprintf("setting version for %s.%s to %s", uaid, appid, vers),
nil)
err = store.UpdateChannel(pk, vers)
if err != nil {
errstr := fmt.Sprintf("Could not update channel %s.%s :: %s", uaid, appid, err)
logger.Warn("main", errstr, nil)
status := sperrors.ErrToStatus(err)
http.Error(resp, errstr, status)
return
}
resp.Header().Set("Content-Type", "application/json")
resp.Write([]byte("{}"))
logger.Info("timer", "Client Update complete",
util.JsMap{"uaid": uaid,
"channelID": appid,
"duration": time.Now().Sub(timer).Nanoseconds()})
// Ping the appropriate server
if client, ok := Clients[uaid]; ok {
Flush(client)
}
return
}