-
-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #9 from damoon/run
add "d8s run" subcommand
- Loading branch information
Showing
7 changed files
with
260 additions
and
172 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,173 @@ | ||
| package d8s | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "log" | ||
| "net" | ||
| "os" | ||
| "os/exec" | ||
| "strconv" | ||
| "time" | ||
| ) | ||
|
|
||
| func Run(ctx context.Context, allowContext string, command []string) error { | ||
| // verify kubernetes context in use | ||
| allowed, err := ContextAllowed(allowContext) | ||
| if err != nil { | ||
| return fmt.Errorf("verify kubernetes context: %v", err) | ||
| } | ||
| if !allowed { | ||
| return fmt.Errorf("kubernetes context not allowed") | ||
| } | ||
|
|
||
| // port forward | ||
| err = awaitDind() | ||
| if err != nil { | ||
| return fmt.Errorf("wait for dind to start: %v", err) | ||
| } | ||
|
|
||
| localPort, err := freePort() | ||
| if err != nil { | ||
| return fmt.Errorf("select free local port: %v", err) | ||
| } | ||
|
|
||
| go portForwardForever(ctx, localPort, dindPort) | ||
|
|
||
| err = awaitPortOpen(ctx, localPort) | ||
| if err != nil { | ||
| return fmt.Errorf("wait for port forward to start: %v", err) | ||
| } | ||
|
|
||
| // execute command | ||
| err = executeCommand(command, fmt.Sprintf("tcp://127.0.0.1:%d", localPort)) | ||
| if err != nil { | ||
| return fmt.Errorf("command failed with %s", err) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func awaitDind() error { | ||
| cmd := exec.Command( | ||
| "kubectl", | ||
| "wait", | ||
| "--for=condition=available", | ||
| "--timeout=600s", | ||
| "deployment/dind", | ||
| ) | ||
| cmd.Stdout = os.Stdout | ||
| cmd.Stderr = os.Stderr | ||
| cmd.Stdin = os.Stdin | ||
| cmd.Env = os.Environ() | ||
|
|
||
| err := cmd.Run() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func freePort() (int, error) { | ||
| addr, err := net.ResolveTCPAddr("tcp", "localhost:0") | ||
| if err != nil { | ||
| return 0, err | ||
| } | ||
|
|
||
| l, err := net.ListenTCP("tcp", addr) | ||
| if err != nil { | ||
| return 0, err | ||
| } | ||
| defer l.Close() | ||
|
|
||
| return l.Addr().(*net.TCPAddr).Port, nil | ||
| } | ||
|
|
||
| func portForwardForever(ctx context.Context, localPort, dindPort int) { | ||
| err := portForward(ctx, localPort, dindPort) | ||
| if err != nil { | ||
| log.Printf("port forward failed: %v", err) | ||
| } | ||
|
|
||
| for { | ||
| select { | ||
| case <-ctx.Done(): | ||
| return | ||
| case <-time.After(100 * time.Millisecond): | ||
| err := portForward(ctx, localPort, dindPort) | ||
| if err != nil { | ||
| log.Printf("port forward failed: %v", err) | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func portForward(ctx context.Context, localPort, dinnerPort int) error { | ||
| cmd := exec.Command( | ||
| "kubectl", | ||
| "port-forward", | ||
| "deployment/dind", | ||
| fmt.Sprintf("%d:%d", localPort, dinnerPort), | ||
| ) | ||
| cmd.Stdout = os.Stdout | ||
| cmd.Stderr = os.Stderr | ||
| cmd.Stdin = os.Stdin | ||
| cmd.Env = os.Environ() | ||
|
|
||
| err := cmd.Run() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func awaitPortOpen(ctx context.Context, localPort int) error { | ||
| for { | ||
| select { | ||
| case <-ctx.Done(): | ||
| return fmt.Errorf("port did not open: %v", ctx.Err()) | ||
| case <-time.After(1 * time.Second): | ||
| timeout, cancel := context.WithTimeout(ctx, time.Second) | ||
| defer cancel() | ||
|
|
||
| open := portOpen(timeout, "127.0.0.1", strconv.Itoa(localPort)) | ||
| if open { | ||
| return nil | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func portOpen(ctx context.Context, host string, port string) bool { | ||
|
|
||
| d := net.Dialer{Timeout: time.Second} | ||
|
|
||
| conn, err := d.DialContext(ctx, "tcp", net.JoinHostPort(host, port)) | ||
| if err != nil { | ||
| return false | ||
| } | ||
| defer conn.Close() | ||
|
|
||
| return true | ||
| } | ||
|
|
||
| func executeCommand(command []string, dockerAddr string) error { | ||
| cmd := exec.Command(command[0], command[1:]...) | ||
| cmd.Stdout = os.Stdout | ||
| cmd.Stderr = os.Stderr | ||
| cmd.Stdin = os.Stdin | ||
| cmd.Env = os.Environ() | ||
| cmd.Env = append(cmd.Env, "DOCKER_HOST="+dockerAddr) | ||
| cmd.Env = append(cmd.Env, "DOCKER_BUILDKIT=1") | ||
|
|
||
| fmt.Printf("Execute command %s\n", cmd.String()) | ||
|
|
||
| err := cmd.Run() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return nil | ||
| } |
Oops, something went wrong.