Minimal golang backend for pullrequests.opencast.org

api.go

package main

import (
	"flag"
	"fmt"

	"github.com/codegangsta/martini"
	"github.com/garyburd/redigo/redis"
	"github.com/martini-contrib/render"
)

var (
	redisAddress   = flag.String("redis-address", ":6379", "Address to the Redis server")
	maxConnections = flag.Int("max-connections", 10, "Max connections to Redis")
)

func main() {
	martini.Env = martini.Prod

	flag.Parse()

	redisPool := redis.NewPool(func() (redis.Conn, error) {
		c, err := redis.Dial("tcp", *redisAddress)

		if err != nil {
			return nil, err
		}

		return c, err
	}, *maxConnections)

	defer redisPool.Close()

	m := martini.Classic()

	m.Map(redisPool)

	m.Use(render.Renderer())

	m.Get("/", func() string {
		return "Hello from Martini!"
	})

	m.Get("/get/:key", func(pool *redis.Pool, params martini.Params) string {
		//key := params["key"]

		c := pool.Get()
		defer c.Close()

		// Get keys
		keys, err := redis.Values(c.Do("KEYS", "*pr_*"))
		if err != nil {
			return fmt.Sprintf("Could not get keys: %s", err)
		}

		last := len(keys) - 1
		result := "["
		for index, key := range keys {
			value, err := redis.String(c.Do("GET", key))

			if err != nil {
				return fmt.Sprintf("Could not GET %s", key)
			} else {
				result += value
				if index < last {
					result += ","
				}
			}
		}

		return result + "]"
	})

	m.Run()
}