kor-elf-shield/kor-elf-shield
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 14 Wiki Activity

Refactor analyzer to use a shared log channel

Browse Source
This commit is contained in:
kor-elf
2026-01-05 22:27:45 +05:00
parent 1f8be77ab3
commit ba23474eab
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/daemon/analyzer/analyzer.go
+9 -6
View File
@@ -22,6 +22,8 @@ type analyzer struct {
notify notifications.Notifications notify notifications.Notifications
systemd analyzerLog.Systemd systemd analyzerLog.Systemd
analysis analyzerLog.Analysis analysis analyzerLog.Analysis
logChan chan analysisServices.Entry
} }
func New(config config2.Config, logger log.Logger, notify notifications.Notifications) Analyzer { func New(config config2.Config, logger log.Logger, notify notifications.Notifications) Analyzer {
@@ -39,23 +41,23 @@ func New(config config2.Config, logger log.Logger, notify notifications.Notifica
notify: notify, notify: notify,
systemd: systemdService, systemd: systemdService,
analysis: analysisService, analysis: analysisService,
logChan: make(chan analysisServices.Entry, 1000),
} }
} }
func (a *analyzer) Run(ctx context.Context) { func (a *analyzer) Run(ctx context.Context) {
logChan := make(chan analysisServices.Entry, 1000) go a.systemd.Run(ctx, a.logChan)
go a.processLogs(ctx)
go a.systemd.Run(ctx, logChan)
go a.processLogs(ctx, logChan)
a.logger.Debug("Analyzer is start") a.logger.Debug("Analyzer is start")
} }
func (a *analyzer) processLogs(ctx context.Context, logChan <-chan analysisServices.Entry) { func (a *analyzer) processLogs(ctx context.Context) {
for { for {
select { select {
case <-ctx.Done(): case <-ctx.Done():
return return
case entry := <-logChan: case entry := <-a.logChan:
a.logger.Debug(fmt.Sprintf("Received log entry: %s", entry)) a.logger.Debug(fmt.Sprintf("Received log entry: %s", entry))
switch entry.Unit { switch entry.Unit {
case "ssh.service": case "ssh.service":
@@ -74,6 +76,7 @@ func (a *analyzer) Close() error {
if err := a.systemd.Close(); err != nil { if err := a.systemd.Close(); err != nil {
return err return err
} }
close(a.logChan)
a.logger.Debug("Analyzer is stop") a.logger.Debug("Analyzer is stop")