mirror of
https://github.com/yunionio/cloudpods.git
synced 2026-05-06 21:52:54 +08:00
- add command executor
- host prepare for deploy on container - procutils adpat to executor
This commit is contained in:
95
cmd/executor-server/main.go
Normal file
95
cmd/executor-server/main.go
Normal file
@@ -0,0 +1,95 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"net"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"google.golang.org/grpc"
|
||||
|
||||
"yunion.io/x/executor/apis"
|
||||
"yunion.io/x/executor/server"
|
||||
"yunion.io/x/log"
|
||||
|
||||
"yunion.io/x/onecloud/pkg/util/sysutils"
|
||||
)
|
||||
|
||||
var socketPath string
|
||||
|
||||
func init() {
|
||||
flag.StringVar(&socketPath, "socket-path", "/var/run/exec.sock", "execute service listen socket path")
|
||||
}
|
||||
|
||||
func main() {
|
||||
if !sysutils.IsRootPermission() {
|
||||
log.Fatalln("executor must run on root permission")
|
||||
}
|
||||
Serve()
|
||||
}
|
||||
|
||||
type SExecuteService struct {
|
||||
}
|
||||
|
||||
func NewExecuteService() *SExecuteService {
|
||||
return &SExecuteService{}
|
||||
}
|
||||
|
||||
func (s *SExecuteService) fixPathEnv() error {
|
||||
var paths = []string{
|
||||
"/usr/local/sbin",
|
||||
"/usr/local/bin",
|
||||
"/sbin",
|
||||
"/bin",
|
||||
"/usr/sbin",
|
||||
"/usr/bin",
|
||||
}
|
||||
return os.Setenv("PATH", strings.Join(paths, ":"))
|
||||
}
|
||||
|
||||
func (s *SExecuteService) prepareEnv() error {
|
||||
if err := s.fixPathEnv(); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *SExecuteService) runService() {
|
||||
grpcServer := grpc.NewServer()
|
||||
apis.RegisterExecutorServer(grpcServer, &server.Executor{})
|
||||
if _, err := os.Stat(socketPath); !os.IsNotExist(err) {
|
||||
// socket file already exist, remove first
|
||||
if err := os.Remove(socketPath); err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
}
|
||||
|
||||
listener, err := net.Listen("unix", socketPath)
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
defer listener.Close()
|
||||
log.Infof("Init net listener on %s succ", socketPath)
|
||||
err = grpcServer.Serve(listener)
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SExecuteService) initService() {
|
||||
if len(socketPath) == 0 {
|
||||
log.Fatalf("missing socket path")
|
||||
}
|
||||
if err := s.prepareEnv(); err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SExecuteService) Run() {
|
||||
s.initService()
|
||||
s.runService()
|
||||
}
|
||||
|
||||
func Serve() {
|
||||
NewExecuteService().Run()
|
||||
}
|
||||
Reference in New Issue
Block a user