github.com/go-spring/spring-base@v1.1.3/log/internal/caller_test.go (about) 1 /* 2 * Copyright 2012-2019 the original author or authors. 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * https://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package internal_test 18 19 import ( 20 "fmt" 21 "testing" 22 23 "github.com/go-spring/spring-base/assert" 24 "github.com/go-spring/spring-base/log/internal" 25 ) 26 27 func TestCaller(t *testing.T) { 28 for i := 0; i < 2; i++ { 29 file, line, loaded := internal.Caller(0, true) 30 if i == 0 { 31 assert.False(t, loaded) 32 } else { 33 assert.True(t, loaded) 34 } 35 _ = fmt.Sprintf("%s:%d\n", file, line) 36 } 37 } 38 39 func BenchmarkCaller(b *testing.B) { 40 // BenchmarkCaller/fast-8 2393902 463.3 ns/op 41 // BenchmarkCaller/slow-8 1470334 814.4 ns/op 42 b.Run("fast", func(b *testing.B) { 43 for i := 0; i < b.N; i++ { 44 file, line, _ := internal.Caller(0, true) 45 _ = fmt.Sprintf("%s:%d\n", file, line) 46 } 47 }) 48 b.Run("slow", func(b *testing.B) { 49 for i := 0; i < b.N; i++ { 50 file, line, _ := internal.Caller(0, false) 51 _ = fmt.Sprintf("%s:%d\n", file, line) 52 } 53 }) 54 }