[config] 初步实现yaml配置读取
This commit is contained in:
parent
eba278904f
commit
226c73711a
|
|
@ -0,0 +1,22 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"sysmonitord/internal/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
configPath := "./config.yaml"
|
||||||
|
|
||||||
|
cfg, err := config.LoadConfig(configPath)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("加载配置失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println("加载配置成功:")
|
||||||
|
fmt.Printf("审计配置: %+v\n", cfg.Audit)
|
||||||
|
fmt.Printf("扫描配置: %+v\n", cfg.Scanner)
|
||||||
|
|
||||||
|
fmt.Printf("审计服务器地址:%s:%d\n", cfg.Audit.Server, cfg.Audit.Port)
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,15 @@
|
||||||
|
server:
|
||||||
|
host: "127.0.0.1"
|
||||||
|
port: 8080
|
||||||
|
|
||||||
|
audit:
|
||||||
|
enabled: true
|
||||||
|
server: "192.168.1.100"
|
||||||
|
port: 9000
|
||||||
|
buffer_size: 1000
|
||||||
|
|
||||||
|
scanner:
|
||||||
|
file:
|
||||||
|
exclude_paths:
|
||||||
|
- /proc
|
||||||
|
- /sys
|
||||||
|
|
@ -0,0 +1,5 @@
|
||||||
|
module sysmonitord
|
||||||
|
|
||||||
|
go 1.26.1
|
||||||
|
|
||||||
|
require gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
|
|
@ -0,0 +1,3 @@
|
||||||
|
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/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
|
@ -0,0 +1,41 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Config struct {
|
||||||
|
Audit AuditConfig `yaml:"audit"`
|
||||||
|
Scanner ScannerConfig `yaml:"scanner"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type AuditConfig struct {
|
||||||
|
Enabled bool `yaml:"enabled"`
|
||||||
|
Server string `yaml:"server"`
|
||||||
|
Port int `yaml:"port"`
|
||||||
|
BufferSize int `yaml:"buffer_size"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ScannerConfig struct {
|
||||||
|
File FileScannerConfig `yaml:"file"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type FileScannerConfig struct {
|
||||||
|
ExcludePaths []string `yaml:"exclude_paths"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoadConfig(path string) (*Config, error) {
|
||||||
|
data, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("无法读取配置文件: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var cfg Config
|
||||||
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
||||||
|
return nil, fmt.Errorf("无法解析配置文件: %w", err)
|
||||||
|
}
|
||||||
|
return &cfg, nil
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user