DataProxy Service Definition (flyteorg#261)

eapolinario · Mar 23, 2022 · 8575d55 · 8575d55
1 parent 3ea75ea
commit 8575d55
Show file tree

Hide file tree

Showing 20 changed files with 5,919 additions and 9,311 deletions.
diff --git a/clients/go/admin/client.go b/clients/go/admin/client.go
@@ -29,6 +29,7 @@ type Clientset struct {
 	authMetadataServiceClient service.AuthMetadataServiceClient
 	healthServiceClient       grpc_health_v1.HealthClient
 	identityServiceClient     service.IdentityServiceClient
+	dataProxyServiceClient    service.DataProxyClient
 	authOpt                   grpc.DialOption
 }
 
@@ -55,6 +56,10 @@ func (c Clientset) IdentityClient() service.IdentityServiceClient {
 	return c.identityServiceClient
 }
 
+func (c Clientset) DataProxyClient() service.DataProxyClient {
+	return c.dataProxyServiceClient
+}
+
 func NewAdminClient(ctx context.Context, conn *grpc.ClientConn) service.AdminServiceClient {
 	logger.Infof(ctx, "Initialized Admin client")
 	return service.NewAdminServiceClient(conn)
@@ -197,6 +202,7 @@ func initializeClients(ctx context.Context, cfg *Config, tokenCache pkce.TokenCa
 	cs.authMetadataServiceClient = service.NewAuthMetadataServiceClient(adminConnection)
 	cs.identityServiceClient = service.NewIdentityServiceClient(adminConnection)
 	cs.healthServiceClient = grpc_health_v1.NewHealthClient(adminConnection)
+	cs.dataProxyServiceClient = service.NewDataProxyClient(adminConnection)
 	if authOpt != nil {
 		cs.authOpt = authOpt
 	}

diff --git a/gen/pb-cpp/flyteidl/service/dataproxy.grpc.pb.cc b/gen/pb-cpp/flyteidl/service/dataproxy.grpc.pb.cc