gophertype/pkg/app.go

47 行
861 B
Go

package gophertype
import (
"log"
"net/http"
"git.kirsle.net/apps/gophertype/pkg/models"
"github.com/gorilla/mux"
"github.com/jinzhu/gorm"
"github.com/urfave/negroni"
)
// Site is the master struct for the Gophertype server.
type Site struct {
n *negroni.Negroni
mux *mux.Router
}
// NewSite initializes the Site.
func NewSite() *Site {
site := &Site{}
n := negroni.New()
n.Use(negroni.NewRecovery())
n.Use(negroni.NewLogger())
site.n = n
return site
}
// UseDB specifies the database to use.
func (s *Site) UseDB(driver string, path string) error {
db, err := gorm.Open(driver, path)
if err != nil {
return err
}
log.Printf("Using database driver '%s'", driver)
models.UseDB(db)
return nil
}
// ListenAndServe starts the HTTP service.
func (s *Site) ListenAndServe(addr string) error {
return http.ListenAndServe(addr, s.n)
}