fix: set scanner buffer (#628)

openimsdk · Jul 30, 2024 · ba25eca · ba25eca
1 parent 04c13b5
commit ba25eca
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/internal/third/log.go b/internal/third/log.go
@@ -20,6 +20,10 @@ import (
 	"time"
 )
 
+const (
+	buffer = 10 * 1024 * 1024
+)
+
 // UploadLogs uploadLogs.
 func (c *Third) UploadLogs(ctx context.Context, line int, ex string, progress Progress) (err error) {
 	if c.logUploadLock.TryLock() {
@@ -160,6 +164,9 @@ func readLastNLines(filename string, n int) ([]string, error) {
 	count := 0
 
 	scanner := bufio.NewScanner(f)
+	buf := make([]byte, buffer)
+	scanner.Buffer(buf, buffer)
+
 	for scanner.Scan() {
 		lines[count%n] = scanner.Text()
 		count++