pod.pb.access.go

// Code generated by protoc-gen-goten-resource
// Resource: Pod
// DO NOT EDIT!!!

package pod

import (
	"context"

	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"

	"github.com/cloudwan/goten-sdk/runtime/api/watch_type"
	gotenobject "github.com/cloudwan/goten-sdk/runtime/object"
	gotenresource "github.com/cloudwan/goten-sdk/runtime/resource"
)

// proto imports
import (
	project "github.com/cloudwan/edgelq-sdk/applications/resources/v1alpha/project"
	ntt_meta "github.com/cloudwan/edgelq-sdk/common/types/meta"
	devices_device "github.com/cloudwan/edgelq-sdk/devices/resources/v1alpha/device"
	timestamp "github.com/golang/protobuf/ptypes/timestamp"
)

// ensure the imports are used
var (
	_ = new(context.Context)

	_ = codes.Internal
	_ = status.Status{}

	_ = watch_type.WatchType_STATEFUL
	_ = new(gotenobject.FieldPath)
	_ = new(gotenresource.ListQuery)
)

// make sure we're using proto imports
var (
	_ = &project.Project{}
	_ = &ntt_meta.Meta{}
	_ = &devices_device.Device{}
	_ = &timestamp.Timestamp{}
)

type PodAccess interface {
	GetPod(context.Context, *GetQuery) (*Pod, error)
	BatchGetPods(context.Context, []*Reference, ...gotenresource.BatchGetOption) error
	QueryPods(context.Context, *ListQuery) (*QueryResultSnapshot, error)
	WatchPod(context.Context, *GetQuery, func(*PodChange) error) error
	WatchPods(context.Context, *WatchQuery, func(*QueryResultChange) error) error
	SavePod(context.Context, *Pod, ...gotenresource.SaveOption) error
	DeletePod(context.Context, *Reference, ...gotenresource.DeleteOption) error
}

type anyCastAccess struct {
	PodAccess
}

func AsAnyCastAccess(access PodAccess) gotenresource.Access {
	return &anyCastAccess{PodAccess: access}
}

func (a *anyCastAccess) Get(ctx context.Context, q gotenresource.GetQuery) (gotenresource.Resource, error) {
	if asPodQuery, ok := q.(*GetQuery); ok {
		return a.GetPod(ctx, asPodQuery)
	}
	return nil, status.Errorf(codes.Internal,
		"Unrecognized descriptor, expected Pod, got: %s",
		q.GetResourceDescriptor().GetResourceTypeName().FullyQualifiedTypeName())
}

func (a *anyCastAccess) Query(ctx context.Context, q gotenresource.ListQuery) (gotenresource.QueryResultSnapshot, error) {
	if asPodQuery, ok := q.(*ListQuery); ok {
		return a.QueryPods(ctx, asPodQuery)
	}
	return nil, status.Errorf(codes.Internal,
		"Unrecognized descriptor, expected Pod, got: %s",
		q.GetResourceDescriptor().GetResourceTypeName().FullyQualifiedTypeName())
}

func (a *anyCastAccess) Search(ctx context.Context, q gotenresource.SearchQuery) (gotenresource.QueryResultSnapshot, error) {
	return nil, status.Errorf(codes.Internal, "Search is not available for Pod")
}

func (a *anyCastAccess) Watch(ctx context.Context, q gotenresource.GetQuery, cb func(ch gotenresource.ResourceChange) error) error {
	if asPodQuery, ok := q.(*GetQuery); ok {
		return a.WatchPod(ctx, asPodQuery, func(change *PodChange) error {
			return cb(change)
		})
	}
	return status.Errorf(codes.Internal,
		"Unrecognized descriptor, expected Pod, got: %s",
		q.GetResourceDescriptor().GetResourceTypeName().FullyQualifiedTypeName())
}

func (a *anyCastAccess) WatchQuery(ctx context.Context, q gotenresource.WatchQuery, cb func(ch gotenresource.QueryResultChange) error) error {
	if asPodQuery, ok := q.(*WatchQuery); ok {
		return a.WatchPods(ctx, asPodQuery, func(change *QueryResultChange) error {
			return cb(change)
		})
	}
	return status.Errorf(codes.Internal,
		"Unrecognized descriptor, expected Pod, got: %s",
		q.GetResourceDescriptor().GetResourceTypeName().FullyQualifiedTypeName())
}

func (a *anyCastAccess) Save(ctx context.Context, res gotenresource.Resource, opts ...gotenresource.SaveOption) error {
	if asPodRes, ok := res.(*Pod); ok {
		return a.SavePod(ctx, asPodRes, opts...)
	}
	return status.Errorf(codes.Internal,
		"Unrecognized descriptor, expected Pod, got: %s",
		res.GetResourceDescriptor().GetResourceTypeName().FullyQualifiedTypeName())
}

func (a *anyCastAccess) Delete(ctx context.Context, ref gotenresource.Reference, opts ...gotenresource.DeleteOption) error {
	if asPodRef, ok := ref.(*Reference); ok {
		return a.DeletePod(ctx, asPodRef, opts...)
	}
	return status.Errorf(codes.Internal,
		"Unrecognized descriptor, expected Pod, got: %s",
		ref.GetResourceDescriptor().GetResourceTypeName().FullyQualifiedTypeName())
}

func (a *anyCastAccess) BatchGet(ctx context.Context, toGet []gotenresource.Reference, opts ...gotenresource.BatchGetOption) error {
	podRefs := make([]*Reference, 0, len(toGet))
	for _, ref := range toGet {
		if asPodRef, ok := ref.(*Reference); !ok {
			return status.Errorf(codes.Internal,
				"Unrecognized descriptor, expected Pod, got: %s",
				ref.GetResourceDescriptor().GetResourceTypeName().FullyQualifiedTypeName())
		} else {
			podRefs = append(podRefs, asPodRef)
		}
	}
	return a.BatchGetPods(ctx, podRefs, opts...)
}

func (a *anyCastAccess) GetResourceDescriptors() []gotenresource.Descriptor {
	return []gotenresource.Descriptor{
		GetDescriptor(),
	}
}