65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
// Copyright (c) 2018 Yandex LLC. All rights reserved.
|
|
// Author: Dmitry Novikov <novikoff@yandex-team.ru>
|
|
|
|
package sdkresolvers
|
|
|
|
import (
|
|
"context"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
"github.com/yandex-cloud/go-genproto/yandex/cloud/compute/v1"
|
|
ycsdk "github.com/yandex-cloud/go-sdk"
|
|
"github.com/yandex-cloud/go-sdk/pkg/sdkerrors"
|
|
)
|
|
|
|
type imageResolver struct {
|
|
BaseNameResolver
|
|
}
|
|
|
|
func ImageResolver(name string, opts ...ResolveOption) ycsdk.Resolver {
|
|
return &imageResolver{
|
|
BaseNameResolver: NewBaseNameResolver(name, "image", opts...),
|
|
}
|
|
}
|
|
|
|
func (r *imageResolver) Run(ctx context.Context, sdk *ycsdk.SDK, opts ...grpc.CallOption) error {
|
|
err := r.ensureFolderID()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
resp, err := sdk.Compute().Image().List(ctx, &compute.ListImagesRequest{
|
|
FolderId: r.FolderID(),
|
|
Filter: CreateResolverFilter("name", r.Name),
|
|
PageSize: DefaultResolverPageSize,
|
|
}, opts...)
|
|
return r.findName(resp.GetImages(), err)
|
|
}
|
|
|
|
type imageByFamilyResolver struct {
|
|
BaseNameResolver
|
|
}
|
|
|
|
func ImageByFamilyResolver(name string, opts ...ResolveOption) ycsdk.Resolver {
|
|
return &imageByFamilyResolver{
|
|
BaseNameResolver: NewBaseNameResolver(name, "image from family", opts...),
|
|
}
|
|
}
|
|
|
|
func (r *imageByFamilyResolver) Run(ctx context.Context, sdk *ycsdk.SDK, opts ...grpc.CallOption) error {
|
|
err := r.ensureFolderID()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
img, err := sdk.Compute().Image().GetLatestByFamily(ctx, &compute.GetImageLatestByFamilyRequest{
|
|
FolderId: r.FolderID(),
|
|
Family: r.Name,
|
|
})
|
|
if err != nil {
|
|
err = sdkerrors.WithMessagef(err, "failed to find image with family \"%v\"", r.Name)
|
|
}
|
|
return r.Set(img, err)
|
|
}
|