Move services to plugins/services

Signed-off-by: Derek McGowan <derek@mcg.dev>

plugins/services/streaming/service.go

@@ -0,0 +1,136 @@
+/*
+   Copyright The containerd Authors.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+package streaming
+
+import (
+	"errors"
+	"io"
+
+	api "github.com/containerd/containerd/v2/api/services/streaming/v1"
+	"github.com/containerd/containerd/v2/errdefs"
+	"github.com/containerd/containerd/v2/pkg/streaming"
+	"github.com/containerd/containerd/v2/plugins"
+	"github.com/containerd/containerd/v2/protobuf"
+	ptypes "github.com/containerd/containerd/v2/protobuf/types"
+	"github.com/containerd/log"
+	"github.com/containerd/plugin"
+	"github.com/containerd/plugin/registry"
+	"github.com/containerd/typeurl/v2"
+	"google.golang.org/grpc"
+)
+
+var emptyResponse typeurl.Any
+
+func init() {
+	// save marshalled empty response to avoid marshaling everytime
+	var err error
+	emptyResponse, err = typeurl.MarshalAny(&ptypes.Empty{})
+	if err != nil {
+		panic(err)
+	}
+
+	registry.Register(&plugin.Registration{
+		Type: plugins.GRPCPlugin,
+		ID:   "streaming",
+		Requires: []plugin.Type{
+			plugins.StreamingPlugin,
+		},
+		InitFn: func(ic *plugin.InitContext) (interface{}, error) {
+			i, err := ic.GetByID(plugins.StreamingPlugin, "manager")
+			if err != nil {
+				return nil, err
+			}
+			return &service{manager: i.(streaming.StreamManager)}, nil
+		},
+	})
+}
+
+type service struct {
+	manager streaming.StreamManager
+	api.UnimplementedStreamingServer
+}
+
+func (s *service) Register(server *grpc.Server) error {
+	api.RegisterStreamingServer(server, s)
+	return nil
+}
+
+func (s *service) Stream(srv api.Streaming_StreamServer) error {
+	// TODO: Timeout waiting
+	a, err := srv.Recv()
+	if err != nil {
+		return err
+	}
+	var i api.StreamInit
+	if err := typeurl.UnmarshalTo(a, &i); err != nil {
+		return err
+	}
+
+	cc := make(chan struct{})
+	ss := &serviceStream{
+		s:  srv,
+		cc: cc,
+	}
+
+	log.G(srv.Context()).WithField("stream", i.ID).Debug("registering stream")
+	if err := s.manager.Register(srv.Context(), i.ID, ss); err != nil {
+		return err
+	}
+
+	// Send response packet after registering stream
+	if err := srv.Send(protobuf.FromAny(emptyResponse)); err != nil {
+		return err
+	}
+
+	select {
+	case <-srv.Context().Done():
+		// TODO: Should return error if not cancelled?
+	case <-cc:
+	}
+
+	return nil
+}
+
+type serviceStream struct {
+	s  api.Streaming_StreamServer
+	cc chan struct{}
+}
+
+func (ss *serviceStream) Send(a typeurl.Any) (err error) {
+	err = errdefs.FromGRPC(ss.s.Send(protobuf.FromAny(a)))
+	if !errors.Is(err, io.EOF) {
+		err = errdefs.FromGRPC(err)
+	}
+	return
+}
+
+func (ss *serviceStream) Recv() (a typeurl.Any, err error) {
+	a, err = ss.s.Recv()
+	if !errors.Is(err, io.EOF) {
+		err = errdefs.FromGRPC(err)
+	}
+	return
+}
+
+func (ss *serviceStream) Close() error {
+	select {
+	case <-ss.cc:
+	default:
+		close(ss.cc)
+	}
+	return nil
+}