mirror of
https://github.com/go-gitea/gitea.git
synced 2026-02-15 07:26:19 +00:00
0d8bd7720d
1. fix a performance regression when using line-by-line highlighting * the root cause is that chroma's `lexers.Get` is slow and a lexer cache is missing during recent changes 2. clarify the chroma lexer detection behavior * now we fully manage our logic to detect lexer, and handle overriding problems, everything is fully under control 3. clarify "code analyze" behavior, now only 2 usages: * only use file name and language to detect lexer (very fast), mainly for "diff" page which contains a lot of files * if no lexer is detected by file name and language, use code content to detect again (slow), mainly for "view file" or "blame" page, which can get best result 4. fix git diff bug, it caused "broken pipe" error for large diff files
91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
// Copyright 2026 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package highlight
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/alecthomas/chroma/v2/lexers"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func BenchmarkDetectChromaLexerByFileName(b *testing.B) {
|
|
for b.Loop() {
|
|
// BenchmarkDetectChromaLexerByFileName-12 18214717 61.35 ns/op
|
|
DetectChromaLexerByFileName("a.sql", "")
|
|
}
|
|
}
|
|
|
|
func BenchmarkDetectChromaLexerWithAnalyze(b *testing.B) {
|
|
b.StopTimer()
|
|
code := []byte(strings.Repeat("SELECT * FROM table;\n", 1000))
|
|
b.StartTimer()
|
|
for b.Loop() {
|
|
// BenchmarkRenderCodeSlowGuess-12 87946 13310 ns/op
|
|
detectChromaLexerWithAnalyze("a", "", code)
|
|
}
|
|
}
|
|
|
|
func BenchmarkChromaAnalyze(b *testing.B) {
|
|
b.StopTimer()
|
|
code := strings.Repeat("SELECT * FROM table;\n", 1000)
|
|
b.StartTimer()
|
|
for b.Loop() {
|
|
// comparing to detectChromaLexerWithAnalyze (go-enry), "chroma/lexers.Analyse" is very slow
|
|
// BenchmarkChromaAnalyze-12 519 2247104 ns/op
|
|
lexers.Analyse(code)
|
|
}
|
|
}
|
|
|
|
func BenchmarkRenderCodeByLexer(b *testing.B) {
|
|
b.StopTimer()
|
|
code := strings.Repeat("SELECT * FROM table;\n", 1000)
|
|
lexer := DetectChromaLexerByFileName("a.sql", "")
|
|
b.StartTimer()
|
|
for b.Loop() {
|
|
// Really slow .......
|
|
// BenchmarkRenderCodeByLexer-12 22 47159038 ns/op
|
|
RenderCodeByLexer(lexer, code)
|
|
}
|
|
}
|
|
|
|
func TestDetectChromaLexer(t *testing.T) {
|
|
globalVars().highlightMapping[".my-html"] = "HTML"
|
|
t.Cleanup(func() { delete(globalVars().highlightMapping, ".my-html") })
|
|
|
|
cases := []struct {
|
|
fileName string
|
|
language string
|
|
content string
|
|
expected string
|
|
}{
|
|
{"test.py", "", "", "Python"},
|
|
|
|
{"any-file", "javascript", "", "JavaScript"},
|
|
{"any-file", "", "/* vim: set filetype=python */", "Python"},
|
|
{"any-file", "", "", "fallback"},
|
|
|
|
{"test.fs", "", "", "FSharp"},
|
|
{"test.fs", "F#", "", "FSharp"},
|
|
{"test.fs", "", "let x = 1", "FSharp"},
|
|
|
|
{"test.c", "", "", "C"},
|
|
{"test.C", "", "", "C++"},
|
|
{"OLD-CODE.PAS", "", "", "ObjectPascal"},
|
|
{"test.my-html", "", "", "HTML"},
|
|
|
|
{"a.php", "", "", "PHP"},
|
|
{"a.sql", "", "", "SQL"},
|
|
{"dhcpd.conf", "", "", "ISCdhcpd"},
|
|
{".env.my-production", "", "", "Bash"},
|
|
}
|
|
for _, c := range cases {
|
|
lexer := detectChromaLexerWithAnalyze(c.fileName, c.language, []byte(c.content))
|
|
if assert.NotNil(t, lexer, "case: %+v", c) {
|
|
assert.Equal(t, c.expected, lexer.Config().Name, "case: %+v", c)
|
|
}
|
|
}
|
|
}
|