main.go

package main

import (
	"log"
	"net/http"
	"time"

	"github.com/3cb/ssc"
	"github.com/boltdb/bolt"
	"github.com/gorilla/mux"
	"github.com/gorilla/websocket"
)

func main() {
	// open database and create buckets of key/value pairs
	db, err := bolt.Open("my.db", 0600, nil)
	if err != nil {
		log.Fatal(err)
	}
	err = createBucket(db, "Messages")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	pool := ssc.NewPool([]string{}, time.Second*0)
	err = pool.Start()
	if err != nil {
		log.Fatal("Unable to start pool to serve websocket connections.")
	}

	// start 5 minute polling goroutine
	go poll(db, pool)

	// create new router instance
	r := mux.NewRouter()

	// serve static files
	r.Handle("/", http.FileServer(http.Dir("./static/")))
	r.PathPrefix("/dist/").Handler(http.FileServer(http.Dir("./static/")))

	// handle websocket requests
	upgrader := &websocket.Upgrader{}
	r.Handle("/ws", WS(db, pool, upgrader))
	r.Handle("/api/day/{date}", SingleDate(db))
	// {date} variable is first day of month
	r.Handle("/api/month/{date}", Month())
	// {date} variable is first day of year
	r.Handle("/api/year/{date}", Year())

	// start server
	log.Fatal(http.ListenAndServe(":3030", r))
}