main.go

package main

import (
	"flag"
	"fmt"
	"log"
	"net"
	"net/http"
	"os"
	"os/signal"
	"syscall"
)

func main() {
	runWorker()
}

func runWorker() {
	// Define a command-line flag for specifying the socket path.
	socketPathPtr := flag.String("socket-path", "", "Unix domain socket path")
	flag.Parse()

	if *socketPathPtr != "" {
		// Create a Unix domain socket and listen for incoming connections.
		socket, err := net.Listen("unix", *socketPathPtr)
		if err != nil {
			panic(err)
		}

		// Cleanup the sockfile.
		c := make(chan os.Signal, 1)
		signal.Notify(c, os.Interrupt, syscall.SIGTERM)
		go func() {
			<-c
			fmt.Println("Cleaning up socket file...")
			if _, err := os.Stat(*socketPathPtr); err == nil {
				os.Remove(*socketPathPtr)
			}
			os.Exit(1)
		}()

		handler := http.NewServeMux()
		handler.HandleFunc("/execute", handleRequest)
		handler.HandleFunc("/kill", handleKill)
		handler.HandleFunc("/health", handleHealthCheck)

		server := http.Server{
			Handler: handler,
		}
		fmt.Println("Worker listening on socket ", *socketPathPtr)
		err = server.Serve(socket)
		if err != nil {
			panic(err)
		}
	} else {
		http.HandleFunc("/execute", handleRequest)
		http.HandleFunc("/kill", handleKill)
		http.HandleFunc("/health", handleHealthCheck)

		log.Println("Listening on :8080...")
		log.Fatal(http.ListenAndServe(":8080", nil))
	}
}