windy/go-caatsm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: Update aviation_parser.go and pattern.go to use BodyConfig struct instead of config.BodyConfig

Browse Source
This commit is contained in:
windyboy
2024-08-06 12:44:32 +08:00
parent bef663ca3d
commit a57e924501
3 changed files with 166 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/config/config.go
+58 -58
View File
@@ -63,10 +63,10 @@ const (
// - ssr: the alphanumeric SSR code // - ssr: the alphanumeric SSR code
// - departure: the four-letter departure airport code // - departure: the four-letter departure airport code
// - arrival: the four-letter arrival airport code // - arrival: the four-letter arrival airport code
arrPatternString = `^\((?P<category>[A-Z]{3})` + // arrPatternString = `^\((?P<category>[A-Z]{3})` +
`-(?P<number>[A-Z0-9]+)(\/?(?P<ssr>[A-Z0-9]+))?` + // `-(?P<number>[A-Z0-9]+)(\/?(?P<ssr>[A-Z0-9]+))?` +
`-(?P<dep>[A-Z]{4})` + // `-(?P<dep>[A-Z]{4})` +
`-(?P<arr>[A-Z]{4})(?P<arr_time>\d{4})\)$` // `-(?P<arr>[A-Z]{4})(?P<arr_time>\d{4})\)$`
// depPatternString represents the regular expression pattern used to match departure patterns. // depPatternString represents the regular expression pattern used to match departure patterns.
// The pattern matches strings in the format: "(TYPE-NUMBER-SSR-DEPARTURE-DEPARTURE_TIME-ARRIVAL)". // The pattern matches strings in the format: "(TYPE-NUMBER-SSR-DEPARTURE-DEPARTURE_TIME-ARRIVAL)".
@@ -77,7 +77,7 @@ const (
// - departure: the four-letter departure airport code // - departure: the four-letter departure airport code
// - departure_time: the four-digit departure time // - departure_time: the four-digit departure time
// - arrival: the four-letter arrival airport code // - arrival: the four-letter arrival airport code
depPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>[A-Z0-9]+)(\/(?P<ssr>[A-Z0-9]+))?-(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})-(?P<arr>[A-Z]{4})\)$` // depPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>[A-Z0-9]+)(\/(?P<ssr>[A-Z0-9]+))?-(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})-(?P<arr>[A-Z]{4})\)$`
// fplPatternString represents the regular expression pattern used to match flight plan patterns. // fplPatternString represents the regular expression pattern used to match flight plan patterns.
// The pattern matches strings in the format: "(CATEGORY-NUMBER-INDICATOR-AIRCRAFT-SURVE-DEPARTURE-DEPARTURE_TIME-SPEED-LEVEL-ROUTE-DESTINATION-ESTT-ALTER-OTHER)". // The pattern matches strings in the format: "(CATEGORY-NUMBER-INDICATOR-AIRCRAFT-SURVE-DEPARTURE-DEPARTURE_TIME-SPEED-LEVEL-ROUTE-DESTINATION-ESTT-ALTER-OTHER)".
@@ -97,65 +97,65 @@ const (
// - alter: one or more sequences of a whitespace character followed by exactly four uppercase letters // - alter: one or more sequences of a whitespace character followed by exactly four uppercase letters
// - other: any other relevant information, starting with three uppercase letters followed by a forward slash and zero or more characters (including newline) // - other: any other relevant information, starting with three uppercase letters followed by a forward slash and zero or more characters (including newline)
fplPatternString = `\((?P<category>[A-Z]{3})-(?P<number>[A-Z]+\d+)-(?P<indicator>[A-Z]{2})\n-(?P<aircraft>[A-Z]+\d+\/?[A-Z]?)\n?-(?P<surve>.*)\n?-(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})\n?-(?P<speed>[A-Z]+\d+)(?P<level>[A-Z0-9]+)\s+(?P<route>(.|\n)+)\n-(?P<dest>[A-Z]{4})(?P<estt>\d{4})\s?(?P<alter>(\s[A-Z]{4})+)\n?-([A-Z]{3}\/(?:[A-Z]{4}\d{4}\s?)+)?(?P<other>(?m)[A-Z]{3}\/(.|\n)*)\)$` // fplPatternString = `\((?P<category>[A-Z]{3})-(?P<number>[A-Z]+\d+)-(?P<indicator>[A-Z]{2})\n-(?P<aircraft>[A-Z]+\d+\/?[A-Z]?)\n?-(?P<surve>.*)\n?-(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})\n?-(?P<speed>[A-Z]+\d+)(?P<level>[A-Z0-9]+)\s+(?P<route>(.|\n)+)\n-(?P<dest>[A-Z]{4})(?P<estt>\d{4})\s?(?P<alter>(\s[A-Z]{4})+)\n?-([A-Z]{3}\/(?:[A-Z]{4}\d{4}\s?)+)?(?P<other>(?m)[A-Z]{3}\/(.|\n)*)\)$`
cnlPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>\w+\d+)-?(?P<dep>[A-Z]{4})?-?(?<arr>[A-Z]{4})\)$` // cnlPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>\w+\d+)-?(?P<dep>[A-Z]{4})?-?(?<arr>[A-Z]{4})\)$`
dlaPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>\w+\d+)-?(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})?-?(?<arr>[A-Z]{4})(?<arr_time>\d{4})?\)$` // dlaPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>\w+\d+)-?(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})?-?(?<arr>[A-Z]{4})(?<arr_time>\d{4})?\)$`
) )
// Initialize the bodyPatterns map // Initialize the bodyPatterns map
var bodyPatterns = map[string]BodyConfig{ // var bodyPatterns = map[string]BodyConfig{
"ARR": { // "ARR": {
Patterns: []PatternConfig{ // Patterns: []PatternConfig{
{ // {
Pattern: arrPatternString, // Pattern: arrPatternString,
Comments: "Pattern for ARR message", // Comments: "Pattern for ARR message",
Expression: regexp.MustCompile(arrPatternString), // Expression: regexp.MustCompile(arrPatternString),
}, // },
}, // },
}, // },
"DEP": { // "DEP": {
Patterns: []PatternConfig{ // Patterns: []PatternConfig{
{ // {
Pattern: depPatternString, // Pattern: depPatternString,
Comments: "Pattern for DEP message", // Comments: "Pattern for DEP message",
Expression: regexp.MustCompile(depPatternString), // Expression: regexp.MustCompile(depPatternString),
}, // },
}, // },
}, // },
"FPL": { // "FPL": {
Patterns: []PatternConfig{ // Patterns: []PatternConfig{
{ // {
Pattern: fplPatternString, // Pattern: fplPatternString,
Comments: "Pattern for FPL message", // Comments: "Pattern for FPL message",
Expression: regexp.MustCompile(fplPatternString), // Expression: regexp.MustCompile(fplPatternString),
}, // },
}, // },
}, // },
"CNL": { // "CNL": {
Patterns: []PatternConfig{ // Patterns: []PatternConfig{
{ // {
Pattern: cnlPatternString, // Pattern: cnlPatternString,
Comments: "Pattern for CNL message", // Comments: "Pattern for CNL message",
Expression: regexp.MustCompile(cnlPatternString), // Expression: regexp.MustCompile(cnlPatternString),
}, // },
}, // },
}, // },
"DLA": { // "DLA": {
Patterns: []PatternConfig{ // Patterns: []PatternConfig{
{ // {
Pattern: dlaPatternString, // Pattern: dlaPatternString,
Comments: "Pattern for DLA message", // Comments: "Pattern for DLA message",
Expression: regexp.MustCompile(dlaPatternString), // Expression: regexp.MustCompile(dlaPatternString),
}, // },
}, // },
}, // },
} // }
func GetBodyPatterns() map[string]BodyConfig { // func GetBodyPatterns() map[string]BodyConfig {
return bodyPatterns // return bodyPatterns
} // }
func SetMyConfig(cfg *Config) { func SetMyConfig(cfg *Config) {
MyConfig = cfg MyConfig = cfg
internal/parsers/aviation_parser.go
+4 -5
View File
@@ -1,7 +1,6 @@
package parsers package parsers
import ( import (
"caatsm/internal/config"
"caatsm/internal/domain" "caatsm/internal/domain"
"caatsm/pkg/utils" "caatsm/pkg/utils"
"fmt" "fmt"
@@ -71,24 +70,24 @@ var (
type BodyParser struct { type BodyParser struct {
body string body string
bodyPatterns map[string]config.BodyConfig bodyPatterns map[string]BodyConfig
mu sync.Mutex mu sync.Mutex
} }
func NewBodyParser(body string) *BodyParser { func NewBodyParser(body string) *BodyParser {
return &BodyParser{ return &BodyParser{
bodyPatterns: config.GetBodyPatterns(), bodyPatterns: bodyPatterns,
body: body, body: body,
} }
} }
func (bp *BodyParser) GetBodyPatterns() map[string]config.BodyConfig { func (bp *BodyParser) GetBodyPatterns() map[string]BodyConfig {
bp.mu.Lock() bp.mu.Lock()
defer bp.mu.Unlock() defer bp.mu.Unlock()
return bp.bodyPatterns return bp.bodyPatterns
} }
func (bp *BodyParser) SetBodyPatterns(patterns map[string]config.BodyConfig) { func (bp *BodyParser) SetBodyPatterns(patterns map[string]BodyConfig) {
bp.mu.Lock() bp.mu.Lock()
defer bp.mu.Unlock() defer bp.mu.Unlock()
bp.bodyPatterns = patterns bp.bodyPatterns = patterns
internal/parsers/pattern.go
+104 -5
View File
@@ -1,20 +1,119 @@
package parsers package parsers
import ( import (
"caatsm/internal/config"
"regexp" "regexp"
) )
type BodyConfig struct {
Patterns []PatternConfig
}
type PatternConfig struct {
Pattern string
Comments string
Expression *regexp.Regexp
}
const (
arrPatternString = `^\((?P<category>[A-Z]{3})` +
`-(?P<number>[A-Z0-9]+)(\/?(?P<ssr>[A-Z0-9]+))?` +
`-(?P<dep>[A-Z]{4})` +
`-(?P<arr>[A-Z]{4})(?P<arr_time>\d{4})\)$`
// depPatternString represents the regular expression pattern used to match departure patterns.
// The pattern matches strings in the format: "(TYPE-NUMBER-SSR-DEPARTURE-DEPARTURE_TIME-ARRIVAL)".
// The pattern captures the following named groups:
// - category: the three-letter category code
// - number: the alphanumeric flight number
// - ssr: the alphanumeric SSR code
// - departure: the four-letter departure airport code
// - departure_time: the four-digit departure time
// - arrival: the four-letter arrival airport code
depPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>[A-Z0-9]+)(\/(?P<ssr>[A-Z0-9]+))?-(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})-(?P<arr>[A-Z]{4})\)$`
// fplPatternString represents the regular expression pattern used to match flight plan patterns.
// The pattern matches strings in the format: "(CATEGORY-NUMBER-INDICATOR-AIRCRAFT-SURVE-DEPARTURE-DEPARTURE_TIME-SPEED-LEVEL-ROUTE-DESTINATION-ESTT-ALTER-OTHER)".
// The pattern captures the following named groups:
// - category: the three-letter category code
// - number: the alphanumeric flight number
// - indicator: the two-letter indicator
// - aircraft: the alphanumeric aircraft type, optionally followed by a slash and an uppercase letter
// - surve: any character sequence (surveillance information)
// - departure: the four-letter departure airport code
// - departure_time: the four-digit departure time
// - speed: one or more uppercase letters followed by one or more digits (e.g., N123)
// - level: one or more alphanumeric characters (e.g., FL350)
// - route: one or more characters (including newline) representing the flight route
// - destination: the four-letter destination airport code
// - estt: the four-digit estimated time of arrival
// - alter: one or more sequences of a whitespace character followed by exactly four uppercase letters
// - other: any other relevant information, starting with three uppercase letters followed by a forward slash and zero or more characters (including newline)
fplPatternString = `\((?P<category>[A-Z]{3})-(?P<number>[A-Z]+\d+)-(?P<indicator>[A-Z]{2})\n-(?P<aircraft>[A-Z]+\d+\/?[A-Z]?)\n?-(?P<surve>.*)\n?-(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})\n?-(?P<speed>[A-Z]+\d+)(?P<level>[A-Z0-9]+)\s+(?P<route>(.|\n)+)\n-(?P<dest>[A-Z]{4})(?P<estt>\d{4})\s?(?P<alter>(\s[A-Z]{4})+)\n?-([A-Z]{3}\/(?:[A-Z]{4}\d{4}\s?)+)?(?P<other>(?m)[A-Z]{3}\/(.|\n)*)\)$`
cnlPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>\w+\d+)-?(?P<dep>[A-Z]{4})?-?(?<arr>[A-Z]{4})\)$`
dlaPatternString = `^\((?P<category>[A-Z]{3})-(?P<number>\w+\d+)-?(?P<dep>[A-Z]{4})(?P<dep_time>\d{4})?-?(?<arr>[A-Z]{4})(?<arr_time>\d{4})?\)$`
)
var ( var (
bodyTypePattern = regexp.MustCompile(`^\(([A-Z]{3})(.*\n?)+\)$`) bodyTypePattern = regexp.MustCompile(`^\(([A-Z]{3})(.*\n?)+\)$`)
bodyPatterns = map[string]BodyConfig{
"ARR": {
Patterns: []PatternConfig{
{
Pattern: arrPatternString,
Comments: "Pattern for ARR message",
Expression: regexp.MustCompile(arrPatternString),
},
},
},
"DEP": {
Patterns: []PatternConfig{
{
Pattern: depPatternString,
Comments: "Pattern for DEP message",
Expression: regexp.MustCompile(depPatternString),
},
},
},
"FPL": {
Patterns: []PatternConfig{
{
Pattern: fplPatternString,
Comments: "Pattern for FPL message",
Expression: regexp.MustCompile(fplPatternString),
},
},
},
"CNL": {
Patterns: []PatternConfig{
{
Pattern: cnlPatternString,
Comments: "Pattern for CNL message",
Expression: regexp.MustCompile(cnlPatternString),
},
},
},
"DLA": {
Patterns: []PatternConfig{
{
Pattern: dlaPatternString,
Comments: "Pattern for DLA message",
Expression: regexp.MustCompile(dlaPatternString),
},
},
},
}
) )
func FindPatterns(messageBody string) *config.BodyConfig { func FindPatterns(messageBody string) *BodyConfig {
if match := bodyTypePattern.FindStringSubmatch(messageBody); len(match) > 1 { if match := bodyTypePattern.FindStringSubmatch(messageBody); len(match) > 1 {
name := match[1] name := match[1]
patters := config.GetBodyPatterns() patters := bodyPatterns
if body, found := patters[name]; found { if bodyConfig, found := patters[name]; found {
return &body return &bodyConfig
} }
} }
return nil return nil