87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
// Copyright 2015 clair 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 clair implements the ability to boot Clair with your own imports
|
|
// that can dynamically register additional functionality.
|
|
package clair
|
|
|
|
import (
|
|
"math/rand"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/coreos/clair/api"
|
|
"github.com/coreos/clair/api/context"
|
|
"github.com/coreos/clair/config"
|
|
"github.com/coreos/clair/database"
|
|
"github.com/coreos/clair/database/mysql"
|
|
"github.com/coreos/clair/database/pgsql"
|
|
"github.com/coreos/clair/notifier"
|
|
"github.com/coreos/clair/updater"
|
|
"github.com/coreos/clair/utils"
|
|
"github.com/coreos/pkg/capnslog"
|
|
)
|
|
|
|
var log = capnslog.NewPackageLogger("github.com/coreos/clair", "main")
|
|
|
|
// Boot starts Clair. By exporting this function, anyone can import their own
|
|
// custom fetchers/updaters into their own package and then call clair.Boot.
|
|
func Boot(config *config.Config) {
|
|
rand.Seed(time.Now().UnixNano())
|
|
st := utils.NewStopper()
|
|
var (
|
|
db database.Datastore
|
|
err error
|
|
)
|
|
// Open database
|
|
if strings.HasPrefix(config.Database.Source, "postgres") {
|
|
db, err = pgsql.Open(config.Database)
|
|
} else if strings.HasPrefix(config.Database.Source, "mysql") {
|
|
db, err = mysql.Open(config.Database)
|
|
} else {
|
|
log.Fatal("database source '%s' does not support, support 'postgres' and 'mysql' for now", config.Database.Source)
|
|
}
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
// Start notifier
|
|
st.Begin()
|
|
go notifier.Run(config.Notifier, db, st)
|
|
|
|
// Start API
|
|
st.Begin()
|
|
go api.Run(config.API, &context.RouteContext{db, config.API}, st)
|
|
st.Begin()
|
|
go api.RunHealth(config.API, &context.RouteContext{db, config.API}, st)
|
|
|
|
// Start updater
|
|
st.Begin()
|
|
go updater.Run(config.Updater, db, st)
|
|
|
|
// Wait for interruption and shutdown gracefully.
|
|
waitForSignals(syscall.SIGINT, syscall.SIGTERM)
|
|
log.Info("Received interruption, gracefully stopping ...")
|
|
st.Stop()
|
|
}
|
|
|
|
func waitForSignals(signals ...os.Signal) {
|
|
interrupts := make(chan os.Signal, 1)
|
|
signal.Notify(interrupts, signals...)
|
|
<-interrupts
|
|
}
|