diff options
Diffstat (limited to 'cleaner.go')
| -rw-r--r-- | cleaner.go | 12 |
1 files changed, 6 insertions, 6 deletions
@@ -64,8 +64,8 @@ func cleanData(conf Config, db_user, db_pass, db_conn, db_name string) (err erro } //save cleaned data - for _, cd := range cDat { - err = insertCleanData(tx, cd) + for ix := range cDat { + err = insertCleanData(tx, &cDat[ix]) if err != nil { tx.Rollback() log.Println("Failed to save cleaned data") @@ -124,8 +124,8 @@ func clean(rDat []RawData, conf Config) (cDat []CleanData, err error) { // collect all ips so we can query for their ip blocks ips := make(map[string]struct{}) for _, rd := range rDat { - ips[rd.ipSrc] = nil - ips[rd.ipDst] = nil + ips[rd.ipSrc] = struct{}{} + ips[rd.ipDst] = struct{}{} } var iplist []string @@ -146,8 +146,8 @@ func clean(rDat []RawData, conf Config) (cDat []CleanData, err error) { } cDat = append(cDat, CleanData{ - ipbSrc: ips[rd.ipSrc].ipBlock, - ipbDst: ips[rd.ipDst].ipBlock, + ipbSrc: pairs[rd.ipSrc], + ipbDst: pairs[rd.ipDst], asSrc: rd.asSrc, asDst: rd.asDst, portSrc: rd.portSrc, |