James Munnelly
88 lines
1.7 KiB
Go
88 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
log "github.com/Sirupsen/logrus"
|
|
|
|
"github.com/munnerz/plex-elastic-transcoder/common"
|
|
"github.com/munnerz/plex-elastic-transcoder/executors"
|
|
|
|
_ "github.com/munnerz/plex-elastic-transcoder/executors/kubernetes"
|
|
)
|
|
|
|
var executor executors.Executor
|
|
|
|
func signals() {
|
|
// Signal handling
|
|
c := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(c, os.Interrupt)
|
|
signal.Notify(c, syscall.SIGTERM)
|
|
go func() {
|
|
<-c
|
|
log.Print("Shutting down...")
|
|
|
|
if executor != nil {
|
|
log.Print("Terminating running job...")
|
|
err := executor.Stop()
|
|
if err != nil {
|
|
log.Fatal("Error terminating job: ", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
log.Print("Successfully terminated running job.")
|
|
}
|
|
|
|
os.Exit(0)
|
|
}()
|
|
}
|
|
func main() {
|
|
// Setup signals
|
|
signals()
|
|
|
|
// Get the arguments passed to Plex New Transcoder
|
|
args := os.Args[1:]
|
|
log.Print("Dispatching job with args: ", args)
|
|
|
|
job := executors.Job{
|
|
Command: []string{"/Plex New Transcoder"},
|
|
Args: args,
|
|
}
|
|
|
|
executor = common.CreateExecutor(job)
|
|
|
|
log.Print("Created executor: ", executor)
|
|
|
|
err := executor.Start()
|
|
if err != nil {
|
|
log.Fatal("Job start failed with error: ", err)
|
|
}
|
|
|
|
log.Print("Waiting for build pod to enter Running state...")
|
|
|
|
err = executor.WaitForState(executors.ExecutorRunning)
|
|
if err != nil {
|
|
log.Fatal("Error waiting for pod to enter running state: ", err)
|
|
}
|
|
|
|
log.Print("Job has started running...")
|
|
log.Print("Waiting for job to complete...")
|
|
|
|
err = executor.WaitForState(executors.ExecutorSucceeded)
|
|
if err != nil {
|
|
log.Fatal("Error waiting for job to complete: ", err)
|
|
}
|
|
|
|
log.Print("Job completed. Destroying pod...")
|
|
|
|
err = executor.Stop()
|
|
if err != nil {
|
|
log.Fatal("Error stopping job: ", err)
|
|
}
|
|
|
|
log.Print("Pod destroyed. Exiting.")
|
|
}
|