Home
Diffusion Flamenco Manager (wip-redesign) Browse

flamenco-manager/shaman/server.go
b71a9ffaa2eawip-redesign

Last Change
Raw File

server.go
View Options

/* (c) 2019, Blender Foundation - Sybren A. Stüvel
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package shaman
import (
"sync"
"github.com/armadillica/flamenco-manager/jwtauth"
"github.com/armadillica/flamenco-manager/shaman/checkout"
"github.com/armadillica/flamenco-manager/shaman/config"
"github.com/armadillica/flamenco-manager/shaman/fileserver"
"github.com/armadillica/flamenco-manager/shaman/filestore"
"github.com/armadillica/flamenco-manager/shaman/httpserver"
"github.com/gorilla/mux"
)
// Server represents a Shaman Server.
type Server struct {
config config.Config
auther jwtauth.Authenticator
fileStore filestore.Storage
fileServer *fileserver.FileServer
checkoutMan *checkout.Manager
shutdownChan chan struct{}
wg sync.WaitGroup
}
// NewServer creates a new Shaman server.
func NewServer(conf config.Config, auther jwtauth.Authenticator) *Server {
fileStore := filestore.New(conf)
checkoutMan := checkout.NewManager(conf, fileStore)
fileServer := fileserver.New(fileStore)
shamanServer := &Server{
conf,
auther,
fileStore,
fileServer,
checkoutMan,
make(chan struct{}),
sync.WaitGroup{},
}
return shamanServer
}
// Go starts goroutines for background operations.
// After Go() has been called, use Close() to stop those goroutines.
func (s *Server) Go() {
packageLogger.Info("Shaman server starting")
s.fileServer.Go()
if s.config.GarbageCollect.Period == 0 {
packageLogger.Warning("garbage collection disabled, set garbageCollect.period > 0 in configuration")
} else if s.config.GarbageCollect.SilentlyDisable {
packageLogger.Debug("not starting garbage collection")
} else {
s.wg.Add(1)
go s.periodicCleanup()
}
}
// Auther returns the JWT authentication manager.
func (s *Server) Auther() jwtauth.Authenticator {
return s.auther
}
// AddRoutes adds the Shaman server endpoints to the given router.
func (s *Server) AddRoutes(router *mux.Router) {
s.checkoutMan.AddRoutes(router, s.auther)
s.fileServer.AddRoutes(router, s.auther)
httpserver.RegisterTestRoutes(router, s.auther)
}
// Close shuts down the Shaman server.
func (s *Server) Close() {
packageLogger.Info("shutting down Shaman server")
close(s.shutdownChan)
s.fileServer.Close()
s.checkoutMan.Close()
s.wg.Wait()
}