rest_server.go

package main

// HTTP REST server for all client interactions

import (
	"fmt"
	"github.com/julienschmidt/httprouter"
	"net/http"
)

var shutdown chan bool = make(chan bool, 1)

var restServer *RestServer

type RestServer struct {
	PrettyPrint bool
}

// Authenticate
func (this *RestServer) auth(r *http.Request) bool {
	// @todo Implement
	return true
}

// Not authorized
func (this *RestServer) notAuthorized(w http.ResponseWriter) {
	w.WriteHeader(http.StatusUnauthorized)
}

// Not found
func (this *RestServer) notFound(w http.ResponseWriter) {
	w.WriteHeader(http.StatusNotFound)
}

// Start
func (this *RestServer) start() {
	go func() {
		// New router
		router := httprouter.New()

		// Debug handlers
		if conf.HttpDebug {
			log.Warn("HTTP debug endpoints are ON")

			// File locator
			router.GET("/v1/debug/file-locator/shards", GetDebugFileLocatorShards)
			router.GET("/v1/debug/file-locator/shard-locations", GetDebugFileLocatorShardLocations)

			// Gossip
			router.GET("/v1/debug/gossip/nodes", GetDebugGossipNodes)

			// Block
			router.POST("/v1/debug/block/allocate", PostDebugBlockAllocate)
			router.PUT("/v1/debug/block/persist", PutDebugBlockPersist)
		}

		// File
		router.POST("/v1/file", PostFile)
		router.GET("/v1/file", GetFile)

		// Local calls
		router.GET("/v1/local/file", GetLocalFile) // Local file will attempt to load file from this server

		// Start server
		log.Infof("Starting REST HTTP server on port TCP/%d", conf.HttpPort)
		log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", conf.HttpPort), router))
	}()
}

func newRestServer() *RestServer {
	o := &RestServer{
		PrettyPrint: true,
	}
	o.start()
	return o
}