Go语言中接口类型到唯一ID的健壮映射策略

type Task interface {
    Do() error
    ID() int64 // 新增：获取任务唯一ID的方法
}

import "math/rand" // 用于生成随机ID

type XTask struct {
    id int64 // 存储唯一ID
    // 其他业务数据，例如任务参数等
}

func (t *XTask) Do() error {
    // 实际的任务执行逻辑
    fmt.Println("XTask Do method called.")
    return nil
}

func (t *XTask) ID() int64 {
    return t.id
}

import (
    "fmt"
    "math/rand"
    "sync" // 用于并发安全
    "time" // 用于初始化rand种子
)

// taskRegistry 存储 ID 到 Task 实例的映射
var taskRegistry = make(map[int64]Task)
var registryMutex sync.RWMutex // 保护 taskRegistry 的并发访问

func init() {
    rand.Seed(time.Now().UnixNano()) // 初始化随机数种子
}

func Register(t Task) int64 {
    registryMutex.Lock()
    defer registryMutex.Unlock()

    var id int64
    for { // 循环生成唯一ID，直到找到一个未使用的ID
        id = rand.Int63()
        if _, exists := taskRegistry[id]; !exists {
            break
        }
    }
    taskRegistry[id] = t // 存储 ID 到 Task 的映射
    return id
}

// GetTaskByID 提供通过ID查找Task实例的功能
func GetTaskByID(id int64) (Task, bool) {
    registryMutex.RLock()
    defer registryMutex.RUnlock()
    task, exists := taskRegistry[id]
    return task, exists
}

func NewXTask(/* task parameters... */) *XTask {
    t := &XTask{ /* 初始化 XTask 的其他成员 */ }
    t.id = Register(t) // 在构造时获取并设置ID
    // 可能更多的初始化逻辑
    return t
}

package main

import (
    "fmt"
    "math/rand"
    "sync"
    "time"
)

// Task 接口定义，包含 Do 和 ID 方法
type Task interface {
    Do() error
    ID() int64
}

// XTask 是 Task 接口的一个实现
type XTask struct {
    id   int64
    name string // 示例：其他业务数据
}

// NewXTask 是 XTask 的构造函数
func NewXTask(name string) *XTask {
    t := &XTask{name: name}
    t.id = Register(t) // 在构造时注册并获取ID
    return t
}

// Do 实现 Task 接口的 Do 方法
func (t *XTask) Do() error {
    fmt.Printf("Task %s (ID: %x) is doing its work.\n", t.name, t.id)
    return nil
}

// ID 实现 Task 接口的 ID 方法
func (t *XTask) ID() int64 {
    return t.id
}

// taskRegistry 存储 ID 到 Task 实例的映射
var taskRegistry = make(map[int64]Task)
var registryMutex sync.RWMutex // 保护 taskRegistry 的并发访问

func init() {
    rand.Seed(time.Now().UnixNano()) // 初始化随机数种子
}

// Register 注册一个 Task 实例，并为其分配一个唯一的 ID
func Register(t Task) int64 {
    registryMutex.Lock()
    defer registryMutex.Unlock()

    var id int64
    for { // 循环生成唯一 ID，直到找到一个未使用的 ID
        id = rand.Int63()
        if _, exists := taskRegistry[id]; !exists {
            break
        }
    }
    taskRegistry[id] = t // 存储 ID 到 Task 的映射
    return id
}

// GetTaskByID 提供通过 ID 查找 Task 实例的功能
func GetTaskByID(id int64) (Task, bool) {
    registryMutex.RLock()
    defer registryMutex.RUnlock()
    task, exists := taskRegistry[id]
    return task, exists
}

func main() {
    // 创建并注册两个 Task 实例
    t1 := NewXTask("TaskA")
    t2 := NewXTask("TaskB")

    fmt.Printf("TaskA ID: %x\n", t1.ID())
    fmt.Printf("TaskB ID: %x\n", t2.ID())

    // 演示通过 ID 查找 Task
    if foundTask, ok := GetTaskByID(t1.ID()); ok {
        fmt.Printf("Found task with ID %x: ", t1.ID())
        foundTask.Do()
    }

    if foundTask, ok := GetTaskByID(t2.ID()); ok {
        fmt.Printf("Found task with ID %x: ", t2.ID())
        foundTask.Do()
    }

    // 尝试查找一个不存在的 ID
    if _, ok := GetTaskByID(999); !ok {
        fmt.Println("Task with ID 999 not found.")
    }
}