|
1 | | -// SPDX-FileCopyrightText: 2022-present Open Networking Foundation <info@opennetworking.org> |
| 1 | +// SPDX-FileCopyrightText: 2022-present Intel Corporation |
2 | 2 | // |
3 | 3 | // SPDX-License-Identifier: Apache-2.0 |
4 | 4 |
|
5 | 5 | package main |
6 | 6 |
|
| 7 | +import ( |
| 8 | + "fmt" |
| 9 | + "github.com/atomix/runtime/proxy/pkg/runtime" |
| 10 | + "github.com/spf13/cobra" |
| 11 | + "os" |
| 12 | + "os/signal" |
| 13 | + "syscall" |
| 14 | +) |
| 15 | + |
7 | 16 | func main() { |
| 17 | + cmd := &cobra.Command{ |
| 18 | + Use: "atomix-proxy", |
| 19 | + Run: func(cmd *cobra.Command, args []string) { |
| 20 | + proxyHost, err := cmd.Flags().GetString("proxy-host") |
| 21 | + if err != nil { |
| 22 | + fmt.Fprintln(cmd.OutOrStderr(), err.Error()) |
| 23 | + os.Exit(1) |
| 24 | + } |
| 25 | + proxyPort, err := cmd.Flags().GetInt("proxy-port") |
| 26 | + if err != nil { |
| 27 | + fmt.Fprintln(cmd.OutOrStderr(), err.Error()) |
| 28 | + os.Exit(1) |
| 29 | + } |
| 30 | + runtimeHost, err := cmd.Flags().GetString("runtime-host") |
| 31 | + if err != nil { |
| 32 | + fmt.Fprintln(cmd.OutOrStderr(), err.Error()) |
| 33 | + os.Exit(1) |
| 34 | + } |
| 35 | + runtimePort, err := cmd.Flags().GetInt("runtime-port") |
| 36 | + if err != nil { |
| 37 | + fmt.Fprintln(cmd.OutOrStderr(), err.Error()) |
| 38 | + os.Exit(1) |
| 39 | + } |
| 40 | + configFile, err := cmd.Flags().GetString("config") |
| 41 | + if err != nil { |
| 42 | + fmt.Fprintln(cmd.OutOrStderr(), err.Error()) |
| 43 | + os.Exit(1) |
| 44 | + } |
| 45 | + cacheDir, err := cmd.Flags().GetString("cache") |
| 46 | + if err != nil { |
| 47 | + fmt.Fprintln(cmd.OutOrStderr(), err.Error()) |
| 48 | + os.Exit(1) |
| 49 | + } |
| 50 | + |
| 51 | + rt := runtime.New( |
| 52 | + runtime.WithProxyHost(proxyHost), |
| 53 | + runtime.WithProxyPort(proxyPort), |
| 54 | + runtime.WithRuntimeHost(runtimeHost), |
| 55 | + runtime.WithRuntimePort(runtimePort), |
| 56 | + runtime.WithConfigFile(configFile), |
| 57 | + runtime.WithCacheDir(cacheDir)) |
| 58 | + if err := rt.Start(); err != nil { |
| 59 | + fmt.Fprintln(cmd.OutOrStderr(), err.Error()) |
| 60 | + os.Exit(1) |
| 61 | + } |
| 62 | + |
| 63 | + // Wait for an interrupt signal |
| 64 | + ch := make(chan os.Signal, 2) |
| 65 | + signal.Notify(ch, os.Interrupt, syscall.SIGTERM) |
| 66 | + <-ch |
| 67 | + |
| 68 | + // Stop the proxy service |
| 69 | + if err := rt.Stop(); err != nil { |
| 70 | + fmt.Println(err) |
| 71 | + os.Exit(1) |
| 72 | + } |
| 73 | + }, |
| 74 | + } |
| 75 | + cmd.Flags().StringP("proxy-host", "h", "", "the host to which to bind the proxy server") |
| 76 | + cmd.Flags().IntP("proxy-port", "p", 5678, "the port to which to bind the proxy server") |
| 77 | + cmd.Flags().String("runtime-host", "", "the host to which to bind the runtime server") |
| 78 | + cmd.Flags().Int("runtime-port", 5679, "the port to which to bind the runtime server") |
| 79 | + cmd.Flags().StringP("config", "c", "~/.atomix/config.yaml", "the path to the Atomix configuration file") |
| 80 | + cmd.Flags().String("cache", "~/.atomix/cache", "the path to the Atomix cache") |
8 | 81 |
|
| 82 | + if err := cmd.Execute(); err != nil { |
| 83 | + panic(err) |
| 84 | + } |
9 | 85 | } |
0 commit comments