James Munnelly
122 lines
2.4 KiB
Go
122 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"fmt"
|
|
"strings"
|
|
|
|
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"
|
|
)
|
|
|
|
const (
|
|
cmdPath = "/plexmediaserver/bootstrap.sh"
|
|
logFilePath = "/var/log/plex/plex-elastic-transcoder.log"
|
|
plexServerURL = "10.20.40.60:32400"
|
|
)
|
|
|
|
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()
|
|
|
|
// Setup logs
|
|
fo, err := os.Create(logFilePath)
|
|
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Error opening log file: %s", err))
|
|
}
|
|
|
|
defer func() {
|
|
if err := fo.Close(); err != nil {
|
|
panic(fmt.Sprintf("Error closing file: %s", err))
|
|
}
|
|
}()
|
|
|
|
log.SetOutput(fo)
|
|
|
|
// Get the arguments passed to Plex New Transcoder
|
|
args := os.Args[1:]
|
|
wd, _ := os.Getwd()
|
|
for i, arg := range args {
|
|
if arg == "-progressurl" {
|
|
// Change the progress URL to report to about the transcode
|
|
args[i + 1] = strings.Replace(args[i+1], "127.0.0.1:32400", plexServerURL, 1)
|
|
}
|
|
}
|
|
args = append([]string{wd}, args...)
|
|
|
|
log.Print("In WD: ", wd)
|
|
log.Print("Dispatching job with args: ", args)
|
|
|
|
job := executors.Job{
|
|
Command: []string{cmdPath},
|
|
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.")
|
|
}
|