48 lines
898 B
Go
48 lines
898 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 {
|
|
log.Printf("Listening on %s", addr)
|
|
return http.ListenAndServe(addr, s.n)
|
|
}
|