]> git.rm.cloudns.org Git - xonotic/xonstat.git/commitdiff
Add a simple check to prevent most blank files.
authorAnt Zucaro <azucaro@gmail.com>
Thu, 4 Jan 2018 21:49:35 +0000 (16:49 -0500)
committerAnt Zucaro <azucaro@gmail.com>
Thu, 4 Jan 2018 21:49:35 +0000 (16:49 -0500)
xonstat/util/xs_parselog.go

index c2a1e743b7f77a24fa8d8e49216c708e8098a0c2..c2f4e77dc49ceccbd78ab283e4a0f7a787b2ccef 100644 (file)
@@ -16,74 +16,76 @@ import "strings"
    anonymize the hashkeys in the requests to prevent exposing player data
    unnecessarily. */
 func main() {
-  var fn = flag.String("file", "xonstat.log", "XonStat log file name")
-  var target_match_id = flag.String("match", "", "the specific match_id (I line) to extract from the log")
-  var anonymize = flag.Bool("anonymize", false, "whether or not to anonymize player hashkeys")
-  flag.Parse()
+       var fn = flag.String("file", "xonstat.log", "XonStat log file name")
+       var target_match_id = flag.String("match", "", "the specific match_id (I line) to extract from the log")
+       var anonymize = flag.Bool("anonymize", false, "whether or not to anonymize player hashkeys")
+       flag.Parse()
 
-  f, err := os.Open(*fn)
-  if err != nil {
-    fmt.Println("Issue opening file")
-    os.Exit(1)
-  }
-  defer f.Close()
+       f, err := os.Open(*fn)
+       if err != nil {
+               fmt.Println("Issue opening file")
+               os.Exit(1)
+       }
+       defer f.Close()
 
-  r := bufio.NewReader(f)
+       r := bufio.NewReader(f)
 
-  var inreq bool
-  var match_id string
-  var game_type_cd string
-  lines := make([]string, 0, 100)
-  hashkeys := make(map[string] int, 500)
-  max_player_id := 0
+       var inreq bool
+       var match_id string
+       var game_type_cd string
+       lines := make([]string, 0, 100)
+       hashkeys := make(map[string]int, 500)
+       max_player_id := 0
 
-  line, err := r.ReadString('\n')
-  for err == nil {
-    switch {
-      case strings.Contains(line, "BEGIN REQUEST BODY"):
-        inreq = true
-      case strings.Contains(line, "END REQUEST BODY"):
-        if *target_match_id == "" || match_id == *target_match_id {
-          create_match_file(game_type_cd, match_id, lines)
-        }
-        inreq = false
-        lines = make([]string, 0, 100)
-      case inreq:
-        if *anonymize && line[0] == 'P' {
-          if line[2:5] != "bot" && line[2:8] != "player" {
-            hashkey := line[2:len(line)-1]
-            if _, ok := hashkeys[hashkey]; !ok {
-              hashkeys[hashkey] = max_player_id
-              max_player_id = max_player_id + 1
-            }
-            line = fmt.Sprintf("P %d\n", hashkeys[hashkey])
-          }
-        }
-        if line[0] == 'I' {
-          match_id = line[2:len(line)-1]
-        }
-        if line[0] == 'G' {
-          game_type_cd = line[2:len(line)-1]
-        }
-        lines = append(lines, line)
-    }
+       line, err := r.ReadString('\n')
+       for err == nil {
+               switch {
+               case strings.Contains(line, "BEGIN REQUEST BODY"):
+                       inreq = true
+               case strings.Contains(line, "END REQUEST BODY"):
+                       if *target_match_id == "" || match_id == *target_match_id {
+                               // a simple heuristic to filter *most* empty games
+                               if len(lines) >= 20 {
+                                       create_match_file(game_type_cd, match_id, lines)
+                               }
+                       }
+                       inreq = false
+                       lines = make([]string, 0, 100)
+               case inreq:
+                       if *anonymize && line[0] == 'P' {
+                               if line[2:5] != "bot" && line[2:8] != "player" {
+                                       hashkey := line[2 : len(line)-1]
+                                       if _, ok := hashkeys[hashkey]; !ok {
+                                               hashkeys[hashkey] = max_player_id
+                                               max_player_id = max_player_id + 1
+                                       }
+                                       line = fmt.Sprintf("P %d\n", hashkeys[hashkey])
+                               }
+                       }
+                       if line[0] == 'I' {
+                               match_id = line[2 : len(line)-1]
+                       }
+                       if line[0] == 'G' {
+                               game_type_cd = line[2 : len(line)-1]
+                       }
+                       lines = append(lines, line)
+               }
 
-    line, err = r.ReadString('\n')
-  }
+               line, err = r.ReadString('\n')
+       }
 }
 
-
 func create_match_file(game_type_cd string, match_id string, body []string) {
-  fn := fmt.Sprintf("%s_%s.txt", game_type_cd, match_id)
-  f_raw, err := os.Create(fn)
-  if err != nil {
-    fmt.Printf("Error creating match_id %s.\n", match_id)
-  }
-  defer f_raw.Close()
+       fn := fmt.Sprintf("%s_%s.txt", game_type_cd, match_id)
+       f_raw, err := os.Create(fn)
+       if err != nil {
+               fmt.Printf("Error creating match_id %s.\n", match_id)
+       }
+       defer f_raw.Close()
 
-  f := bufio.NewWriter(f_raw)
-  for _, v := range body {
-    fmt.Fprint(f, v)
-  }
-  f.Flush()
+       f := bufio.NewWriter(f_raw)
+       for _, v := range body {
+               fmt.Fprint(f, v)
+       }
+       f.Flush()
 }