ing
This commit is contained in:
commit
b1adb63ea8
68
.gitignore
vendored
Normal file
68
.gitignore
vendored
Normal file
|
@ -0,0 +1,68 @@
|
|||
# Created by .ignore support plugin (hsz.mobi)
|
||||
### JetBrains template
|
||||
# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and Webstorm
|
||||
# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
|
||||
|
||||
# User-specific stuff:
|
||||
.idea/**/workspace.xml
|
||||
.idea/**/tasks.xml
|
||||
.idea/dictionaries
|
||||
|
||||
# Sensitive or high-churn files:
|
||||
.idea/**/dataSources/
|
||||
.idea/**/dataSources.ids
|
||||
.idea/**/dataSources.xml
|
||||
.idea/**/dataSources.local.xml
|
||||
.idea/**/sqlDataSources.xml
|
||||
.idea/**/dynamic.xml
|
||||
.idea/**/uiDesigner.xml
|
||||
|
||||
# Gradle:
|
||||
.idea/**/gradle.xml
|
||||
.idea/**/libraries
|
||||
|
||||
# Mongo Explorer plugin:
|
||||
.idea/**/mongoSettings.xml
|
||||
|
||||
## File-based project format:
|
||||
*.iws
|
||||
|
||||
## Plugin-specific files:
|
||||
|
||||
# IntelliJ
|
||||
/out/
|
||||
|
||||
# mpeltonen/sbt-idea plugin
|
||||
.idea_modules/
|
||||
|
||||
# JIRA plugin
|
||||
atlassian-ide-plugin.xml
|
||||
|
||||
# Crashlytics plugin (for Android Studio and IntelliJ)
|
||||
com_crashlytics_export_strings.xml
|
||||
crashlytics.properties
|
||||
crashlytics-build.properties
|
||||
fabric.properties
|
||||
### Go template
|
||||
# Binaries for programs and plugins
|
||||
*.exe
|
||||
*.dll
|
||||
*.so
|
||||
*.dylib
|
||||
|
||||
# Test binary, build with `go test -c`
|
||||
*.test
|
||||
|
||||
# Output of the go coverage tool, specifically when used with LiteIDE
|
||||
*.out
|
||||
|
||||
# Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736
|
||||
.glide/
|
||||
.idea/
|
||||
*.iml
|
||||
|
||||
vendor/
|
||||
glide.lock
|
||||
.DS_Store
|
||||
dist/
|
||||
debug
|
32
.vscode/launch.json
vendored
Normal file
32
.vscode/launch.json
vendored
Normal file
|
@ -0,0 +1,32 @@
|
|||
{
|
||||
"version": "0.2.0",
|
||||
"configurations": [
|
||||
{
|
||||
"name": "Debug",
|
||||
"type": "go",
|
||||
"request": "launch",
|
||||
"mode": "debug",
|
||||
"remotePath": "",
|
||||
"port": 2345,
|
||||
"host": "127.0.0.1",
|
||||
"program": "${workspaceRoot}/main.go",
|
||||
"env": {},
|
||||
"args": [],
|
||||
"showLog": true
|
||||
},
|
||||
{
|
||||
"name": "File Debug",
|
||||
"type": "go",
|
||||
"request": "launch",
|
||||
"mode": "debug",
|
||||
"remotePath": "",
|
||||
"port": 2345,
|
||||
"host": "127.0.0.1",
|
||||
"program": "${fileDirname}",
|
||||
"env": {},
|
||||
"args": [],
|
||||
"showLog": true
|
||||
}
|
||||
|
||||
]
|
||||
}
|
3
.vscode/settings.json
vendored
Normal file
3
.vscode/settings.json
vendored
Normal file
|
@ -0,0 +1,3 @@
|
|||
// Place your settings in this file to overwrite default and user settings.
|
||||
{
|
||||
}
|
1
central/client.go
Normal file
1
central/client.go
Normal file
|
@ -0,0 +1 @@
|
|||
package central
|
9
config.json
Normal file
9
config.json
Normal file
|
@ -0,0 +1,9 @@
|
|||
{
|
||||
"domain": {
|
||||
"apikey": ""
|
||||
},
|
||||
"central": {
|
||||
"url": "ws://192.168.1.50:19190",
|
||||
"tls": false
|
||||
}
|
||||
}
|
11
glide.yaml
Normal file
11
glide.yaml
Normal file
|
@ -0,0 +1,11 @@
|
|||
package: git.loafle.net/overflow/overflow_probes
|
||||
import:
|
||||
- package: git.loafle.net/commons_go/config
|
||||
- package: git.loafle.net/commons_go/logging
|
||||
- package: github.com/gorilla/websocket
|
||||
version: v1.2.0
|
||||
- package: git.loafle.net/commons_go/util
|
||||
- package: github.com/shirou/gopsutil
|
||||
version: v2.17.08
|
||||
- package: github.com/takama/daemon
|
||||
version: 0.9.1
|
27
logging.json
Normal file
27
logging.json
Normal file
|
@ -0,0 +1,27 @@
|
|||
{
|
||||
"level": "debug",
|
||||
"development": true,
|
||||
"disableCaller": true,
|
||||
"disableStacktrace": true,
|
||||
"sampling": {
|
||||
"initial": 100,
|
||||
"thereafter": 100
|
||||
},
|
||||
"encoding": "console",
|
||||
"encoderConfig": {
|
||||
"messageKey": "message",
|
||||
"levelKey": "level",
|
||||
"timeKey": "time",
|
||||
"nameKey": "name",
|
||||
"callerKey": "caller",
|
||||
"stacktraceKey": "stacktrace",
|
||||
"lineEnding": "\n",
|
||||
"levelEncoder": "color",
|
||||
"timeEncoder": "ISO8601",
|
||||
"durationEncoder": "string",
|
||||
"callerEncoder": "full",
|
||||
"nameEncoder": "full"
|
||||
},
|
||||
"outputPaths": ["stdout", "/tmp/logs"],
|
||||
"errorOutputPaths": ["stderr"]
|
||||
}
|
179
main.go
Normal file
179
main.go
Normal file
|
@ -0,0 +1,179 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"git.loafle.net/commons_go/logging"
|
||||
"git.loafle.net/overflow/overflow_probes/probe"
|
||||
"github.com/takama/daemon"
|
||||
)
|
||||
|
||||
const (
|
||||
version = "1.0.0"
|
||||
website = "https://www.overflow.cloud"
|
||||
banner = `
|
||||
|
||||
██████╗ ██╗ ██╗███████╗██████╗ ███████╗██╗ ██████╗ ██╗ ██╗
|
||||
██╔═══██╗██║ ██║██╔════╝██╔══██╗██╔════╝██║ ██╔═══██╗██║ ██║
|
||||
██║ ██║██║ ██║█████╗ ██████╔╝█████╗ ██║ ██║ ██║██║ █╗ ██║
|
||||
██║ ██║╚██╗ ██╔╝██╔══╝ ██╔══██╗██╔══╝ ██║ ██║ ██║██║███╗██║
|
||||
╚██████╔╝ ╚████╔╝ ███████╗██║ ██║██║ ███████╗╚██████╔╝╚███╔███╔╝
|
||||
╚═════╝ ╚═══╝ ╚══════╝╚═╝ ╚═╝╚═╝ ╚══════╝ ╚═════╝ ╚══╝╚══╝
|
||||
|
||||
`
|
||||
)
|
||||
|
||||
const (
|
||||
// name of the service
|
||||
serviceName = "Probe"
|
||||
serviceDescription = "Probe Service of overFlow"
|
||||
)
|
||||
|
||||
type daemonHandler struct {
|
||||
daemon.Daemon
|
||||
}
|
||||
|
||||
// Manage by daemon commands or run the daemon
|
||||
// cmd install -config=./path
|
||||
// cmd install
|
||||
// cmd remove
|
||||
// cmd start
|
||||
// cmd stop
|
||||
// cmd status
|
||||
// cmd -config=./path
|
||||
// cmd
|
||||
func (d *daemonHandler) Manage() (isRunning bool, status string, err error) {
|
||||
|
||||
isRunning = true
|
||||
|
||||
if nil != probe.Args.Daemon {
|
||||
switch *probe.Args.Daemon {
|
||||
case "install":
|
||||
var runArgs = []string{}
|
||||
runArgs = append(runArgs, fmt.Sprintf("-config=%s", *probe.Args.ConfigPath))
|
||||
|
||||
isRunning = false
|
||||
status, err = d.Install(runArgs...)
|
||||
case "remove":
|
||||
isRunning = false
|
||||
status, err = d.Remove()
|
||||
case "start":
|
||||
isRunning = false
|
||||
status, err = d.Start()
|
||||
case "stop":
|
||||
isRunning = false
|
||||
status, err = d.Stop()
|
||||
case "status":
|
||||
isRunning = false
|
||||
status, err = d.Status()
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func init() {
|
||||
flag.Usage = func() {
|
||||
fmt.Printf("Usage of %s\n", os.Args[0])
|
||||
fmt.Printf(" [install | remove | start | stop | status]\n")
|
||||
flag.PrintDefaults()
|
||||
}
|
||||
|
||||
if len(os.Args) > 1 {
|
||||
command := os.Args[1]
|
||||
switch command {
|
||||
case "install", "remove", "start", "stop", "status":
|
||||
*probe.Args.Daemon = command
|
||||
}
|
||||
}
|
||||
|
||||
probe.Args.ConfigPath = flag.String("config", ".", "The path of config")
|
||||
flag.Parse()
|
||||
|
||||
}
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
var srv daemon.Daemon
|
||||
var status string
|
||||
isRunning := true
|
||||
|
||||
defer logging.Logger.Sync()
|
||||
|
||||
fmt.Println(banner)
|
||||
fmt.Printf("Version: %s\n", version)
|
||||
fmt.Printf("URL: %s\n", website)
|
||||
fmt.Println()
|
||||
|
||||
if srv, err = daemon.New(serviceName, serviceDescription); nil != err {
|
||||
logging.Logger.Panic(fmt.Sprintf("Probe: %v", err))
|
||||
}
|
||||
|
||||
s := &daemonHandler{srv}
|
||||
if isRunning, status, err = s.Manage(); nil != err {
|
||||
logging.Logger.Error(fmt.Sprintf("Probe: status[%s] error: %v", status, err))
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if !isRunning {
|
||||
logging.Logger.Info(fmt.Sprintf("Probe: status[%s]", status))
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
// // Do something, call your goroutines, etc
|
||||
p := probe.New()
|
||||
|
||||
go func() {
|
||||
if err := p.Start(); err != nil {
|
||||
logging.Logger.Error(fmt.Sprintf("Probe: cannot start probe error: %v", err))
|
||||
}
|
||||
}()
|
||||
|
||||
// // Set up channel on which to send signal notifications.
|
||||
// // We must use a buffered channel or risk missing the signal
|
||||
// // if we're not ready to receive when the signal is sent.
|
||||
interrupt := make(chan os.Signal, 1)
|
||||
signal.Notify(interrupt,
|
||||
syscall.SIGKILL,
|
||||
syscall.SIGSTOP,
|
||||
syscall.SIGHUP,
|
||||
syscall.SIGINT,
|
||||
syscall.SIGTERM,
|
||||
syscall.SIGQUIT)
|
||||
|
||||
<-interrupt
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
if err := p.Shutdown(ctx); err != nil {
|
||||
logging.Logger.Error(fmt.Sprintf("Probe: status[%s] error: %v", status, err))
|
||||
}
|
||||
|
||||
// // loop work cycle with accept connections or interrupt
|
||||
// // by system signal
|
||||
// ListenLoop:
|
||||
// for {
|
||||
// select {
|
||||
// case s := <-interrupt:
|
||||
// logging.Logger.Info(fmt.Sprintf("Probe: signal[%v]", s))
|
||||
// ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
// defer cancel()
|
||||
// if err := p.Shutdown(ctx); err != nil {
|
||||
// logging.Logger.Error(fmt.Sprintf("Probe: status[%s] error: %v", status, err))
|
||||
// }
|
||||
|
||||
// if s == os.Interrupt {
|
||||
// logging.Logger.Info("Probe was interruped by system signal")
|
||||
// } else {
|
||||
// logging.Logger.Info("Probe was killed")
|
||||
// }
|
||||
// break ListenLoop
|
||||
// }
|
||||
// }
|
||||
|
||||
}
|
3
probe.json
Normal file
3
probe.json
Normal file
|
@ -0,0 +1,3 @@
|
|||
{
|
||||
"id": ""
|
||||
}
|
63
probe/handler/auth/auth_handler.go
Normal file
63
probe/handler/auth/auth_handler.go
Normal file
|
@ -0,0 +1,63 @@
|
|||
package auth
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
|
||||
"git.loafle.net/commons_go/logging"
|
||||
"git.loafle.net/overflow/overflow_probes/probe/handler"
|
||||
|
||||
"github.com/gorilla/websocket"
|
||||
)
|
||||
|
||||
type AuthHandler interface {
|
||||
handler.Handler
|
||||
}
|
||||
|
||||
type authHandlers struct {
|
||||
c *websocket.Conn
|
||||
apiKey *string
|
||||
tempKey *string
|
||||
}
|
||||
|
||||
type NoAuthProbe struct {
|
||||
Host *Host `json:"host"`
|
||||
Network *Network `json:"network"`
|
||||
}
|
||||
|
||||
func NewHandler() AuthHandler {
|
||||
h := &authHandlers{}
|
||||
return h
|
||||
}
|
||||
|
||||
func (h *authHandlers) Start() {
|
||||
// c, _, err := websocket.DefaultDialer.Dial("ws://192.168.1.50:19190/auth", nil)
|
||||
// if err != nil {
|
||||
// logging.Logger.Fatal(fmt.Sprintf("auth: %v", err))
|
||||
// }
|
||||
// h.c = c
|
||||
p := &NoAuthProbe{}
|
||||
|
||||
if h, err := getHost(); nil == err {
|
||||
p.Host = h
|
||||
}
|
||||
|
||||
if n, err := getNetwork(); nil == err {
|
||||
p.Network = n
|
||||
}
|
||||
|
||||
if buf, err := json.Marshal(*p); nil == err {
|
||||
logging.Logger.Debug(fmt.Sprintf("p: %s", string(buf)))
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func (h *authHandlers) listen() {
|
||||
// 1. regist
|
||||
// 2. wait for accept auth
|
||||
}
|
||||
|
||||
func (h *authHandlers) Shutdown(ctx context.Context) {
|
||||
|
||||
}
|
32
probe/handler/auth/host.go
Normal file
32
probe/handler/auth/host.go
Normal file
|
@ -0,0 +1,32 @@
|
|||
package auth
|
||||
|
||||
import (
|
||||
"github.com/shirou/gopsutil/host"
|
||||
)
|
||||
|
||||
type Host struct {
|
||||
Name string `json:"name"`
|
||||
OS string `json:"os"`
|
||||
Platform string `json:"paltform"`
|
||||
PlatformFamily string `json:"platformFamily"`
|
||||
PlatformVersion string `json:"platformVersion"`
|
||||
KernelVersion string `json:"kernelVersion"`
|
||||
HostID string `json:"hostID"`
|
||||
}
|
||||
|
||||
func getHost() (*Host, error) {
|
||||
h := &Host{}
|
||||
|
||||
if i, err := host.Info(); nil == err {
|
||||
h.Name = i.Hostname
|
||||
h.OS = i.OS
|
||||
h.Platform = i.Platform
|
||||
h.PlatformFamily = i.PlatformFamily
|
||||
h.KernelVersion = i.KernelVersion
|
||||
h.HostID = i.HostID
|
||||
} else {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return h, nil
|
||||
}
|
64
probe/handler/auth/network.go
Normal file
64
probe/handler/auth/network.go
Normal file
|
@ -0,0 +1,64 @@
|
|||
package auth
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"net"
|
||||
|
||||
"git.loafle.net/commons_go/util/net/gateway"
|
||||
)
|
||||
|
||||
type Network struct {
|
||||
Name string `json:"name"`
|
||||
Address string `json:"address"`
|
||||
Gateway string `json:"gateway"`
|
||||
MacAddress string `json:"macAddress"`
|
||||
}
|
||||
|
||||
func getNetwork() (*Network, error) {
|
||||
var ip net.IP
|
||||
var iface string
|
||||
var err error
|
||||
if ip, iface, err = gateway.DiscoverGateway(); nil != err {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
interfaces, err := net.Interfaces()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
idx := -1
|
||||
|
||||
for _idx, i := range interfaces {
|
||||
if i.Name == iface {
|
||||
idx = _idx
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if -1 == idx {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
n := &Network{}
|
||||
i := interfaces[idx]
|
||||
|
||||
n.Name = i.Name
|
||||
n.MacAddress = i.HardwareAddr.String()
|
||||
n.Gateway = ip.String()
|
||||
|
||||
if addrs, err := i.Addrs(); nil == err {
|
||||
var buffer bytes.Buffer
|
||||
for _idx, a := range addrs {
|
||||
if 0 < _idx {
|
||||
buffer.WriteString("|")
|
||||
}
|
||||
buffer.WriteString(a.String())
|
||||
}
|
||||
n.Address = buffer.String()
|
||||
} else {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return n, nil
|
||||
}
|
8
probe/handler/handler.go
Normal file
8
probe/handler/handler.go
Normal file
|
@ -0,0 +1,8 @@
|
|||
package handler
|
||||
|
||||
import "context"
|
||||
|
||||
type Handler interface {
|
||||
Start()
|
||||
Shutdown(ctx context.Context)
|
||||
}
|
22
probe/handler/probe/probe_handler.go
Normal file
22
probe/handler/probe/probe_handler.go
Normal file
|
@ -0,0 +1,22 @@
|
|||
package handler
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"git.loafle.net/overflow/overflow_probes/probe/handler"
|
||||
)
|
||||
|
||||
type ProbeHandler interface {
|
||||
handler.Handler
|
||||
}
|
||||
|
||||
type probeHandlers struct {
|
||||
}
|
||||
|
||||
func (h *probeHandlers) Start() {
|
||||
|
||||
}
|
||||
|
||||
func (h *probeHandlers) Shutdown(ctx context.Context) {
|
||||
|
||||
}
|
75
probe/probe.go
Normal file
75
probe/probe.go
Normal file
|
@ -0,0 +1,75 @@
|
|||
package probe
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"git.loafle.net/commons_go/config"
|
||||
"git.loafle.net/overflow/overflow_probes/probe/handler"
|
||||
"git.loafle.net/overflow/overflow_probes/probe/handler/auth"
|
||||
)
|
||||
|
||||
type Arguments struct {
|
||||
Daemon *string
|
||||
ConfigPath *string
|
||||
}
|
||||
|
||||
var Args Arguments
|
||||
|
||||
func New() Probe {
|
||||
p := &probe{}
|
||||
|
||||
return p
|
||||
}
|
||||
|
||||
type Probe interface {
|
||||
Start() error
|
||||
Shutdown(ctx context.Context) error
|
||||
}
|
||||
|
||||
type probe struct {
|
||||
handler handler.Handler
|
||||
probeConfig config.Configurator
|
||||
}
|
||||
|
||||
func (p *probe) Start() error {
|
||||
// conf := loadConfig(*configPath)
|
||||
probeConf := loadProbeConfig(*Args.ConfigPath)
|
||||
|
||||
probeID := probeConf.GetString("id")
|
||||
|
||||
if "" == probeID {
|
||||
a := auth.NewHandler()
|
||||
a.Start()
|
||||
} else {
|
||||
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *probe) Shutdown(ctx context.Context) error {
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func loadConfig(path string) config.Configurator {
|
||||
conf := config.New()
|
||||
conf.SetConfigName("config")
|
||||
conf.AddConfigPath(path)
|
||||
err := conf.ReadInConfig()
|
||||
if nil != err {
|
||||
panic(err)
|
||||
}
|
||||
return conf
|
||||
}
|
||||
|
||||
func loadProbeConfig(path string) config.Configurator {
|
||||
conf := config.New()
|
||||
conf.SetConfigName("probe")
|
||||
conf.AddConfigPath(path)
|
||||
err := conf.ReadInConfig()
|
||||
if nil != err {
|
||||
panic(err)
|
||||
}
|
||||
return conf
|
||||
}
|
Loading…
Reference in New Issue
Block a user