@@ -18,7 +18,6 @@ package main
1818
1919import (
2020 "context"
21- "crypto/x509/pkix"
2221 "net/http"
2322 "os"
2423 "os/signal"
@@ -28,8 +27,6 @@ import (
2827 pkg "github.com/kloeckner-i/db-auth-gateway/internal"
2928 "github.com/kloeckner-i/db-auth-gateway/internal/api"
3029 "github.com/kloeckner-i/db-auth-gateway/internal/config"
31- "github.com/kloeckner-i/db-auth-gateway/internal/mock"
32- "github.com/kloeckner-i/db-auth-gateway/internal/pubkey"
3330 "github.com/prometheus/client_golang/prometheus/promhttp"
3431 log "github.com/sirupsen/logrus"
3532 "github.com/spf13/cobra"
@@ -83,32 +80,7 @@ func execute() error {
8380 return err
8481 }
8582
86- mockCmd := & cobra.Command {
87- Use : "mock" ,
88- Short : "Run a local Google managed database mock (for testing)" ,
89- RunE : startMock ,
90- }
91-
92- mockCmd .Flags ().String ("instance" , "" , "Fully qualified database instance to connect to (project:region:name)" )
93- mockCmd .Flags ().String ("db-address" , "" , "Address of the database for which to proxy" )
94- mockCmd .Flags ().Int ("api-port" , 8080 , "Port on which to serve the mocked sqladmin API" )
95- mockCmd .Flags ().Int ("proxy-port" , 3307 , "Port on which to serve the database proxy" )
96-
97- if err := mockCmd .MarkFlagRequired ("instance" ); err != nil {
98- return err
99- }
100-
101- if err := mockCmd .MarkFlagRequired ("db-address" ); err != nil {
102- return err
103- }
104-
105- rootCmd .AddCommand (mockCmd )
106-
107- if err := rootCmd .Execute (); err != nil {
108- return err
109- }
110-
111- return nil
83+ return rootCmd .Execute ()
11284}
11385
11486func startGateway (cmd * cobra.Command , args []string ) error {
@@ -206,54 +178,3 @@ func startGateway(cmd *cobra.Command, args []string) error {
206178
207179 return gateway .Run (ctx )
208180}
209-
210- func startMock (cmd * cobra.Command , args []string ) error {
211- instance , err := cmd .Flags ().GetString ("instance" )
212- if err != nil {
213- return err
214- }
215-
216- dbAddress , err := cmd .Flags ().GetString ("db-address" )
217- if err != nil {
218- return err
219- }
220-
221- apiPort , err := cmd .Flags ().GetInt ("api-port" )
222- if err != nil {
223- return err
224- }
225-
226- proxyPort , err := cmd .Flags ().GetInt ("proxy-port" )
227- if err != nil {
228- return err
229- }
230-
231- authority , err := pubkey .NewAuthority (pkix.Name {
232- CommonName : "Mock CA" ,
233- }, time .Hour )
234- if err != nil {
235- log .Fatal (err )
236- }
237-
238- dbProxy , err := mock .NewDatabaseProxy (authority , proxyPort , instance , dbAddress )
239- if err != nil {
240- log .Fatal (err )
241- }
242-
243- sqlAdminAPI , err := mock .NewSQLAdminAPI (dbProxy , authority , apiPort )
244- if err != nil {
245- log .Fatal (err )
246- }
247-
248- go func () {
249- if err := dbProxy .Run (); err != nil {
250- log .Fatal (err )
251- }
252- }()
253-
254- if err := sqlAdminAPI .Run (); err != nil {
255- log .Fatal (err )
256- }
257-
258- return nil
259- }
0 commit comments