Compare commits
1 Commits
v1.2.0
..
bece440c47
| Author | SHA1 | Date | |
|---|---|---|---|
| bece440c47 |
@@ -1,23 +1,20 @@
|
|||||||
module github.com/loveuer/go-alived
|
module github.com/loveuer/go-alived
|
||||||
|
|
||||||
go 1.24.0
|
go 1.25.0
|
||||||
|
|
||||||
require (
|
|
||||||
github.com/mdlayher/arp v0.0.0-20220512170110-6706a2966875
|
|
||||||
github.com/spf13/cobra v1.10.2
|
|
||||||
github.com/vishvananda/netlink v1.3.1
|
|
||||||
golang.org/x/net v0.47.0
|
|
||||||
gopkg.in/yaml.v3 v3.0.1
|
|
||||||
)
|
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/inconshreveable/mousetrap v1.1.0 // indirect
|
github.com/inconshreveable/mousetrap v1.1.0 // indirect
|
||||||
github.com/josharian/native v1.0.0 // indirect
|
github.com/josharian/native v1.0.0 // indirect
|
||||||
|
github.com/mdlayher/arp v0.0.0-20220512170110-6706a2966875 // indirect
|
||||||
github.com/mdlayher/ethernet v0.0.0-20220221185849-529eae5b6118 // indirect
|
github.com/mdlayher/ethernet v0.0.0-20220221185849-529eae5b6118 // indirect
|
||||||
github.com/mdlayher/packet v1.0.0 // indirect
|
github.com/mdlayher/packet v1.0.0 // indirect
|
||||||
github.com/mdlayher/socket v0.2.1 // indirect
|
github.com/mdlayher/socket v0.2.1 // indirect
|
||||||
|
github.com/spf13/cobra v1.10.2 // indirect
|
||||||
github.com/spf13/pflag v1.0.9 // indirect
|
github.com/spf13/pflag v1.0.9 // indirect
|
||||||
|
github.com/vishvananda/netlink v1.3.1 // indirect
|
||||||
github.com/vishvananda/netns v0.0.5 // indirect
|
github.com/vishvananda/netns v0.0.5 // indirect
|
||||||
|
golang.org/x/net v0.47.0 // indirect
|
||||||
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c // indirect
|
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c // indirect
|
||||||
golang.org/x/sys v0.38.0 // indirect
|
golang.org/x/sys v0.38.0 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
|
github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
|
||||||
github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
github.com/google/go-cmp v0.5.7 h1:81/ik6ipDQS2aGcBfIN5dHDB36BwrStyeAQquSYCV4o=
|
|
||||||
github.com/google/go-cmp v0.5.7/go.mod h1:n+brtR0CgQNWTVd5ZUFpTBC8YFBDLK/h/bpaJ8/DtOE=
|
github.com/google/go-cmp v0.5.7/go.mod h1:n+brtR0CgQNWTVd5ZUFpTBC8YFBDLK/h/bpaJ8/DtOE=
|
||||||
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
|
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
|
||||||
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
||||||
@@ -26,6 +25,7 @@ github.com/vishvananda/netns v0.0.5/go.mod h1:SpkAiCQRtJ6TvvxPnOSyH3BMl6unz3xZla
|
|||||||
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
|
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
golang.org/x/net v0.0.0-20190503192946-f4e77d36d62c/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
golang.org/x/net v0.0.0-20190503192946-f4e77d36d62c/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
|
golang.org/x/net v0.0.0-20190603091049-60506f45cf65 h1:+rhAzEzT3f4JtomfC371qB+0Ola2caSKcY69NUBZrRQ=
|
||||||
golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=
|
golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=
|
||||||
golang.org/x/net v0.47.0 h1:Mx+4dIFzqraBXUugkia1OOvlD6LemFo1ALMHjrXDOhY=
|
golang.org/x/net v0.47.0 h1:Mx+4dIFzqraBXUugkia1OOvlD6LemFo1ALMHjrXDOhY=
|
||||||
golang.org/x/net v0.47.0/go.mod h1:/jNxtkgq5yWUGYkaZGqo27cfGZ1c5Nen03aYrrKpVRU=
|
golang.org/x/net v0.47.0/go.mod h1:/jNxtkgq5yWUGYkaZGqo27cfGZ1c5Nen03aYrrKpVRU=
|
||||||
@@ -35,12 +35,12 @@ golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5h
|
|||||||
golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20210927094055-39ccf1dd6fa6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.0.0-20220209214540-3681064d5158/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.0.0-20220209214540-3681064d5158/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.2.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.2.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.10.0 h1:SqMFp9UcQJZa+pmYuAKjd9xq1f0j5rLcDIk0mj4qAsA=
|
||||||
golang.org/x/sys v0.10.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.10.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.38.0 h1:3yZWxaJjBmCWXqhN1qh02AkOnCQ1poK6oF+a7xWL6Gc=
|
golang.org/x/sys v0.38.0 h1:3yZWxaJjBmCWXqhN1qh02AkOnCQ1poK6oF+a7xWL6Gc=
|
||||||
golang.org/x/sys v0.38.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
golang.org/x/sys v0.38.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
|||||||
@@ -1,347 +0,0 @@
|
|||||||
package cmd
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"net"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/spf13/cobra"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
defaultBinaryPath = "/usr/local/bin/go-alived"
|
|
||||||
defaultConfigDir = "/etc/go-alived"
|
|
||||||
defaultConfigFile = "/etc/go-alived/config.yaml"
|
|
||||||
systemdServicePath = "/etc/systemd/system/go-alived.service"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
installMethod string
|
|
||||||
)
|
|
||||||
|
|
||||||
var installCmd = &cobra.Command{
|
|
||||||
Use: "install",
|
|
||||||
Aliases: []string{"i"},
|
|
||||||
Short: "Install go-alived as a system service",
|
|
||||||
Long: `Install go-alived binary and configuration files to system paths.
|
|
||||||
|
|
||||||
Supported installation methods:
|
|
||||||
- systemd: Install as a systemd service (default, recommended for modern Linux)
|
|
||||||
- service: Install binary and config only (manual startup)
|
|
||||||
|
|
||||||
Examples:
|
|
||||||
sudo go-alived install
|
|
||||||
sudo go-alived install --method systemd
|
|
||||||
sudo go-alived i -m service`,
|
|
||||||
Run: runInstall,
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
rootCmd.AddCommand(installCmd)
|
|
||||||
|
|
||||||
installCmd.Flags().StringVarP(&installMethod, "method", "m", "systemd",
|
|
||||||
"installation method: systemd, service")
|
|
||||||
}
|
|
||||||
|
|
||||||
func runInstall(cmd *cobra.Command, args []string) {
|
|
||||||
// Check root privileges
|
|
||||||
if os.Geteuid() != 0 {
|
|
||||||
fmt.Println("Error: This command requires root privileges")
|
|
||||||
fmt.Println("Please run with: sudo go-alived install")
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Validate method
|
|
||||||
method := strings.ToLower(installMethod)
|
|
||||||
if method != "systemd" && method != "service" {
|
|
||||||
fmt.Printf("Error: Invalid installation method '%s'\n", installMethod)
|
|
||||||
fmt.Println("Supported methods: systemd, service")
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Println("=== Go-Alived Installation ===")
|
|
||||||
fmt.Println()
|
|
||||||
|
|
||||||
totalSteps := 2
|
|
||||||
if method == "systemd" {
|
|
||||||
totalSteps = 3
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 1: Copy binary
|
|
||||||
if err := installBinary(1, totalSteps); err != nil {
|
|
||||||
fmt.Printf("Error installing binary: %v\n", err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 2: Create config directory and file
|
|
||||||
configCreated, err := installConfig(2, totalSteps)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Printf("Error installing config: %v\n", err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 3: Install systemd service if requested
|
|
||||||
if method == "systemd" {
|
|
||||||
if err := installSystemdService(3, totalSteps); err != nil {
|
|
||||||
fmt.Printf("Error installing systemd service: %v\n", err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Print completion message
|
|
||||||
printCompletionMessage(method, configCreated)
|
|
||||||
}
|
|
||||||
|
|
||||||
func installBinary(step, total int) error {
|
|
||||||
fmt.Printf("[%d/%d] Installing binary... ", step, total)
|
|
||||||
|
|
||||||
// Get current executable path
|
|
||||||
execPath, err := os.Executable()
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to get executable path: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Resolve symlinks
|
|
||||||
execPath, err = filepath.EvalSymlinks(execPath)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to resolve symlinks: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if already installed at target path
|
|
||||||
if execPath == defaultBinaryPath {
|
|
||||||
fmt.Println("already installed")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Open source file
|
|
||||||
src, err := os.Open(execPath)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to open source binary: %w", err)
|
|
||||||
}
|
|
||||||
defer src.Close()
|
|
||||||
|
|
||||||
// Create destination file
|
|
||||||
dst, err := os.OpenFile(defaultBinaryPath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0755)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to create destination binary: %w", err)
|
|
||||||
}
|
|
||||||
defer dst.Close()
|
|
||||||
|
|
||||||
// Copy binary
|
|
||||||
if _, err := io.Copy(dst, src); err != nil {
|
|
||||||
return fmt.Errorf("failed to copy binary: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Printf("done (%s)\n", defaultBinaryPath)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func installConfig(step, total int) (bool, error) {
|
|
||||||
fmt.Printf("[%d/%d] Setting up configuration... ", step, total)
|
|
||||||
|
|
||||||
// Create config directory
|
|
||||||
if err := os.MkdirAll(defaultConfigDir, 0755); err != nil {
|
|
||||||
return false, fmt.Errorf("failed to create config directory: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if config file already exists
|
|
||||||
if _, err := os.Stat(defaultConfigFile); err == nil {
|
|
||||||
fmt.Println("config already exists")
|
|
||||||
return false, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generate config content
|
|
||||||
configContent := generateDefaultConfig()
|
|
||||||
|
|
||||||
// Write config file
|
|
||||||
if err := os.WriteFile(defaultConfigFile, []byte(configContent), 0644); err != nil {
|
|
||||||
return false, fmt.Errorf("failed to write config file: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Printf("done (%s)\n", defaultConfigFile)
|
|
||||||
return true, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func installSystemdService(step, total int) error {
|
|
||||||
fmt.Printf("[%d/%d] Installing systemd service... ", step, total)
|
|
||||||
|
|
||||||
serviceContent := generateSystemdService()
|
|
||||||
|
|
||||||
if err := os.WriteFile(systemdServicePath, []byte(serviceContent), 0644); err != nil {
|
|
||||||
return fmt.Errorf("failed to write service file: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Printf("done (%s)\n", systemdServicePath)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func generateDefaultConfig() string {
|
|
||||||
// Auto-detect network interface
|
|
||||||
iface := detectNetworkInterface()
|
|
||||||
hostname, _ := os.Hostname()
|
|
||||||
if hostname == "" {
|
|
||||||
hostname = "node1"
|
|
||||||
}
|
|
||||||
|
|
||||||
return fmt.Sprintf(`# Go-Alived Configuration
|
|
||||||
# Generated by: go-alived install
|
|
||||||
# Documentation: https://github.com/loveuer/go-alived
|
|
||||||
|
|
||||||
global:
|
|
||||||
router_id: "%s"
|
|
||||||
|
|
||||||
vrrp_instances:
|
|
||||||
- name: "VI_1"
|
|
||||||
interface: "%s"
|
|
||||||
state: "BACKUP"
|
|
||||||
virtual_router_id: 51
|
|
||||||
priority: 100
|
|
||||||
advert_interval: 1
|
|
||||||
auth_type: "PASS"
|
|
||||||
auth_pass: "changeme" # TODO: Change this password
|
|
||||||
virtual_ips:
|
|
||||||
- "192.168.1.100/24" # TODO: Change to your VIP
|
|
||||||
|
|
||||||
# Optional: Health checkers
|
|
||||||
# health_checkers:
|
|
||||||
# - name: "check_nginx"
|
|
||||||
# type: "tcp"
|
|
||||||
# interval: 3s
|
|
||||||
# timeout: 2s
|
|
||||||
# rise: 3
|
|
||||||
# fall: 2
|
|
||||||
# config:
|
|
||||||
# host: "127.0.0.1"
|
|
||||||
# port: 80
|
|
||||||
`, hostname, iface)
|
|
||||||
}
|
|
||||||
|
|
||||||
func generateSystemdService() string {
|
|
||||||
return `[Unit]
|
|
||||||
Description=Go-Alived - VRRP High Availability Service
|
|
||||||
Documentation=https://github.com/loveuer/go-alived
|
|
||||||
After=network-online.target
|
|
||||||
Wants=network-online.target
|
|
||||||
|
|
||||||
[Service]
|
|
||||||
Type=simple
|
|
||||||
User=root
|
|
||||||
Group=root
|
|
||||||
|
|
||||||
ExecStart=/usr/local/bin/go-alived run --config /etc/go-alived/config.yaml
|
|
||||||
ExecReload=/bin/kill -HUP $MAINPID
|
|
||||||
|
|
||||||
Restart=on-failure
|
|
||||||
RestartSec=5s
|
|
||||||
|
|
||||||
StandardOutput=journal
|
|
||||||
StandardError=journal
|
|
||||||
SyslogIdentifier=go-alived
|
|
||||||
|
|
||||||
# Security settings
|
|
||||||
NoNewPrivileges=false
|
|
||||||
PrivateTmp=true
|
|
||||||
ProtectSystem=strict
|
|
||||||
ProtectHome=true
|
|
||||||
ReadWritePaths=/etc/go-alived
|
|
||||||
|
|
||||||
# Resource limits
|
|
||||||
LimitNOFILE=65535
|
|
||||||
LimitNPROC=512
|
|
||||||
|
|
||||||
# Capabilities required for VRRP operations
|
|
||||||
AmbientCapabilities=CAP_NET_ADMIN CAP_NET_RAW CAP_NET_BIND_SERVICE
|
|
||||||
CapabilityBoundingSet=CAP_NET_ADMIN CAP_NET_RAW CAP_NET_BIND_SERVICE
|
|
||||||
|
|
||||||
[Install]
|
|
||||||
WantedBy=multi-user.target
|
|
||||||
`
|
|
||||||
}
|
|
||||||
|
|
||||||
func detectNetworkInterface() string {
|
|
||||||
interfaces, err := net.Interfaces()
|
|
||||||
if err != nil {
|
|
||||||
return "eth0"
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, iface := range interfaces {
|
|
||||||
// Skip loopback and down interfaces
|
|
||||||
if iface.Flags&net.FlagLoopback != 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if iface.Flags&net.FlagUp == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check if interface has IPv4 address
|
|
||||||
addrs, err := iface.Addrs()
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, addr := range addrs {
|
|
||||||
if ipNet, ok := addr.(*net.IPNet); ok {
|
|
||||||
if ipv4 := ipNet.IP.To4(); ipv4 != nil && !ipv4.IsLoopback() {
|
|
||||||
return iface.Name
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return "eth0"
|
|
||||||
}
|
|
||||||
|
|
||||||
func printCompletionMessage(method string, configCreated bool) {
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println("=== Installation Complete ===")
|
|
||||||
fmt.Println()
|
|
||||||
|
|
||||||
// What needs to be modified
|
|
||||||
fmt.Println(">>> Configuration Required:")
|
|
||||||
fmt.Printf(" Edit: %s\n", defaultConfigFile)
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println(" Modify the following settings:")
|
|
||||||
if configCreated {
|
|
||||||
fmt.Println(" - auth_pass: Change 'changeme' to a secure password")
|
|
||||||
fmt.Println(" - virtual_ips: Set your Virtual IP address(es)")
|
|
||||||
fmt.Println(" - interface: Verify the network interface is correct")
|
|
||||||
fmt.Println(" - priority: Adjust based on node role (higher = more likely to be master)")
|
|
||||||
} else {
|
|
||||||
fmt.Println(" - Review your existing configuration")
|
|
||||||
}
|
|
||||||
fmt.Println()
|
|
||||||
|
|
||||||
// How to start
|
|
||||||
fmt.Println(">>> Next Steps:")
|
|
||||||
if method == "systemd" {
|
|
||||||
fmt.Println(" 1. Edit configuration:")
|
|
||||||
fmt.Printf(" sudo vim %s\n", defaultConfigFile)
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println(" 2. Reload systemd and start service:")
|
|
||||||
fmt.Println(" sudo systemctl daemon-reload")
|
|
||||||
fmt.Println(" sudo systemctl enable go-alived")
|
|
||||||
fmt.Println(" sudo systemctl start go-alived")
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println(" 3. Check service status:")
|
|
||||||
fmt.Println(" sudo systemctl status go-alived")
|
|
||||||
fmt.Println(" sudo journalctl -u go-alived -f")
|
|
||||||
} else {
|
|
||||||
fmt.Println(" 1. Edit configuration:")
|
|
||||||
fmt.Printf(" sudo vim %s\n", defaultConfigFile)
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println(" 2. Run manually:")
|
|
||||||
fmt.Printf(" sudo %s run -c %s\n", defaultBinaryPath, defaultConfigFile)
|
|
||||||
fmt.Println()
|
|
||||||
fmt.Println(" 3. Or run in debug mode:")
|
|
||||||
fmt.Printf(" sudo %s run -c %s -d\n", defaultBinaryPath, defaultConfigFile)
|
|
||||||
}
|
|
||||||
fmt.Println()
|
|
||||||
|
|
||||||
// Test environment
|
|
||||||
fmt.Println(">>> Test Environment (Optional):")
|
|
||||||
fmt.Printf(" sudo %s test\n", defaultBinaryPath)
|
|
||||||
fmt.Println()
|
|
||||||
}
|
|
||||||
@@ -11,7 +11,7 @@ var rootCmd = &cobra.Command{
|
|||||||
Short: "Go-Alived - VRRP High Availability Service",
|
Short: "Go-Alived - VRRP High Availability Service",
|
||||||
Long: `go-alived is a lightweight, dependency-free VRRP implementation in Go.
|
Long: `go-alived is a lightweight, dependency-free VRRP implementation in Go.
|
||||||
It provides high availability for IP addresses with health checking support.`,
|
It provides high availability for IP addresses with health checking support.`,
|
||||||
Version: "1.2.0",
|
Version: "1.0.0",
|
||||||
}
|
}
|
||||||
|
|
||||||
func Execute() {
|
func Execute() {
|
||||||
|
|||||||
@@ -367,7 +367,7 @@ func (t *EnvironmentTest) TestCloudEnvironment() {
|
|||||||
if err == nil {
|
if err == nil {
|
||||||
cloudDetected = true
|
cloudDetected = true
|
||||||
t.AddResult("云环境", !test.isFatal, fmt.Sprintf("检测到%s环境", test.name), test.isFatal)
|
t.AddResult("云环境", !test.isFatal, fmt.Sprintf("检测到%s环境", test.name), test.isFatal)
|
||||||
t.log.Warn("%s", test.solution)
|
t.log.Warn(test.solution)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -8,13 +8,9 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func CreateChecker(cfg *config.HealthChecker) (Checker, error) {
|
func CreateChecker(cfg *config.HealthChecker) (Checker, error) {
|
||||||
if cfg.Config == nil {
|
|
||||||
return nil, fmt.Errorf("missing config for checker %s", cfg.Name)
|
|
||||||
}
|
|
||||||
|
|
||||||
configMap, ok := cfg.Config.(map[string]interface{})
|
configMap, ok := cfg.Config.(map[string]interface{})
|
||||||
if !ok {
|
if !ok {
|
||||||
return nil, fmt.Errorf("invalid config type for checker %s: expected map[string]interface{}", cfg.Name)
|
return nil, fmt.Errorf("invalid config for checker %s", cfg.Name)
|
||||||
}
|
}
|
||||||
|
|
||||||
switch cfg.Type {
|
switch cfg.Type {
|
||||||
@@ -40,7 +36,6 @@ func LoadFromConfig(cfg *config.Config, log *logger.Logger) (*Manager, error) {
|
|||||||
return nil, fmt.Errorf("failed to create checker %s: %w", healthCfg.Name, err)
|
return nil, fmt.Errorf("failed to create checker %s: %w", healthCfg.Name, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
configMap, _ := healthCfg.Config.(map[string]interface{})
|
|
||||||
monitorCfg := &CheckerConfig{
|
monitorCfg := &CheckerConfig{
|
||||||
Name: healthCfg.Name,
|
Name: healthCfg.Name,
|
||||||
Type: healthCfg.Type,
|
Type: healthCfg.Type,
|
||||||
@@ -48,7 +43,7 @@ func LoadFromConfig(cfg *config.Config, log *logger.Logger) (*Manager, error) {
|
|||||||
Timeout: healthCfg.Timeout,
|
Timeout: healthCfg.Timeout,
|
||||||
Rise: healthCfg.Rise,
|
Rise: healthCfg.Rise,
|
||||||
Fall: healthCfg.Fall,
|
Fall: healthCfg.Fall,
|
||||||
Config: configMap,
|
Config: healthCfg.Config.(map[string]interface{}),
|
||||||
}
|
}
|
||||||
|
|
||||||
monitor := NewMonitor(checker, monitorCfg, log)
|
monitor := NewMonitor(checker, monitorCfg, log)
|
||||||
|
|||||||
@@ -1,74 +0,0 @@
|
|||||||
package health
|
|
||||||
|
|
||||||
import (
|
|
||||||
"sync"
|
|
||||||
|
|
||||||
"github.com/loveuer/go-alived/pkg/logger"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Manager manages multiple health check monitors.
|
|
||||||
type Manager struct {
|
|
||||||
monitors map[string]*Monitor
|
|
||||||
mu sync.RWMutex
|
|
||||||
log *logger.Logger
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewManager creates a new health check Manager.
|
|
||||||
func NewManager(log *logger.Logger) *Manager {
|
|
||||||
return &Manager{
|
|
||||||
monitors: make(map[string]*Monitor),
|
|
||||||
log: log,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// AddMonitor adds a monitor to the manager.
|
|
||||||
func (m *Manager) AddMonitor(monitor *Monitor) {
|
|
||||||
m.mu.Lock()
|
|
||||||
defer m.mu.Unlock()
|
|
||||||
m.monitors[monitor.config.Name] = monitor
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetMonitor retrieves a monitor by name.
|
|
||||||
func (m *Manager) GetMonitor(name string) (*Monitor, bool) {
|
|
||||||
m.mu.RLock()
|
|
||||||
defer m.mu.RUnlock()
|
|
||||||
monitor, ok := m.monitors[name]
|
|
||||||
return monitor, ok
|
|
||||||
}
|
|
||||||
|
|
||||||
// StartAll starts all registered monitors.
|
|
||||||
func (m *Manager) StartAll() {
|
|
||||||
m.mu.RLock()
|
|
||||||
defer m.mu.RUnlock()
|
|
||||||
|
|
||||||
for _, monitor := range m.monitors {
|
|
||||||
monitor.Start()
|
|
||||||
}
|
|
||||||
|
|
||||||
m.log.Info("started %d health check monitor(s)", len(m.monitors))
|
|
||||||
}
|
|
||||||
|
|
||||||
// StopAll stops all registered monitors.
|
|
||||||
func (m *Manager) StopAll() {
|
|
||||||
m.mu.RLock()
|
|
||||||
defer m.mu.RUnlock()
|
|
||||||
|
|
||||||
for _, monitor := range m.monitors {
|
|
||||||
monitor.Stop()
|
|
||||||
}
|
|
||||||
|
|
||||||
m.log.Info("stopped all health check monitors")
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetAllStates returns the current state of all monitors.
|
|
||||||
func (m *Manager) GetAllStates() map[string]*CheckerState {
|
|
||||||
m.mu.RLock()
|
|
||||||
defer m.mu.RUnlock()
|
|
||||||
|
|
||||||
states := make(map[string]*CheckerState)
|
|
||||||
for name, monitor := range m.monitors {
|
|
||||||
states[name] = monitor.GetState()
|
|
||||||
}
|
|
||||||
|
|
||||||
return states
|
|
||||||
}
|
|
||||||
+61
-10
@@ -8,7 +8,6 @@ import (
|
|||||||
"github.com/loveuer/go-alived/pkg/logger"
|
"github.com/loveuer/go-alived/pkg/logger"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Monitor runs periodic health checks and tracks state.
|
|
||||||
type Monitor struct {
|
type Monitor struct {
|
||||||
checker Checker
|
checker Checker
|
||||||
config *CheckerConfig
|
config *CheckerConfig
|
||||||
@@ -22,7 +21,6 @@ type Monitor struct {
|
|||||||
mu sync.RWMutex
|
mu sync.RWMutex
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewMonitor creates a new Monitor for the given checker.
|
|
||||||
func NewMonitor(checker Checker, config *CheckerConfig, log *logger.Logger) *Monitor {
|
func NewMonitor(checker Checker, config *CheckerConfig, log *logger.Logger) *Monitor {
|
||||||
return &Monitor{
|
return &Monitor{
|
||||||
checker: checker,
|
checker: checker,
|
||||||
@@ -37,7 +35,6 @@ func NewMonitor(checker Checker, config *CheckerConfig, log *logger.Logger) *Mon
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Start begins the health check loop.
|
|
||||||
func (m *Monitor) Start() {
|
func (m *Monitor) Start() {
|
||||||
m.mu.Lock()
|
m.mu.Lock()
|
||||||
if m.running {
|
if m.running {
|
||||||
@@ -54,7 +51,6 @@ func (m *Monitor) Start() {
|
|||||||
go m.checkLoop()
|
go m.checkLoop()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Stop stops the health check loop.
|
|
||||||
func (m *Monitor) Stop() {
|
func (m *Monitor) Stop() {
|
||||||
m.mu.Lock()
|
m.mu.Lock()
|
||||||
if !m.running {
|
if !m.running {
|
||||||
@@ -75,7 +71,6 @@ func (m *Monitor) checkLoop() {
|
|||||||
ticker := time.NewTicker(m.config.Interval)
|
ticker := time.NewTicker(m.config.Interval)
|
||||||
defer ticker.Stop()
|
defer ticker.Stop()
|
||||||
|
|
||||||
// Perform initial check immediately
|
|
||||||
m.performCheck()
|
m.performCheck()
|
||||||
|
|
||||||
for {
|
for {
|
||||||
@@ -100,8 +95,7 @@ func (m *Monitor) performCheck() {
|
|||||||
oldHealthy := m.state.Healthy
|
oldHealthy := m.state.Healthy
|
||||||
stateChanged := m.state.Update(result, m.config.Rise, m.config.Fall)
|
stateChanged := m.state.Update(result, m.config.Rise, m.config.Fall)
|
||||||
newHealthy := m.state.Healthy
|
newHealthy := m.state.Healthy
|
||||||
callbacks := make([]StateChangeCallback, len(m.callbacks))
|
callbacks := m.callbacks
|
||||||
copy(callbacks, m.callbacks)
|
|
||||||
m.mu.Unlock()
|
m.mu.Unlock()
|
||||||
|
|
||||||
m.log.Debug("[HealthCheck:%s] check completed: result=%s, duration=%s, healthy=%v",
|
m.log.Debug("[HealthCheck:%s] check completed: result=%s, duration=%s, healthy=%v",
|
||||||
@@ -117,14 +111,12 @@ func (m *Monitor) performCheck() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// OnStateChange registers a callback for health state changes.
|
|
||||||
func (m *Monitor) OnStateChange(callback StateChangeCallback) {
|
func (m *Monitor) OnStateChange(callback StateChangeCallback) {
|
||||||
m.mu.Lock()
|
m.mu.Lock()
|
||||||
defer m.mu.Unlock()
|
defer m.mu.Unlock()
|
||||||
m.callbacks = append(m.callbacks, callback)
|
m.callbacks = append(m.callbacks, callback)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetState returns a copy of the current checker state.
|
|
||||||
func (m *Monitor) GetState() *CheckerState {
|
func (m *Monitor) GetState() *CheckerState {
|
||||||
m.mu.RLock()
|
m.mu.RLock()
|
||||||
defer m.mu.RUnlock()
|
defer m.mu.RUnlock()
|
||||||
@@ -133,9 +125,68 @@ func (m *Monitor) GetState() *CheckerState {
|
|||||||
return &stateCopy
|
return &stateCopy
|
||||||
}
|
}
|
||||||
|
|
||||||
// IsHealthy returns whether the checker is currently healthy.
|
|
||||||
func (m *Monitor) IsHealthy() bool {
|
func (m *Monitor) IsHealthy() bool {
|
||||||
m.mu.RLock()
|
m.mu.RLock()
|
||||||
defer m.mu.RUnlock()
|
defer m.mu.RUnlock()
|
||||||
return m.state.Healthy
|
return m.state.Healthy
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Manager struct {
|
||||||
|
monitors map[string]*Monitor
|
||||||
|
mu sync.RWMutex
|
||||||
|
log *logger.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewManager(log *logger.Logger) *Manager {
|
||||||
|
return &Manager{
|
||||||
|
monitors: make(map[string]*Monitor),
|
||||||
|
log: log,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Manager) AddMonitor(monitor *Monitor) {
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
m.monitors[monitor.config.Name] = monitor
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Manager) GetMonitor(name string) (*Monitor, bool) {
|
||||||
|
m.mu.RLock()
|
||||||
|
defer m.mu.RUnlock()
|
||||||
|
monitor, ok := m.monitors[name]
|
||||||
|
return monitor, ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Manager) StartAll() {
|
||||||
|
m.mu.RLock()
|
||||||
|
defer m.mu.RUnlock()
|
||||||
|
|
||||||
|
for _, monitor := range m.monitors {
|
||||||
|
monitor.Start()
|
||||||
|
}
|
||||||
|
|
||||||
|
m.log.Info("started %d health check monitor(s)", len(m.monitors))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Manager) StopAll() {
|
||||||
|
m.mu.RLock()
|
||||||
|
defer m.mu.RUnlock()
|
||||||
|
|
||||||
|
for _, monitor := range m.monitors {
|
||||||
|
monitor.Stop()
|
||||||
|
}
|
||||||
|
|
||||||
|
m.log.Info("stopped all health check monitors")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Manager) GetAllStates() map[string]*CheckerState {
|
||||||
|
m.mu.RLock()
|
||||||
|
defer m.mu.RUnlock()
|
||||||
|
|
||||||
|
states := make(map[string]*CheckerState)
|
||||||
|
for name, monitor := range m.monitors {
|
||||||
|
states[name] = monitor.GetState()
|
||||||
|
}
|
||||||
|
|
||||||
|
return states
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,95 +0,0 @@
|
|||||||
package vrrp
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// StateTransition represents a single state transition event.
|
|
||||||
type StateTransition struct {
|
|
||||||
From State
|
|
||||||
To State
|
|
||||||
Timestamp time.Time
|
|
||||||
Reason string
|
|
||||||
}
|
|
||||||
|
|
||||||
// StateHistory maintains a bounded history of state transitions.
|
|
||||||
type StateHistory struct {
|
|
||||||
transitions []StateTransition
|
|
||||||
maxSize int
|
|
||||||
mu sync.RWMutex
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewStateHistory creates a new StateHistory with the specified maximum size.
|
|
||||||
func NewStateHistory(maxSize int) *StateHistory {
|
|
||||||
return &StateHistory{
|
|
||||||
transitions: make([]StateTransition, 0, maxSize),
|
|
||||||
maxSize: maxSize,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add records a new state transition.
|
|
||||||
func (sh *StateHistory) Add(from, to State, reason string) {
|
|
||||||
sh.mu.Lock()
|
|
||||||
defer sh.mu.Unlock()
|
|
||||||
|
|
||||||
transition := StateTransition{
|
|
||||||
From: from,
|
|
||||||
To: to,
|
|
||||||
Timestamp: time.Now(),
|
|
||||||
Reason: reason,
|
|
||||||
}
|
|
||||||
|
|
||||||
sh.transitions = append(sh.transitions, transition)
|
|
||||||
|
|
||||||
// Maintain bounded size using ring buffer style
|
|
||||||
if len(sh.transitions) > sh.maxSize {
|
|
||||||
// Copy to new slice to allow garbage collection of old backing array
|
|
||||||
newTransitions := make([]StateTransition, len(sh.transitions)-1, sh.maxSize)
|
|
||||||
copy(newTransitions, sh.transitions[1:])
|
|
||||||
sh.transitions = newTransitions
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetRecent returns the most recent n transitions.
|
|
||||||
func (sh *StateHistory) GetRecent(n int) []StateTransition {
|
|
||||||
sh.mu.RLock()
|
|
||||||
defer sh.mu.RUnlock()
|
|
||||||
|
|
||||||
if n > len(sh.transitions) {
|
|
||||||
n = len(sh.transitions)
|
|
||||||
}
|
|
||||||
|
|
||||||
start := len(sh.transitions) - n
|
|
||||||
result := make([]StateTransition, n)
|
|
||||||
copy(result, sh.transitions[start:])
|
|
||||||
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
|
|
||||||
// Len returns the number of recorded transitions.
|
|
||||||
func (sh *StateHistory) Len() int {
|
|
||||||
sh.mu.RLock()
|
|
||||||
defer sh.mu.RUnlock()
|
|
||||||
return len(sh.transitions)
|
|
||||||
}
|
|
||||||
|
|
||||||
// String returns a formatted string representation of the history.
|
|
||||||
func (sh *StateHistory) String() string {
|
|
||||||
sh.mu.RLock()
|
|
||||||
defer sh.mu.RUnlock()
|
|
||||||
|
|
||||||
if len(sh.transitions) == 0 {
|
|
||||||
return "No state transitions"
|
|
||||||
}
|
|
||||||
|
|
||||||
result := "State transition history:\n"
|
|
||||||
for _, t := range sh.transitions {
|
|
||||||
result += fmt.Sprintf(" %s: %s -> %s (%s)\n",
|
|
||||||
t.Timestamp.Format("2006-01-02 15:04:05"),
|
|
||||||
t.From, t.To, t.Reason)
|
|
||||||
}
|
|
||||||
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
@@ -17,7 +17,6 @@ type Instance struct {
|
|||||||
AdvertInterval uint8
|
AdvertInterval uint8
|
||||||
Interface string
|
Interface string
|
||||||
VirtualIPs []net.IP
|
VirtualIPs []net.IP
|
||||||
VirtualIPCIDRs []string // preserve original CIDR notation
|
|
||||||
AuthType uint8
|
AuthType uint8
|
||||||
AuthPass string
|
AuthPass string
|
||||||
TrackScripts []string
|
TrackScripts []string
|
||||||
@@ -65,14 +64,12 @@ func NewInstance(
|
|||||||
}
|
}
|
||||||
|
|
||||||
virtualIPs := make([]net.IP, 0, len(vips))
|
virtualIPs := make([]net.IP, 0, len(vips))
|
||||||
virtualIPCIDRs := make([]string, 0, len(vips))
|
|
||||||
for _, vip := range vips {
|
for _, vip := range vips {
|
||||||
ip, _, err := net.ParseCIDR(vip)
|
ip, _, err := net.ParseCIDR(vip)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("invalid VIP %s: %w", vip, err)
|
return nil, fmt.Errorf("invalid VIP %s: %w", vip, err)
|
||||||
}
|
}
|
||||||
virtualIPs = append(virtualIPs, ip)
|
virtualIPs = append(virtualIPs, ip)
|
||||||
virtualIPCIDRs = append(virtualIPCIDRs, vip)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var authTypeNum uint8
|
var authTypeNum uint8
|
||||||
@@ -97,7 +94,6 @@ func NewInstance(
|
|||||||
AdvertInterval: advertInt,
|
AdvertInterval: advertInt,
|
||||||
Interface: iface,
|
Interface: iface,
|
||||||
VirtualIPs: virtualIPs,
|
VirtualIPs: virtualIPs,
|
||||||
VirtualIPCIDRs: virtualIPCIDRs,
|
|
||||||
AuthType: authTypeNum,
|
AuthType: authTypeNum,
|
||||||
AuthPass: authPass,
|
AuthPass: authPass,
|
||||||
TrackScripts: trackScripts,
|
TrackScripts: trackScripts,
|
||||||
@@ -196,15 +192,8 @@ func (inst *Instance) receiveLoop() {
|
|||||||
default:
|
default:
|
||||||
}
|
}
|
||||||
|
|
||||||
// Set read deadline to allow periodic check of stop channel
|
|
||||||
inst.socket.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
|
|
||||||
|
|
||||||
pkt, srcIP, err := inst.socket.Receive()
|
pkt, srcIP, err := inst.socket.Receive()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// Check if it's a timeout error, which is expected
|
|
||||||
if netErr, ok := err.(interface{ Timeout() bool }); ok && netErr.Timeout() {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
inst.log.Debug("[%s] failed to receive packet: %v", inst.Name, err)
|
inst.log.Debug("[%s] failed to receive packet: %v", inst.Name, err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@@ -382,7 +371,11 @@ func (inst *Instance) removeVIPs() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (inst *Instance) getVIPsWithCIDR() []string {
|
func (inst *Instance) getVIPsWithCIDR() []string {
|
||||||
return inst.VirtualIPCIDRs
|
result := make([]string, len(inst.VirtualIPs))
|
||||||
|
for i, ip := range inst.VirtualIPs {
|
||||||
|
result[i] = ip.String() + "/32"
|
||||||
|
}
|
||||||
|
return result
|
||||||
}
|
}
|
||||||
|
|
||||||
func (inst *Instance) GetState() State {
|
func (inst *Instance) GetState() State {
|
||||||
@@ -409,8 +402,6 @@ func (inst *Instance) AdjustPriority(delta int) {
|
|||||||
|
|
||||||
if delta < 0 {
|
if delta < 0 {
|
||||||
inst.priorityCalc.DecreasePriority(uint8(-delta))
|
inst.priorityCalc.DecreasePriority(uint8(-delta))
|
||||||
} else if delta > 0 {
|
|
||||||
inst.priorityCalc.IncreasePriority(uint8(delta))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
newPriority := inst.priorityCalc.GetPriority()
|
newPriority := inst.priorityCalc.GetPriority()
|
||||||
|
|||||||
@@ -1,99 +0,0 @@
|
|||||||
package vrrp
|
|
||||||
|
|
||||||
import (
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// PriorityCalculator manages VRRP priority with support for dynamic adjustment.
|
|
||||||
type PriorityCalculator struct {
|
|
||||||
basePriority uint8
|
|
||||||
currentPriority uint8
|
|
||||||
mu sync.RWMutex
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewPriorityCalculator creates a new PriorityCalculator with the specified base priority.
|
|
||||||
func NewPriorityCalculator(basePriority uint8) *PriorityCalculator {
|
|
||||||
return &PriorityCalculator{
|
|
||||||
basePriority: basePriority,
|
|
||||||
currentPriority: basePriority,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetPriority returns the current priority.
|
|
||||||
func (pc *PriorityCalculator) GetPriority() uint8 {
|
|
||||||
pc.mu.RLock()
|
|
||||||
defer pc.mu.RUnlock()
|
|
||||||
return pc.currentPriority
|
|
||||||
}
|
|
||||||
|
|
||||||
// DecreasePriority decreases the current priority by the specified amount.
|
|
||||||
// The priority will not go below 0.
|
|
||||||
func (pc *PriorityCalculator) DecreasePriority(amount uint8) {
|
|
||||||
pc.mu.Lock()
|
|
||||||
defer pc.mu.Unlock()
|
|
||||||
|
|
||||||
if pc.currentPriority > amount {
|
|
||||||
pc.currentPriority -= amount
|
|
||||||
} else {
|
|
||||||
pc.currentPriority = 0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// IncreasePriority increases the current priority by the specified amount.
|
|
||||||
// The priority will not exceed 255 or the base priority.
|
|
||||||
func (pc *PriorityCalculator) IncreasePriority(amount uint8) {
|
|
||||||
pc.mu.Lock()
|
|
||||||
defer pc.mu.Unlock()
|
|
||||||
|
|
||||||
newPriority := pc.currentPriority + amount
|
|
||||||
if newPriority > pc.basePriority {
|
|
||||||
newPriority = pc.basePriority
|
|
||||||
}
|
|
||||||
if newPriority < pc.currentPriority { // overflow check
|
|
||||||
newPriority = pc.basePriority
|
|
||||||
}
|
|
||||||
pc.currentPriority = newPriority
|
|
||||||
}
|
|
||||||
|
|
||||||
// ResetPriority resets the priority to the base value.
|
|
||||||
func (pc *PriorityCalculator) ResetPriority() {
|
|
||||||
pc.mu.Lock()
|
|
||||||
defer pc.mu.Unlock()
|
|
||||||
pc.currentPriority = pc.basePriority
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetBasePriority sets a new base priority and resets current priority to match.
|
|
||||||
func (pc *PriorityCalculator) SetBasePriority(priority uint8) {
|
|
||||||
pc.mu.Lock()
|
|
||||||
defer pc.mu.Unlock()
|
|
||||||
pc.basePriority = priority
|
|
||||||
pc.currentPriority = priority
|
|
||||||
}
|
|
||||||
|
|
||||||
// ShouldBecomeMaster determines if the local node should become master
|
|
||||||
// based on priority comparison and IP address tie-breaking.
|
|
||||||
func ShouldBecomeMaster(localPriority, remotePriority uint8, localIP, remoteIP string) bool {
|
|
||||||
if localPriority > remotePriority {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
if localPriority == remotePriority {
|
|
||||||
return localIP > remoteIP
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// CalculateMasterDownInterval calculates the master down interval
|
|
||||||
// according to VRRP specification: (3 * Advertisement_Interval).
|
|
||||||
func CalculateMasterDownInterval(advertInt uint8) time.Duration {
|
|
||||||
return time.Duration(3*int(advertInt)) * time.Second
|
|
||||||
}
|
|
||||||
|
|
||||||
// CalculateSkewTime calculates the skew time for master down timer
|
|
||||||
// according to VRRP specification: ((256 - Priority) / 256).
|
|
||||||
func CalculateSkewTime(priority uint8) time.Duration {
|
|
||||||
skew := float64(256-int(priority)) / 256.0
|
|
||||||
return time.Duration(skew * float64(time.Second))
|
|
||||||
}
|
|
||||||
@@ -5,7 +5,6 @@ import (
|
|||||||
"net"
|
"net"
|
||||||
"os"
|
"os"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
|
||||||
|
|
||||||
"golang.org/x/net/ipv4"
|
"golang.org/x/net/ipv4"
|
||||||
)
|
)
|
||||||
@@ -16,7 +15,6 @@ const (
|
|||||||
|
|
||||||
type Socket struct {
|
type Socket struct {
|
||||||
conn *ipv4.RawConn
|
conn *ipv4.RawConn
|
||||||
packetConn net.PacketConn
|
|
||||||
iface *net.Interface
|
iface *net.Interface
|
||||||
localIP net.IP
|
localIP net.IP
|
||||||
groupIP net.IP
|
groupIP net.IP
|
||||||
@@ -58,8 +56,9 @@ func NewSocket(ifaceName string) (*Socket, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
file := os.NewFile(uintptr(fd), "vrrp-socket")
|
file := os.NewFile(uintptr(fd), "vrrp-socket")
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
packetConn, err := net.FilePacketConn(file)
|
packetConn, err := net.FilePacketConn(file)
|
||||||
file.Close()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to create packet connection: %w", err)
|
return nil, fmt.Errorf("failed to create packet connection: %w", err)
|
||||||
}
|
}
|
||||||
@@ -88,7 +87,6 @@ func NewSocket(ifaceName string) (*Socket, error) {
|
|||||||
|
|
||||||
return &Socket{
|
return &Socket{
|
||||||
conn: rawConn,
|
conn: rawConn,
|
||||||
packetConn: packetConn,
|
|
||||||
iface: iface,
|
iface: iface,
|
||||||
localIP: localIP,
|
localIP: localIP,
|
||||||
groupIP: groupIP,
|
groupIP: groupIP,
|
||||||
@@ -135,10 +133,6 @@ func (s *Socket) Receive() (*VRRPPacket, net.IP, error) {
|
|||||||
return pkt, header.Src, nil
|
return pkt, header.Src, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Socket) SetReadDeadline(t time.Time) error {
|
|
||||||
return s.packetConn.SetReadDeadline(t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Socket) Close() error {
|
func (s *Socket) Close() error {
|
||||||
if err := s.conn.LeaveGroup(s.iface, &net.IPAddr{IP: s.groupIP}); err != nil {
|
if err := s.conn.LeaveGroup(s.iface, &net.IPAddr{IP: s.groupIP}); err != nil {
|
||||||
return err
|
return err
|
||||||
|
|||||||
+196
-14
@@ -1,8 +1,11 @@
|
|||||||
package vrrp
|
package vrrp
|
||||||
|
|
||||||
import "sync"
|
import (
|
||||||
|
"fmt"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
// State represents the VRRP instance state.
|
|
||||||
type State int
|
type State int
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -12,7 +15,6 @@ const (
|
|||||||
StateFault
|
StateFault
|
||||||
)
|
)
|
||||||
|
|
||||||
// String returns the string representation of the state.
|
|
||||||
func (s State) String() string {
|
func (s State) String() string {
|
||||||
switch s {
|
switch s {
|
||||||
case StateInit:
|
case StateInit:
|
||||||
@@ -28,39 +30,33 @@ func (s State) String() string {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// StateMachine manages VRRP state transitions with thread-safe callbacks.
|
|
||||||
type StateMachine struct {
|
type StateMachine struct {
|
||||||
currentState State
|
currentState State
|
||||||
|
previousState State
|
||||||
mu sync.RWMutex
|
mu sync.RWMutex
|
||||||
stateChangeCallbacks []func(old, new State)
|
stateChangeCallbacks []func(old, new State)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewStateMachine creates a new StateMachine with the specified initial state.
|
|
||||||
func NewStateMachine(initialState State) *StateMachine {
|
func NewStateMachine(initialState State) *StateMachine {
|
||||||
return &StateMachine{
|
return &StateMachine{
|
||||||
currentState: initialState,
|
currentState: initialState,
|
||||||
|
previousState: StateInit,
|
||||||
stateChangeCallbacks: make([]func(old, new State), 0),
|
stateChangeCallbacks: make([]func(old, new State), 0),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetState returns the current state.
|
|
||||||
func (sm *StateMachine) GetState() State {
|
func (sm *StateMachine) GetState() State {
|
||||||
sm.mu.RLock()
|
sm.mu.RLock()
|
||||||
defer sm.mu.RUnlock()
|
defer sm.mu.RUnlock()
|
||||||
return sm.currentState
|
return sm.currentState
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetState transitions to a new state and triggers registered callbacks.
|
|
||||||
func (sm *StateMachine) SetState(newState State) {
|
func (sm *StateMachine) SetState(newState State) {
|
||||||
sm.mu.Lock()
|
sm.mu.Lock()
|
||||||
oldState := sm.currentState
|
oldState := sm.currentState
|
||||||
if oldState == newState {
|
sm.previousState = oldState
|
||||||
sm.mu.Unlock()
|
|
||||||
return
|
|
||||||
}
|
|
||||||
sm.currentState = newState
|
sm.currentState = newState
|
||||||
callbacks := make([]func(old, new State), len(sm.stateChangeCallbacks))
|
callbacks := sm.stateChangeCallbacks
|
||||||
copy(callbacks, sm.stateChangeCallbacks)
|
|
||||||
sm.mu.Unlock()
|
sm.mu.Unlock()
|
||||||
|
|
||||||
for _, callback := range callbacks {
|
for _, callback := range callbacks {
|
||||||
@@ -68,9 +64,195 @@ func (sm *StateMachine) SetState(newState State) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// OnStateChange registers a callback to be invoked on state changes.
|
|
||||||
func (sm *StateMachine) OnStateChange(callback func(old, new State)) {
|
func (sm *StateMachine) OnStateChange(callback func(old, new State)) {
|
||||||
sm.mu.Lock()
|
sm.mu.Lock()
|
||||||
defer sm.mu.Unlock()
|
defer sm.mu.Unlock()
|
||||||
sm.stateChangeCallbacks = append(sm.stateChangeCallbacks, callback)
|
sm.stateChangeCallbacks = append(sm.stateChangeCallbacks, callback)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Timer struct {
|
||||||
|
duration time.Duration
|
||||||
|
timer *time.Timer
|
||||||
|
callback func()
|
||||||
|
mu sync.Mutex
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewTimer(duration time.Duration, callback func()) *Timer {
|
||||||
|
return &Timer{
|
||||||
|
duration: duration,
|
||||||
|
callback: callback,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *Timer) Start() {
|
||||||
|
t.mu.Lock()
|
||||||
|
defer t.mu.Unlock()
|
||||||
|
|
||||||
|
if t.timer != nil {
|
||||||
|
t.timer.Stop()
|
||||||
|
}
|
||||||
|
|
||||||
|
t.timer = time.AfterFunc(t.duration, t.callback)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *Timer) Stop() {
|
||||||
|
t.mu.Lock()
|
||||||
|
defer t.mu.Unlock()
|
||||||
|
|
||||||
|
if t.timer != nil {
|
||||||
|
t.timer.Stop()
|
||||||
|
t.timer = nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *Timer) Reset() {
|
||||||
|
t.mu.Lock()
|
||||||
|
defer t.mu.Unlock()
|
||||||
|
|
||||||
|
if t.timer != nil {
|
||||||
|
t.timer.Stop()
|
||||||
|
}
|
||||||
|
|
||||||
|
t.timer = time.AfterFunc(t.duration, t.callback)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *Timer) SetDuration(duration time.Duration) {
|
||||||
|
t.mu.Lock()
|
||||||
|
defer t.mu.Unlock()
|
||||||
|
t.duration = duration
|
||||||
|
}
|
||||||
|
|
||||||
|
type PriorityCalculator struct {
|
||||||
|
basePriority uint8
|
||||||
|
currentPriority uint8
|
||||||
|
mu sync.RWMutex
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewPriorityCalculator(basePriority uint8) *PriorityCalculator {
|
||||||
|
return &PriorityCalculator{
|
||||||
|
basePriority: basePriority,
|
||||||
|
currentPriority: basePriority,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *PriorityCalculator) GetPriority() uint8 {
|
||||||
|
pc.mu.RLock()
|
||||||
|
defer pc.mu.RUnlock()
|
||||||
|
return pc.currentPriority
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *PriorityCalculator) DecreasePriority(amount uint8) {
|
||||||
|
pc.mu.Lock()
|
||||||
|
defer pc.mu.Unlock()
|
||||||
|
|
||||||
|
if pc.currentPriority > amount {
|
||||||
|
pc.currentPriority -= amount
|
||||||
|
} else {
|
||||||
|
pc.currentPriority = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *PriorityCalculator) ResetPriority() {
|
||||||
|
pc.mu.Lock()
|
||||||
|
defer pc.mu.Unlock()
|
||||||
|
pc.currentPriority = pc.basePriority
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *PriorityCalculator) SetBasePriority(priority uint8) {
|
||||||
|
pc.mu.Lock()
|
||||||
|
defer pc.mu.Unlock()
|
||||||
|
pc.basePriority = priority
|
||||||
|
pc.currentPriority = priority
|
||||||
|
}
|
||||||
|
|
||||||
|
func ShouldBecomeMaster(localPriority, remotePriority uint8, localIP, remoteIP string) bool {
|
||||||
|
if localPriority > remotePriority {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
if localPriority == remotePriority {
|
||||||
|
return localIP > remoteIP
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func CalculateMasterDownInterval(advertInt uint8) time.Duration {
|
||||||
|
return time.Duration(3*int(advertInt)) * time.Second
|
||||||
|
}
|
||||||
|
|
||||||
|
func CalculateSkewTime(priority uint8) time.Duration {
|
||||||
|
skew := float64(256-int(priority)) / 256.0
|
||||||
|
return time.Duration(skew * float64(time.Second))
|
||||||
|
}
|
||||||
|
|
||||||
|
type StateTransition struct {
|
||||||
|
From State
|
||||||
|
To State
|
||||||
|
Timestamp time.Time
|
||||||
|
Reason string
|
||||||
|
}
|
||||||
|
|
||||||
|
type StateHistory struct {
|
||||||
|
transitions []StateTransition
|
||||||
|
maxSize int
|
||||||
|
mu sync.RWMutex
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewStateHistory(maxSize int) *StateHistory {
|
||||||
|
return &StateHistory{
|
||||||
|
transitions: make([]StateTransition, 0, maxSize),
|
||||||
|
maxSize: maxSize,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *StateHistory) Add(from, to State, reason string) {
|
||||||
|
sh.mu.Lock()
|
||||||
|
defer sh.mu.Unlock()
|
||||||
|
|
||||||
|
transition := StateTransition{
|
||||||
|
From: from,
|
||||||
|
To: to,
|
||||||
|
Timestamp: time.Now(),
|
||||||
|
Reason: reason,
|
||||||
|
}
|
||||||
|
|
||||||
|
sh.transitions = append(sh.transitions, transition)
|
||||||
|
|
||||||
|
if len(sh.transitions) > sh.maxSize {
|
||||||
|
sh.transitions = sh.transitions[1:]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *StateHistory) GetRecent(n int) []StateTransition {
|
||||||
|
sh.mu.RLock()
|
||||||
|
defer sh.mu.RUnlock()
|
||||||
|
|
||||||
|
if n > len(sh.transitions) {
|
||||||
|
n = len(sh.transitions)
|
||||||
|
}
|
||||||
|
|
||||||
|
start := len(sh.transitions) - n
|
||||||
|
result := make([]StateTransition, n)
|
||||||
|
copy(result, sh.transitions[start:])
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sh *StateHistory) String() string {
|
||||||
|
sh.mu.RLock()
|
||||||
|
defer sh.mu.RUnlock()
|
||||||
|
|
||||||
|
if len(sh.transitions) == 0 {
|
||||||
|
return "No state transitions"
|
||||||
|
}
|
||||||
|
|
||||||
|
result := "State transition history:\n"
|
||||||
|
for _, t := range sh.transitions {
|
||||||
|
result += fmt.Sprintf(" %s: %s -> %s (%s)\n",
|
||||||
|
t.Timestamp.Format("2006-01-02 15:04:05"),
|
||||||
|
t.From, t.To, t.Reason)
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
@@ -1,64 +0,0 @@
|
|||||||
package vrrp
|
|
||||||
|
|
||||||
import (
|
|
||||||
"sync"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Timer provides a thread-safe timer with callback support.
|
|
||||||
type Timer struct {
|
|
||||||
duration time.Duration
|
|
||||||
timer *time.Timer
|
|
||||||
callback func()
|
|
||||||
mu sync.Mutex
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewTimer creates a new Timer with the specified duration and callback.
|
|
||||||
func NewTimer(duration time.Duration, callback func()) *Timer {
|
|
||||||
return &Timer{
|
|
||||||
duration: duration,
|
|
||||||
callback: callback,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Start starts or restarts the timer.
|
|
||||||
func (t *Timer) Start() {
|
|
||||||
t.mu.Lock()
|
|
||||||
defer t.mu.Unlock()
|
|
||||||
|
|
||||||
if t.timer != nil {
|
|
||||||
t.timer.Stop()
|
|
||||||
}
|
|
||||||
|
|
||||||
t.timer = time.AfterFunc(t.duration, t.callback)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Stop stops the timer if it's running.
|
|
||||||
func (t *Timer) Stop() {
|
|
||||||
t.mu.Lock()
|
|
||||||
defer t.mu.Unlock()
|
|
||||||
|
|
||||||
if t.timer != nil {
|
|
||||||
t.timer.Stop()
|
|
||||||
t.timer = nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Reset stops the current timer and starts a new one with the same duration.
|
|
||||||
func (t *Timer) Reset() {
|
|
||||||
t.mu.Lock()
|
|
||||||
defer t.mu.Unlock()
|
|
||||||
|
|
||||||
if t.timer != nil {
|
|
||||||
t.timer.Stop()
|
|
||||||
}
|
|
||||||
|
|
||||||
t.timer = time.AfterFunc(t.duration, t.callback)
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetDuration updates the timer's duration for future starts.
|
|
||||||
func (t *Timer) SetDuration(duration time.Duration) {
|
|
||||||
t.mu.Lock()
|
|
||||||
defer t.mu.Unlock()
|
|
||||||
t.duration = duration
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user