116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
package leases
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"time"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
"github.com/boltdb/bolt"
|
|
api "github.com/containerd/containerd/api/services/leases/v1"
|
|
"github.com/containerd/containerd/metadata"
|
|
"github.com/containerd/containerd/plugin"
|
|
ptypes "github.com/gogo/protobuf/types"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func init() {
|
|
plugin.Register(&plugin.Registration{
|
|
Type: plugin.GRPCPlugin,
|
|
ID: "leases",
|
|
Requires: []plugin.Type{
|
|
plugin.MetadataPlugin,
|
|
},
|
|
InitFn: func(ic *plugin.InitContext) (interface{}, error) {
|
|
m, err := ic.Get(plugin.MetadataPlugin)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewService(m.(*metadata.DB)), nil
|
|
},
|
|
})
|
|
}
|
|
|
|
type service struct {
|
|
db *metadata.DB
|
|
}
|
|
|
|
// NewService returns the GRPC metadata server
|
|
func NewService(db *metadata.DB) api.LeasesServer {
|
|
return &service{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (s *service) Register(server *grpc.Server) error {
|
|
api.RegisterLeasesServer(server, s)
|
|
return nil
|
|
}
|
|
|
|
func (s *service) Create(ctx context.Context, r *api.CreateRequest) (*api.CreateResponse, error) {
|
|
lid := r.ID
|
|
if lid == "" {
|
|
lid = generateLeaseID()
|
|
}
|
|
var trans metadata.Lease
|
|
if err := s.db.Update(func(tx *bolt.Tx) error {
|
|
var err error
|
|
trans, err = metadata.NewLeaseManager(tx).Create(ctx, lid, r.Labels)
|
|
return err
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
return &api.CreateResponse{
|
|
Lease: txToGRPC(trans),
|
|
}, nil
|
|
}
|
|
|
|
func (s *service) Delete(ctx context.Context, r *api.DeleteRequest) (*ptypes.Empty, error) {
|
|
if err := s.db.Update(func(tx *bolt.Tx) error {
|
|
return metadata.NewLeaseManager(tx).Delete(ctx, r.ID)
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
return &ptypes.Empty{}, nil
|
|
}
|
|
|
|
func (s *service) List(ctx context.Context, r *api.ListRequest) (*api.ListResponse, error) {
|
|
var leases []metadata.Lease
|
|
if err := s.db.View(func(tx *bolt.Tx) error {
|
|
var err error
|
|
leases, err = metadata.NewLeaseManager(tx).List(ctx, false, r.Filters...)
|
|
return err
|
|
}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
apileases := make([]*api.Lease, len(leases))
|
|
for i := range leases {
|
|
apileases[i] = txToGRPC(leases[i])
|
|
}
|
|
|
|
return &api.ListResponse{
|
|
Leases: apileases,
|
|
}, nil
|
|
}
|
|
|
|
func txToGRPC(tx metadata.Lease) *api.Lease {
|
|
return &api.Lease{
|
|
ID: tx.ID,
|
|
Labels: tx.Labels,
|
|
CreatedAt: tx.CreatedAt,
|
|
// TODO: Snapshots
|
|
// TODO: Content
|
|
}
|
|
}
|
|
|
|
func generateLeaseID() string {
|
|
t := time.Now()
|
|
var b [3]byte
|
|
// Ignore read failures, just decreases uniqueness
|
|
rand.Read(b[:])
|
|
return fmt.Sprintf("%d-%s", t.Nanosecond(), base64.URLEncoding.EncodeToString(b[:]))
|
|
}
|