基于Consul的分布式信号量实现

package main
 
import (
    "fmt"
    "time"
 
    consul "github.com/hashicorp/consul/api"
)
 
// 使用Consul实现分布式信号量
func main() {
    config := consul.DefaultConfig()
    client, err := consul.NewClient(config)
    if err != nil {
        panic(err)
    }
 
    kv := client.KV()
    semaphore := NewConsulSemaphore(kv, "my-key", 3)
 
    for i := 0; i < 5; i++ {
        go func(i int) {
            if err := semaphore.Acquire(nil, time.Second*10); err != nil {
                fmt.Printf("Goroutine %d failed to acquire semaphore\n", i)
                return
            }
            fmt.Printf("Goroutine %d acquired semaphore\n", i)
            time.Sleep(time.Second * 5)
            semaphore.Release()
            fmt.Printf("Goroutine %d released semaphore\n", i)
        }(i)
    }
 
    // 阻塞主线程，以便goroutine能够运行
    time.Sleep(time.Minute)
}

这个示例代码展示了如何使用Consul KV接口来实现一个简单的分布式信号量。它创建了一个客户端连接到本地Consul代理，然后创建了一个新的ConsulSemaphore实例。接下来，它启动了5个goroutine，每个goroutine尝试获取信号量，并在获取后睡眠一段时间来模拟工作负载。每个goroutine在释放信号量之前和之后都会打印出消息。这个例子演示了如何在分布式系统中使用信号量来控制访问某些资源或者进行同步。