모듈 업데이트

client/client.go

@@ -103,7 +103,6 @@ type procmeta struct {
 }
 
 type houstonClient struct {
-	client        *grpc.ClientConn
 	childProcs    []*procmeta
 	extraMetrics  unsafe.Pointer // map[string]float32
 	deploys       map[string][]*protos.VersionAndArgs
@@ -111,9 +110,10 @@ type houstonClient struct {
 	ctx           context.Context
 	operationChan chan *protos.OperationQueryResponse
 	exitChan      chan *exec.Cmd
-	httpAddr      string
+	clientChan    chan *grpc.ClientConn
 	timestamp     string
 	wg            sync.WaitGroup
+	config        clientConfig
 }
 
 func unmarshal[T any](val *T, src map[string]string) {
@@ -195,22 +195,6 @@ func NewClient() (HoustonClient, error) {
 		return nil, errors.New("client.http_server_address is missing")
 	}
 
-	var client *grpc.ClientConn
-	for {
-		logger.Println("grpc.DialContext :", clientConfig.GrpcAddress)
-		dialContext, cancelDial := context.WithTimeout(context.Background(), 15*time.Second)
-		client, err = grpc.DialContext(dialContext, clientConfig.GrpcAddress, grpc.WithBlock(), grpc.WithTransportCredentials(insecure.NewCredentials()))
-		if err == nil {
-			cancelDial()
-			break
-		}
-
-		cancelDial()
-		logger.Println("grpc.DialContext failed :", err)
-		time.Sleep(5 * time.Second)
-	}
-
-	logger.Println("grpc.DialContext succeeded")
 	exefile, err := os.Executable()
 	if err != nil {
 		return nil, err
@@ -234,10 +218,10 @@ func NewClient() (HoustonClient, error) {
 	}
 
 	hc := &houstonClient{
-		client:       client,
+		config:       clientConfig,
+		clientChan:   make(chan *grpc.ClientConn),
 		extraMetrics: unsafe.Pointer(&map[string]float32{}),
 		deploys:      deploys,
-		httpAddr:     clientConfig.HttpAddress,
 		timestamp:    exefi.ModTime().String(),
 	}
 
@@ -251,7 +235,7 @@ func NewClient() (HoustonClient, error) {
 		defer hc.wg.Done()
 
 		// 메인 operator
-		op := protos.NewOperationClient(hc.client)
+		var op protos.OperationClient
 		myname, _ := os.Executable()
 		myname = path.Base(filepath.ToSlash(myname))
 		if len(path.Ext(myname)) > 0 {
@@ -266,6 +250,9 @@ func NewClient() (HoustonClient, error) {
 			case <-ctx.Done():
 				return
 
+			case newClient := <-hc.clientChan:
+				op = protos.NewOperationClient(newClient)
+
 			case exited := <-exitChan:
 				var newprocs []*procmeta
 				for _, proc := range hc.childProcs {
@@ -329,21 +316,21 @@ func NewClient() (HoustonClient, error) {
 				case shared.Start:
 					var sr shared.StartProcessRequest
 					unmarshal(&sr, resp.Args)
-					if err := hc.startChildProcess(&sr); err != nil {
+					if err := hc.startChildProcess(&sr, op); err != nil {
 						logger.Println(err)
 					}
 
 				case shared.Stop:
 					var sr shared.StopProcessRequest
 					unmarshal(&sr, resp.Args)
-					if err := hc.stopChildProcess(&sr); err != nil {
+					if err := hc.stopChildProcess(&sr, op); err != nil {
 						logger.Println(err)
 					}
 
 				case shared.Restart:
 					var rr shared.RestartProcessRequest
 					unmarshal(&rr, resp.Args)
-					if err := hc.restartChildProcess(&rr); err != nil {
+					if err := hc.restartChildProcess(&rr, op); err != nil {
 						logger.Println(err)
 					}
 
@@ -392,15 +379,37 @@ func (hc *houstonClient) Start() {
 		hc.shutdownFunc()
 	}()
 
+	var client *grpc.ClientConn
+	reconnCount := 0
 	for {
 		select {
 		case <-hc.ctx.Done():
 			return
 
 		default:
-			err := hc.checkOperation()
-			if err != nil {
-				logger.Println("hc.checkUpdate failed :", err)
+			if client == nil {
+				if reconnCount == 0 {
+					logger.Println("grpc.DialContext :", hc.config.GrpcAddress)
+				}
+
+				reconnCount++
+				dialContext, cancelDial := context.WithTimeout(context.Background(), 15*time.Second)
+				client, _ = grpc.DialContext(dialContext, hc.config.GrpcAddress, grpc.WithBlock(), grpc.WithTransportCredentials(insecure.NewCredentials()))
+				cancelDial()
+				if client != nil {
+					reconnCount = 0
+					logger.Println("grpc.DialContext succeeded")
+					hc.clientChan <- client
+				}
+			}
+
+			if client != nil {
+				err := hc.checkOperation(client)
+				if err != nil {
+					logger.Println("hc.checkUpdate failed :", err)
+
+					client = nil
+				}
 			}
 		}
 	}
@@ -410,7 +419,7 @@ func (hc *houstonClient) Shutdown() {
 	hc.shutdownFunc()
 }
 
-func (hc *houstonClient) checkOperation() error {
+func (hc *houstonClient) checkOperation(client *grpc.ClientConn) error {
 	defer func() {
 		r := recover()
 		if r != nil {
@@ -418,7 +427,7 @@ func (hc *houstonClient) checkOperation() error {
 		}
 	}()
 
-	op := protos.NewOperationClient(hc.client)
+	op := protos.NewOperationClient(client)
 	cl, err := op.Query(hc.ctx, grpc.WaitForReady(true))
 	if err != nil {
 		return err

client/deploy.go

@@ -194,7 +194,7 @@ func (hc *houstonClient) prepareDeploy(name string, version string) (destPath st
 func (hc *houstonClient) makeDownloadUrl(rel string) string {
 	out := rel
 	if !strings.HasPrefix(out, "http") {
-		tks := strings.SplitN(hc.httpAddr, "://", 2)
+		tks := strings.SplitN(hc.config.HttpAddress, "://", 2)
 		out = fmt.Sprintf("%s://%s", tks[0], path.Join(tks[1], rel))
 	}
 	return out

client/operation.go

@@ -53,7 +53,7 @@ func (hc *houstonClient) uploadZipLogFile(zipFile string, name string, version s
 
 	defer zf.Close()
 
-	req, err := http.NewRequest("POST", hc.httpAddr+"/upload", zf)
+	req, err := http.NewRequest("POST", hc.config.HttpAddress+"/upload", zf)
 	if err != nil {
 		logger.Println(err)
 	}
@@ -323,7 +323,7 @@ func (hc *houstonClient) launch(meta *procmeta) error {
 
 var errPrepareprocessLaunchFailed = errors.New("prepareProcessLaunch failed")
 
-func (hc *houstonClient) startChildProcess(req *shared.StartProcessRequest) error {
+func (hc *houstonClient) startChildProcess(req *shared.StartProcessRequest, op protos.OperationClient) error {
 	logger.Println("startChildProcess :", *req)
 	if req.Version == "latest" {
 		// 최신 버전을 찾음
@@ -363,8 +363,6 @@ func (hc *houstonClient) startChildProcess(req *shared.StartProcessRequest) erro
 	}
 
 	hc.childProcs = append(hc.childProcs, meta)
-
-	op := protos.NewOperationClient(hc.client)
 	op.Refresh(context.Background(), hc.makeOperationQueryRequest())
 
 	return nil
@@ -372,7 +370,7 @@ func (hc *houstonClient) startChildProcess(req *shared.StartProcessRequest) erro
 
 var errNoRunningProcess = errors.New("no running processed")
 
-func (hc *houstonClient) stopChildProcess(req *shared.StopProcessRequest) error {
+func (hc *houstonClient) stopChildProcess(req *shared.StopProcessRequest, op protos.OperationClient) error {
 	if req.Version == "latest" {
 		// 최신 버전을 찾음
 		latest, err := shared.FindLastestVersion(req.Name)
@@ -422,7 +420,6 @@ func (hc *houstonClient) stopChildProcess(req *shared.StopProcessRequest) error
 			}
 		}
 
-		op := protos.NewOperationClient(hc.client)
 		op.Refresh(context.Background(), hc.makeOperationQueryRequest())
 
 		for _, proc := range killing {
@@ -439,7 +436,7 @@ func (hc *houstonClient) stopChildProcess(req *shared.StopProcessRequest) error
 	return errNoRunningProcess
 }
 
-func (hc *houstonClient) restartChildProcess(req *shared.RestartProcessRequest) error {
+func (hc *houstonClient) restartChildProcess(req *shared.RestartProcessRequest, op protos.OperationClient) error {
 	if req.Version == "latest" {
 		// 최신 버전을 찾음
 		latest, err := shared.FindLastestVersion(req.Name)
@@ -472,7 +469,6 @@ func (hc *houstonClient) restartChildProcess(req *shared.RestartProcessRequest)
 		proc.state = protos.ProcessState_Stopping
 	}
 
-	op := protos.NewOperationClient(hc.client)
 	op.Refresh(context.Background(), hc.makeOperationQueryRequest())
 
 	for _, proc := range restarts {

go.mod

@@ -6,7 +6,7 @@ require (
 	golang.org/x/text v0.9.0
 	google.golang.org/grpc v1.55.0
 	google.golang.org/protobuf v1.30.0
-	repositories.action2quare.com/ayo/gocommon v0.0.0-20230613151240-0e6b155201c3
+	repositories.action2quare.com/ayo/gocommon v0.0.0-20230613162208-4ebf231bbe72
 )
 
 require (

go.sum

@@ -18,5 +18,5 @@ google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp0
 google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
 google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
 google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
-repositories.action2quare.com/ayo/gocommon v0.0.0-20230613151240-0e6b155201c3 h1:EOWhQ3rBoNGlFmTxqF1ujOI+bXtMbExdRcUo0arLkS8=
-repositories.action2quare.com/ayo/gocommon v0.0.0-20230613151240-0e6b155201c3/go.mod h1:ng62uGMGXyQSeuxePG5gJAMtip4Rnspu5Tu7hgvaXns=
+repositories.action2quare.com/ayo/gocommon v0.0.0-20230613162208-4ebf231bbe72 h1:2qaxct6dumM2JjguSp5nofeWZwWsozKSJWC36q9Q0qc=
+repositories.action2quare.com/ayo/gocommon v0.0.0-20230613162208-4ebf231bbe72/go.mod h1:ng62uGMGXyQSeuxePG5gJAMtip4Rnspu5Tu7hgvaXns=

replacer/main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"encoding/json"
 	"errors"
 	"io"
 	"log"
@@ -70,7 +71,7 @@ func main() {
 
 	selfext, _ := os.Executable()
 	selfext = path.Base(selfext)
-
+	nextArgs := args[4:]
 	entries, _ := os.ReadDir(args[2])
 	for _, ent := range entries {
 		if ent.Name() == selfext {
@@ -82,7 +83,14 @@ func main() {
 				stdlog.Fatal(err)
 			}
 		} else {
-			if err := copy(path.Join(args[2], ent.Name()), ent.Name()); err != nil {
+			if ent.Name() == "@args" {
+				var tempArgs []string
+				argfile, _ := os.Open(path.Join(args[2], ent.Name()))
+				dec := json.NewDecoder(argfile)
+				if dec.Decode(&tempArgs) == nil {
+					nextArgs = tempArgs
+				}
+			} else if err := copy(path.Join(args[2], ent.Name()), ent.Name()); err != nil {
 				stdlog.Println("copy failed :", path.Join(args[2], ent.Name()), ent.Name())
 				stdlog.Fatal(err)
 			}
@@ -100,6 +108,6 @@ func main() {
 	}
 
 	stdlog.Println("exec.Command :", args)
-	cmd := exec.Command(args[3], args[4:]...)
+	cmd := exec.Command(args[3], nextArgs...)
 	cmd.Start()
 }