func main() {
logger := lager.NewLogger("backup")
logger.RegisterSink(log.NewCliSink(lager.INFO))
logger.RegisterSink(lager.NewWriterSink(os.Stdout, lager.INFO))
var configPath string
flag.StringVar(&configPath, "config", "", "Path to YML config file")
flag.Parse()
if _, err := os.Stat(configPath); os.IsNotExist(err) {
logger.Error("snapshot-main", err, lager.Data{"event": "failed", "config_file": configPath})
os.Exit(2)
}
backupConfig, err := backup.LoadBackupConfig(configPath)
exitOnError(err, 78, logger, configPath)
logFile, err := os.OpenFile(backupConfig.LogFilepath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0660)
if err != nil {
logger.Error("snapshot-main", err, lager.Data{"event": "failed", "LogFilepath": backupConfig.LogFilepath})
}
logger.RegisterSink(lager.NewWriterSink(logFile, lager.INFO))
// TODO: Should this really be a pointer not a struct?
if (backupConfig.S3Config == backup.S3Configuration{}) {
logger.Info("snapshot-main", lager.Data{"event": "cancelled", "message": "S3 configuration not found - skipping backup"})
os.Exit(0)
}
backuper, err := backup.NewInstanceBackuper(*backupConfig, logger)
exitOnError(err, 70, logger, configPath)
backupResults := backuper.Backup()
hasError := false
for _, result := range backupResults {
if result.Err != nil {
hasError = true
logger.Error(
"snapshot-main",
result.Err,
lager.Data{
"event": "failed",
"InstanceID": result.InstanceID,
"RedisConfigPath": result.RedisConfigPath,
"NodeIP": result.NodeIP,
},
)
}
}
if hasError {
os.Exit(70)
}
}
. "github.com/onsi/gomega"
"github.com/pivotal-cf/cf-redis-broker/log"
"github.com/pivotal-golang/lager"
)
var _ = Describe("CliSink", func() {
var (
stdoutReader io.Reader
originalStdout *os.File
sink lager.Sink
)
BeforeEach(func() {
originalStdout = os.Stdout
stdoutReader, os.Stdout, _ = os.Pipe()
sink = log.NewCliSink(lager.INFO)
})
AfterEach(func() {
os.Stdout = originalStdout
})
Describe(".Log", func() {
Context("when the payload doesn't have lager data with event key", func() {
It("does not write to stdout", func() {
sink.Log(lager.INFO, []byte("Logging to stdout"))
os.Stdout.Close()
output, _ := ioutil.ReadAll(stdoutReader)
Expect(string(output)).To(Equal(""))
})