- add command executor

- host prepare for deploy on container
- procutils adpat to executor
This commit is contained in:
wanyaoqi
2019-07-13 10:37:01 +08:00
parent b5cddd294e
commit 7bf48c992e
56 changed files with 3091 additions and 771 deletions

View 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()
}