feat: yocsef api

JingYiJun · May 24, 2024 · d191aca · d191aca
1 parent 9fd3858
commit d191aca
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/service/yocsef.go b/service/yocsef.go
@@ -80,11 +80,12 @@ func InferYocsef(
 		if err != nil {
 			return nil, err
 		}
+		line = bytes.Trim(line, " \n\r")
 		if strings.HasPrefix(string(line), "event") {
 			continue
 		}
 		if strings.HasPrefix(string(line), "data") {
-			line = line[6:]
+			line = bytes.TrimPrefix(line, []byte("data:"))
 		}
 		line = bytes.Trim(line, " \n\r")
 		if len(line) == 0 {
@@ -98,7 +99,7 @@ func InferYocsef(
 		var response map[string]any
 		err = json.Unmarshal(line, &response)
 		if err != nil {
-			return nil, err
+			continue
 		}
 
 		var ok bool