forked from rook/rook
88 lines
2.3 KiB
Go
88 lines
2.3 KiB
Go
/*
|
|
Copyright 2017 The Rook Authors. 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.
|
|
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 agent to manage Kubernetes storage attach events.
|
|
package agent
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/coreos/pkg/capnslog"
|
|
"github.com/rook/rook/pkg/clusterd"
|
|
"github.com/rook/rook/pkg/daemon/agent/cluster"
|
|
"github.com/rook/rook/pkg/daemon/agent/flexvolume"
|
|
"github.com/rook/rook/pkg/daemon/agent/flexvolume/attachment"
|
|
"github.com/rook/rook/pkg/daemon/agent/flexvolume/manager/ceph"
|
|
"k8s.io/api/core/v1"
|
|
)
|
|
|
|
var logger = capnslog.NewPackageLogger("github.com/rook/rook", "rook-agent")
|
|
|
|
// Agent represent all the references needed to manage a Rook agent
|
|
type Agent struct {
|
|
context *clusterd.Context
|
|
}
|
|
|
|
// New creates an Agent instance
|
|
func New(context *clusterd.Context) *Agent {
|
|
return &Agent{context: context}
|
|
}
|
|
|
|
// Run the agent
|
|
func (a *Agent) Run() error {
|
|
|
|
volumeAttachmentController, err := attachment.New(a.context)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create volume attachment controller: %+v", err)
|
|
}
|
|
|
|
volumeManager := ceph.NewVolumeManager(a.context)
|
|
|
|
flexvolumeController := flexvolume.NewController(a.context, volumeAttachmentController, volumeManager)
|
|
|
|
flexvolumeServer := flexvolume.NewFlexvolumeServer(
|
|
a.context,
|
|
flexvolumeController,
|
|
volumeManager,
|
|
)
|
|
|
|
flexvolumeServer.Start()
|
|
|
|
// create a cluster controller and tell it to start watching for changes to clusters
|
|
clusterController := cluster.NewClusterController(
|
|
a.context,
|
|
flexvolumeController,
|
|
volumeAttachmentController,
|
|
volumeManager)
|
|
stopChan := make(chan struct{})
|
|
clusterController.StartWatch(v1.NamespaceAll, stopChan)
|
|
|
|
sigc := make(chan os.Signal, 1)
|
|
signal.Notify(sigc, syscall.SIGTERM)
|
|
for {
|
|
select {
|
|
case <-sigc:
|
|
logger.Infof("shutdown signal received, exiting...")
|
|
flexvolumeServer.Stop()
|
|
close(stopChan)
|
|
return nil
|
|
}
|
|
}
|
|
}
|