Go语言开发实现分布式流式计算系统的方法与实践

package main

import (
    "fmt"
    "sync"
)

type Job struct {
    ID     int
    Input  string
    Result map[string]int
}

type Worker struct {
    ID  int
    Job chan Job
    wg  *sync.WaitGroup
}

func (w *Worker) Process(input string) map[string]int {
    result := make(map[string]int)
    // 处理逻辑，此处以Word Count为例
    words := strings.Split(input, " ")
    for _, word := range words {
        result[word]++
    }
    return result
}

func (w *Worker) Run() {
    defer w.wg.Done()
    for job := range w.Job {
        result := w.Process(job.Input)
        job.Result = result
        fmt.Printf("Worker %d completed job %d
", w.ID, job.ID)
    }
}

func main() {
    // 初始化Job Manager和Worker
    jobManager := make(chan Job)
    workers := []*Worker{}
    var wg sync.WaitGroup

    // 启动多个Worker协程
    for i := 0; i < numWorkers; i++ {
        wg.Add(1)
        worker := &Worker{
            ID:  i,
            Job: jobManager,
            wg:  &wg,
        }
        workers = append(workers, worker)
        go worker.Run()
    }

    // 创建任务并发送给Job Manager
    for i := 0; i < numJobs; i++ {
        job := Job{
            ID:    i,
            Input: "Hello World",
        }
        jobManager <- job
    }

    close(jobManager)
    wg.Wait()

    // 处理计算结果
    results := make(map[string]int)
    for _, worker := range workers {
        for word, count := range worker.Result {
            results[word] += count
        }
    }

    // 打印结果
    for word, count := range results {
        fmt.Printf("%s: %d
", word, count)
    }
}