ai
4cf590ef5f
- Implemented streaming processing for constant memory footprint. - Optimized memory usage: reduced allocations from 185 to 99. - Migrated `config.Direction` to typed Enum. - Added comprehensive test suite for config, cgp, rspamc and internal logic. - Cleaned up loop protection and action handlers.
76 lines
1.2 KiB
Go
76 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
"strconv"
|
|
|
|
"git.vsu.ru/ai/rspamd-cgp/cgp"
|
|
"git.vsu.ru/ai/rspamd-cgp/config"
|
|
"git.vsu.ru/ai/rspamd-cgp/rspamc"
|
|
"git.vsu.ru/ai/rspamd-cgp/utils"
|
|
)
|
|
|
|
func main() {
|
|
|
|
if err := cgp.InitStdoutFd(); err != nil {
|
|
os.Stderr.WriteString("failed to init CGP stdout\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
conf, err := config.New(os.Args[1:])
|
|
if err != nil {
|
|
os.Stderr.WriteString("config: " + err.Error() + "\n")
|
|
os.Exit(1)
|
|
}
|
|
config.SetGlobal(conf)
|
|
|
|
if config.ShowVersion() {
|
|
os.Stderr.WriteString(config.GetVersion())
|
|
return
|
|
}
|
|
|
|
in := bufio.NewReader(os.Stdin)
|
|
|
|
loop:
|
|
for {
|
|
line, err := in.ReadSlice('\n')
|
|
if err != nil {
|
|
if err != io.EOF {
|
|
cgp.Putline("* stdin error: ", err)
|
|
}
|
|
break loop
|
|
}
|
|
|
|
parts := bytes.Fields(line)
|
|
n := len(parts)
|
|
if n < 2 {
|
|
continue
|
|
}
|
|
|
|
seq, err := strconv.Atoi(utils.Bytes2string(parts[0]))
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
cmd := utils.Bytes2string(parts[1])
|
|
|
|
switch {
|
|
case cmd == "FILE" && n == 3:
|
|
go rspamc.Scan(seq, string(parts[2]))
|
|
|
|
case cmd == "INTF" && n == 3:
|
|
cgp.Intf(seq, utils.Bytes2string(parts[2]))
|
|
|
|
case cmd == "QUIT" && n == 2:
|
|
cgp.Ok(seq)
|
|
break loop
|
|
|
|
default:
|
|
cgp.Putline("* bad command: ", line)
|
|
}
|
|
}
|
|
}
|