Go语言中统计函数或方法调用次数的实用技巧

package main

import (
    "fmt"
    "sync/atomic" // 导入atomic包
)

var functionCallCount uint64 // 定义全局计数器，使用uint64以支持大数值

// Foo 是一个示例函数，每次调用都会递增全局计数器
func Foo() {
    atomic.AddUint64(&functionCallCount, 1) // 原子性地递增计数器
    fmt.Println("Foo() called!")
}

func main() {
    Foo()
    Foo()
    Foo()
    fmt.Printf("Foo() 已被调用 %d 次\n", functionCallCount)
}

package main

import (
    "fmt"
    "sync/atomic"
)

// Foo 是一个通过闭包创建的函数，它拥有一个私有且独立的调用计数器
var Foo = func() func() uint64 {
    var called uint64 // 闭包捕获的私有计数器
    return func() uint64 {
        atomic.AddUint64(&called, 1) // 原子性递增私有计数器
        fmt.Println("Foo() called!")
        return called // 返回当前调用次数
    }
}() // 注意这里的立即执行，Foo现在是一个可直接调用的函数

func main() {
    Foo()      // 第一次调用
    c := Foo() // 第二次调用，并获取当前计数
    fmt.Printf("Foo() 已被调用 %d 次\n", c)
}

package main

import (
    "fmt"
    "sync/atomic"
)

// RequestHandler 是一个示例结构体，包含一个方法调用计数器
type RequestHandler struct {
    CalledCount uint64 // 结构体字段作为方法调用计数器
}

// HandleRequest 是 RequestHandler 的一个方法，每次调用都会递增其CalledCount
func (h *RequestHandler) HandleRequest() {
    atomic.AddUint64(&h.CalledCount, 1) // 原子性递增结构体实例的计数器
    fmt.Println("RequestHandler.HandleRequest() called!")
}

func main() {
    var myHandler RequestHandler // 创建一个RequestHandler实例
    myHandler.HandleRequest()
    myHandler.HandleRequest()
    myHandler.HandleRequest()
    fmt.Printf("myHandler.HandleRequest() 已被调用 %d 次\n", myHandler.CalledCount)

    var anotherHandler RequestHandler // 另一个独立的实例
    anotherHandler.HandleRequest()
    fmt.Printf("anotherHandler.HandleRequest() 已被调用 %d 次\n", anotherHandler.CalledCount)
}

package main

import (
    "fmt"
    "sync/atomic"
    // 假设有一个外部包，其中包含一个名为ExternalFunction的函数
    // import "externalpackage"
)

// 模拟一个外部包函数，实际应用中会是外部库提供的函数
func ExternalFunction() {
    fmt.Println("ExternalPackage.ExternalFunction() called!")
}

var externalFunctionCallCount uint64

// WrappedExternalFunction 是ExternalFunction的包装器
func WrappedExternalFunction() {
    atomic.AddUint64(&externalFunctionCallCount, 1) // 递增计数器
    ExternalFunction()                               // 调用原始外部函数
}

func main() {
    WrappedExternalFunction()
    WrappedExternalFunction()
    fmt.Printf("ExternalFunction() (通过包装器) 已被调用 %d 次\n", externalFunctionCallCount)
}