/
main.go
93 lines (75 loc) · 2.38 KB
/
main.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
84
85
86
87
88
89
90
91
92
93
// Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
// http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file 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 main
import (
"context"
"net"
"os"
"os/signal"
"syscall"
"github.com/awslabs/containerd-firecracker/snapshotter"
snapshotsapi "github.com/containerd/containerd/api/services/snapshots/v1"
"github.com/containerd/containerd/contrib/snapshotservice"
"github.com/containerd/containerd/log"
"golang.org/x/sync/errgroup"
"google.golang.org/grpc"
)
func main() {
if len(os.Args) < 3 {
log.L.Fatalf("invalid args: usage %s <unix addr> <root>", os.Args[0])
}
var unixAddr, rootPath = os.Args[1], os.Args[2]
stop := make(chan os.Signal, 1)
signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM, syscall.SIGPIPE, syscall.SIGHUP, syscall.SIGQUIT)
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
group, ctx := errgroup.WithContext(ctx)
rpc := grpc.NewServer()
snap, err := snapshotter.NewSnapshotter(ctx, rootPath)
if err != nil {
log.G(ctx).WithError(err).Fatal("failed to create snapshotter")
}
// Convert the snapshotter interface to gRPC service and run server
log.G(ctx).WithField("unix_addr", unixAddr).Info("running gRPC server")
service := snapshotservice.FromSnapshotter(snap)
snapshotsapi.RegisterSnapshotsServer(rpc, service)
listener, err := net.Listen("unix", unixAddr)
if err != nil {
log.G(ctx).WithError(err).Fatalf("failed to listen socket at %s", os.Args[1])
}
group.Go(func() error {
return rpc.Serve(listener)
})
group.Go(func() error {
defer func() {
log.G(ctx).Info("stopping server")
rpc.Stop()
if err := snap.Close(); err != nil {
log.G(ctx).WithError(err).Error("failed to close snapshotter")
}
}()
for {
select {
case <-stop:
cancel()
return nil
case <-ctx.Done():
return ctx.Err()
}
}
})
if err := group.Wait(); err != nil {
log.G(ctx).WithError(err).Warn("snapshotter error")
}
log.G(ctx).Info("done")
}