forked from kyma-project/kyma
-
Notifications
You must be signed in to change notification settings - Fork 0
/
asset_resolver.go
83 lines (69 loc) · 2.55 KB
/
asset_resolver.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package assetstore
import (
"context"
"github.com/golang/glog"
"github.com/kyma-project/kyma/components/asset-store-controller-manager/pkg/apis/assetstore/v1alpha2"
"github.com/kyma-project/kyma/components/console-backend-service/internal/domain/assetstore/listener"
"github.com/kyma-project/kyma/components/console-backend-service/internal/domain/assetstore/pretty"
"github.com/kyma-project/kyma/components/console-backend-service/internal/gqlerror"
"github.com/kyma-project/kyma/components/console-backend-service/internal/gqlschema"
"github.com/pkg/errors"
)
type assetResolver struct {
assetSvc assetSvc
assetConverter gqlAssetConverter
fileSvc fileSvc
fileConverter gqlFileConverter
}
func newAssetResolver(assetService assetSvc) *assetResolver {
return &assetResolver{
assetSvc: assetService,
assetConverter: &assetConverter{},
fileSvc: &fileService{},
fileConverter: &fileConverter{},
}
}
func (r *assetResolver) AssetFilesField(ctx context.Context, obj *gqlschema.Asset, filterExtensions []string) ([]gqlschema.File, error) {
if obj == nil {
glog.Error(errors.New("%s cannot be empty in order to resolve `files` field"), pretty.Asset)
return nil, gqlerror.NewInternal()
}
asset, err := r.assetSvc.Find(obj.Namespace, obj.Name)
if err != nil {
glog.Error(errors.Wrapf(err, "while gathering %s for %s %s", pretty.Asset, pretty.Asset, obj.Name))
return nil, gqlerror.New(err, pretty.Asset)
}
if asset == nil {
return nil, nil
}
var items []*File
if len(filterExtensions) == 0 {
items, err = r.fileSvc.Extract(&asset.Status.AssetRef)
} else {
items, err = r.fileSvc.FilterByExtensionsAndExtract(&asset.Status.AssetRef, filterExtensions)
}
if err != nil {
glog.Error(errors.Wrapf(err, "while gathering %s for %s %s", pretty.Files, pretty.Asset, obj.Name))
return nil, gqlerror.New(err, pretty.Asset)
}
files, err := r.fileConverter.ToGQLs(items)
if err != nil {
glog.Error(errors.Wrapf(err, "while converting %s", pretty.Files))
return nil, gqlerror.New(err, pretty.Asset)
}
return files, nil
}
func (r *assetResolver) AssetEventSubscription(ctx context.Context, namespace string) (<-chan gqlschema.AssetEvent, error) {
channel := make(chan gqlschema.AssetEvent, 1)
filter := func(entity *v1alpha2.Asset) bool {
return entity != nil && entity.Namespace == namespace
}
assetListener := listener.NewAsset(channel, filter, r.assetConverter)
r.assetSvc.Subscribe(assetListener)
go func() {
defer close(channel)
defer r.assetSvc.Unsubscribe(assetListener)
<-ctx.Done()
}()
return channel, nil
}