Move beater interface to cmd package

This commit is contained in:
Blerim Sheqa 2017-10-17 10:19:42 +02:00
parent dd3104e1c6
commit 136ab9b250
2 changed files with 14 additions and 5 deletions

12
cmd/root.go Normal file
View File

@ -0,0 +1,12 @@
package cmd
import (
cmd "github.com/elastic/beats/libbeat/cmd"
"github.com/icinga/icingabeat/beater"
)
// Name of this beat
var Name = "icingabeat"
// RootCmd to handle beats cli
var RootCmd = cmd.GenRootCmd(Name, "", beater.New)

View File

@ -3,14 +3,11 @@ package main
import (
"os"
"github.com/elastic/beats/libbeat/beat"
"github.com/icinga/icingabeat/beater"
"github.com/icinga/icingabeat/cmd"
)
func main() {
err := beat.Run("icingabeat", "", beater.New)
if err != nil {
if err := cmd.RootCmd.Execute(); err != nil {
os.Exit(1)
}
}