Image digests for referencing bundles

- Change schema of OperatorBundle to include field 'bundlepath' - Extend GetBundle API to return bundle image path - Modify GetBundle API to return struct with avaiable fields instead of error while querrying empty bundles
operator-framework · Oct 25, 2019 · d48ff71 · d48ff71
1 parent 161f66a
commit d48ff71
Show file tree

Hide file tree

Showing 14 changed files with 646 additions and 154 deletions.
diff --git a/pkg/api/conversion.go b/pkg/api/conversion.go
@@ -55,7 +55,7 @@ func BundleStringToObjectStrings(bundleString string) ([]string, error) {
 	return objs, nil
 }
 
-func BundleStringToAPIBundle(bundleString string, entry *registry.ChannelEntry) (*Bundle, error) {
+func BundleStringToAPIBundle(bundleString string, bundlepathString string, entry *registry.ChannelEntry) (*Bundle, error) {
 	objs, err := BundleStringToObjectStrings(bundleString)
 	if err != nil {
 		return nil, err
@@ -80,5 +80,6 @@ func BundleStringToAPIBundle(bundleString string, entry *registry.ChannelEntry)
 	}
 	out.ChannelName = entry.ChannelName
 	out.PackageName = entry.PackageName
+	out.BundlePath = bundlepathString
 	return out, nil
 }
diff --git a/pkg/api/registry.pb.go b/pkg/api/registry.pb.go
diff --git a/pkg/api/registry.proto b/pkg/api/registry.proto
@@ -36,6 +36,7 @@ message Bundle{
 	string channelName = 3;
 	string csvJson = 4;
 	repeated string object = 5;
+	string bundlePath = 6;
 }
 
 message ChannelEntry{

diff --git a/pkg/registry/empty.go b/pkg/registry/empty.go
@@ -22,20 +22,20 @@ func (EmptyQuery) GetPackage(ctx context.Context, name string) (*PackageManifest
 	return nil, errors.New("empty querier: cannot get package")
 }
 
-func (EmptyQuery) GetBundle(ctx context.Context, pkgName, channelName, csvName string) (string, error) {
-	return "", errors.New("empty querier: cannot get bundle")
+func (EmptyQuery) GetBundle(ctx context.Context, pkgName, channelName, csvName string) (string, string, error) {
+	return "", "", errors.New("empty querier: cannot get bundle")
 }
 
-func (EmptyQuery) GetBundleForChannel(ctx context.Context, pkgName string, channelName string) (string, error) {
-	return "", errors.New("empty querier: cannot get bundle for channel")
+func (EmptyQuery) GetBundleForChannel(ctx context.Context, pkgName string, channelName string) (string, string, error) {
+	return "", "", errors.New("empty querier: cannot get bundle for channel")
 }
 
 func (EmptyQuery) GetChannelEntriesThatReplace(ctx context.Context, name string) (entries []*ChannelEntry, err error) {
 	return nil, errors.New("empty querier: cannot get channel entries that replace")
 }
 
-func (EmptyQuery) GetBundleThatReplaces(ctx context.Context, name, pkgName, channelName string) (string, error) {
-	return "", errors.New("empty querier: cannot get bundle that replaces")
+func (EmptyQuery) GetBundleThatReplaces(ctx context.Context, name, pkgName, channelName string) (string, string, error) {
+	return "", "", errors.New("empty querier: cannot get bundle that replaces")
 }
 
 func (EmptyQuery) GetChannelEntriesThatProvide(ctx context.Context, group, version, kind string) (entries []*ChannelEntry, err error) {
@@ -46,8 +46,8 @@ func (EmptyQuery) GetLatestChannelEntriesThatProvide(ctx context.Context, group,
 	return nil, errors.New("empty querier: cannot get latest channel entries that provide")
 }
 
-func (EmptyQuery) GetBundleThatProvides(ctx context.Context, group, version, kind string) (string, *ChannelEntry, error) {
-	return "", nil, errors.New("empty querier: cannot get bundle that provides")
+func (EmptyQuery) GetBundleThatProvides(ctx context.Context, group, version, kind string) (string, string, *ChannelEntry, error) {
+	return "", "", nil, errors.New("empty querier: cannot get bundle that provides")
 }
 
 func (EmptyQuery) ListImages(ctx context.Context) ([]string, error) {

diff --git a/pkg/registry/interface.go b/pkg/registry/interface.go
@@ -13,18 +13,18 @@ type Query interface {
 	ListTables(ctx context.Context) ([]string, error)
 	ListPackages(ctx context.Context) ([]string, error)
 	GetPackage(ctx context.Context, name string) (*PackageManifest, error)
-	GetBundle(ctx context.Context, pkgName, channelName, csvName string) (string, error)
-	GetBundleForChannel(ctx context.Context, pkgName string, channelName string) (string, error)
+	GetBundle(ctx context.Context, pkgName, channelName, csvName string) (string, string, error)
+	GetBundleForChannel(ctx context.Context, pkgName string, channelName string) (string, string, error)
 	// Get all channel entries that say they replace this one
 	GetChannelEntriesThatReplace(ctx context.Context, name string) (entries []*ChannelEntry, err error)
 	// Get the bundle in a package/channel that replace this one
-	GetBundleThatReplaces(ctx context.Context, name, pkgName, channelName string) (string, error)
+	GetBundleThatReplaces(ctx context.Context, name, pkgName, channelName string) (string, string, error)
 	// Get all channel entries that provide an api
 	GetChannelEntriesThatProvide(ctx context.Context, group, version, kind string) (entries []*ChannelEntry, err error)
 	// Get latest channel entries that provide an api
 	GetLatestChannelEntriesThatProvide(ctx context.Context, group, version, kind string) (entries []*ChannelEntry, err error)
 	// Get the the latest bundle that provides the API in a default channel
-	GetBundleThatProvides(ctx context.Context, group, version, kind string) (string, *ChannelEntry, error)
+	GetBundleThatProvides(ctx context.Context, group, version, kind string) (string, string, *ChannelEntry, error)
 	// List all images in the database
 	ListImages(ctx context.Context) ([]string, error)
 	// List all images for a particular bundle

diff --git a/pkg/server/server.go b/pkg/server/server.go
@@ -39,27 +39,40 @@ func (s *RegistryServer) GetPackage(ctx context.Context, req *api.GetPackageRequ
 }
 
 func (s *RegistryServer) GetBundle(ctx context.Context, req *api.GetBundleRequest) (*api.Bundle, error) {
-	bundleString, err := s.store.GetBundle(ctx, req.GetPkgName(), req.GetChannelName(), req.GetCsvName())
+	bundleString, bundlepathString, err := s.store.GetBundle(ctx, req.GetPkgName(), req.GetChannelName(), req.GetCsvName())
 	if err != nil {
 		return nil, err
 	}
+	// If the value of the `bundle` field in the OperatorBundle table is NULL, return a
+	// Bundle struct with available fields
+	if bundleString == "" {
+		bundle := &api.Bundle{
+			PackageName: req.PkgName,
+			ChannelName: req.ChannelName,
+			CsvName:     req.CsvName,
+			CsvJson:     "",
+			Object:      []string{},
+			BundlePath:  bundlepathString,
+		}
+		return bundle, nil
+	}
 	entry := &registry.ChannelEntry{
 		PackageName: req.GetPkgName(),
 		ChannelName: req.GetChannelName(),
 	}
-	return api.BundleStringToAPIBundle(bundleString, entry)
+	return api.BundleStringToAPIBundle(bundleString, bundlepathString, entry)
 }
 
 func (s *RegistryServer) GetBundleForChannel(ctx context.Context, req *api.GetBundleInChannelRequest) (*api.Bundle, error) {
-	bundleString, err := s.store.GetBundleForChannel(ctx, req.GetPkgName(), req.GetChannelName())
+	bundleString, bundlepathString, err := s.store.GetBundleForChannel(ctx, req.GetPkgName(), req.GetChannelName())
 	if err != nil {
 		return nil, err
 	}
 	entry := &registry.ChannelEntry{
 		PackageName: req.GetPkgName(),
 		ChannelName: req.GetChannelName(),
 	}
-	return api.BundleStringToAPIBundle(bundleString, entry)
+	return api.BundleStringToAPIBundle(bundleString, bundlepathString, entry)
 }
 
 func (s *RegistryServer) GetChannelEntriesThatReplace(req *api.GetAllReplacementsRequest, stream api.Registry_GetChannelEntriesThatReplaceServer) error {
@@ -76,7 +89,7 @@ func (s *RegistryServer) GetChannelEntriesThatReplace(req *api.GetAllReplacement
 }
 
 func (s *RegistryServer) GetBundleThatReplaces(ctx context.Context, req *api.GetReplacementRequest) (*api.Bundle, error) {
-	bundleString, err := s.store.GetBundleThatReplaces(ctx, req.GetCsvName(), req.GetPkgName(), req.GetChannelName())
+	bundleString, bundlepathString, err := s.store.GetBundleThatReplaces(ctx, req.GetCsvName(), req.GetPkgName(), req.GetChannelName())
 	if err != nil {
 		return nil, err
 	}
@@ -85,7 +98,7 @@ func (s *RegistryServer) GetBundleThatReplaces(ctx context.Context, req *api.Get
 		ChannelName: req.GetChannelName(),
 		Replaces:    req.GetCsvName(),
 	}
-	return api.BundleStringToAPIBundle(bundleString, entry)
+	return api.BundleStringToAPIBundle(bundleString, bundlepathString, entry)
 }
 
 func (s *RegistryServer) GetChannelEntriesThatProvide(req *api.GetAllProvidersRequest, stream api.Registry_GetChannelEntriesThatProvideServer) error {
@@ -115,9 +128,9 @@ func (s *RegistryServer) GetLatestChannelEntriesThatProvide(req *api.GetLatestPr
 }
 
 func (s *RegistryServer) GetDefaultBundleThatProvides(ctx context.Context, req *api.GetDefaultProviderRequest) (*api.Bundle, error) {
-	bundleString, channelEntry, err := s.store.GetBundleThatProvides(ctx, req.GetGroup(), req.GetVersion(), req.GetKind())
+	bundleString, bundlepathString, channelEntry, err := s.store.GetBundleThatProvides(ctx, req.GetGroup(), req.GetVersion(), req.GetKind())
 	if err != nil {
 		return nil, err
 	}
-	return api.BundleStringToAPIBundle(bundleString, channelEntry)
+	return api.BundleStringToAPIBundle(bundleString, bundlepathString, channelEntry)
 }
diff --git a/pkg/server/server_test.go b/pkg/server/server_test.go
diff --git a/pkg/sqlite/configmap_test.go b/pkg/sqlite/configmap_test.go
diff --git a/pkg/sqlite/directory_test.go b/pkg/sqlite/directory_test.go
diff --git a/pkg/sqlite/load.go b/pkg/sqlite/load.go
@@ -56,7 +56,7 @@ func (s *SQLLoader) AddOperatorBundle(bundle *registry.Bundle) error {
 		tx.Rollback()
 	}()
 
-	stmt, err := tx.Prepare("insert into operatorbundle(name, csv, bundle) values(?, ?, ?)")
+	stmt, err := tx.Prepare("insert into operatorbundle(name, csv, bundle, bundlepath) values(?, ?, ?, ?)")
 	if err != nil {
 		return err
 	}
@@ -77,7 +77,7 @@ func (s *SQLLoader) AddOperatorBundle(bundle *registry.Bundle) error {
 		return fmt.Errorf("csv name not found")
 	}
 
-	if _, err := stmt.Exec(csvName, csvBytes, bundleBytes); err != nil {
+	if _, err := stmt.Exec(csvName, csvBytes, bundleBytes, nil); err != nil {
 		return err
 	}