Go工作流與分布式鎖機制的結合

在Go語言中，工作流和分布式鎖機制可以結合使用，以實現高效、安全的并發控制。工作流通常用于管理和協調多個任務的執行，而分布式鎖機制則用于確保在分布式系統中的資源訪問是同步的。

以下是一個簡單的示例，展示了如何在Go中結合使用工作流和分布式鎖機制：

1. 首先，我們需要一個分布式鎖的實現。這里我們使用go-redis庫來實現一個基于Redis的分布式鎖：

package main

import (
	"context"
	"fmt"
	"time"

	"github.com/go-redis/redis/v8"
)

type RedisLock struct {
	client *redis.Client
	key    string
	value  string
}

func NewRedisLock(client *redis.Client, key, value string) *RedisLock {
	return &RedisLock{
		client: client,
		key:    key,
		value:  value,
	}
}

func (l *RedisLock) Lock() error {
	ctx := context.Background()
	for {
		if l.client.SetNX(ctx, l.key, l.value, time.Second*10).Val() {
			return nil
		}
		time.Sleep(time.Millisecond * 100)
	}
}

func (l *RedisLock) Unlock() error {
	ctx := context.Background()
	val, err := l.client.Get(ctx, l.key).Result()
	if err != nil {
		return err
	}
	if val == l.value {
		l.client.Del(ctx, l.key)
	}
	return nil
}

2. 接下來，我們創建一個工作流管理器，用于協調多個任務的執行：

package main

import (
	"sync"
)

type WorkflowManager struct {
	lock sync.Mutex
	wg   sync.WaitGroup
}

func NewWorkflowManager() *WorkflowManager {
	return &WorkflowManager{}
}

func (m *WorkflowManager) AddTask(task func()) {
	m.lock.Lock()
	defer m.lock.Unlock()
	m.wg.Add(1)
	go func() {
		defer m.wg.Done()
		task()
	}()
}

func (m *WorkflowManager) Wait() {
	m.wg.Wait()
}

3. 最后，我們將工作流管理器和分布式鎖結合起來，實現一個簡單的并發控制示例：

package main

import (
	"fmt"
	"os"

	"github.com/go-redis/redis/v8"
)

func main() {
	// 初始化Redis客戶端
	client := redis.NewClient(&redis.Options{
		Addr:     "localhost:6379",
		Password: "", // no password set
		DB:       0,  // use default DB
	})

	// 初始化工作流管理器
	manager := NewWorkflowManager()

	// 添加任務到工作流
	for i := 0; i < 10; i++ {
		manager.AddTask(func() {
			// 創建分布式鎖
			lock := NewRedisLock(client, "my_lock", fmt.Sprintf("worker_%d", os.Getpid()))

			// 獲取鎖
			err := lock.Lock()
			if err != nil {
				fmt.Println("Failed to acquire lock:", err)
				return
			}
			defer lock.Unlock()

			// 執行任務
			fmt.Println("Executing task...")
			time.Sleep(time.Second)
		})
	}

	// 等待所有任務完成
	manager.Wait()
}

在這個示例中，我們創建了一個工作流管理器，用于協調多個任務的執行。每個任務都會嘗試獲取一個分布式鎖，然后執行相應的操作。通過結合工作流和分布式鎖機制，我們可以確保在分布式系統中的資源訪問是同步的。