Files
goLearn/timu.md
2025-08-15 03:55:10 +08:00

3.5 KiB
Raw Blame History

​​基础要求​​​​生产者​​: 2 个协程,每秒生成 1 个随机数0-100发送到缓冲通道容量为 10。 使用 sync.Mutex保护以下操作 记录每个数字的生成次数map[int]int 统计总生产数量int消费者 3 个协程从通道读取数据并打印格式消费者ID: 处理数据X (历史出现次数: Y)。 使用 sync.RWMutex保护对 map[int]int的读取查询历史次数写操作仍由 sync.Mutex保护。主协程 5 秒后关闭通道,等待所有协程退出。 程序退出前,打印最终统计数据: 总生产/消费次数 每个数字的出现频率(按频率降序排列)​​进阶要求​​​​性能优化​​: 消费者读取历史次数时,必须使用 RLock()允许多个消费者并发查询。 生产者更新 map和计数器时用最短的锁持有时间例如合并写锁操作异常处理 若通道已满,生产者需打印警告并丢弃数据。 消费者检测到通道关闭后立即退出,打印退出日志。​​扩展功能​​(可选): 添加一个监控协程,每秒打印当前最频繁生成的 3 个数字(需使用 RWMutex读取数据

package model

type Task struct {
	Id      string
	Content string
}

type Data struct {
	Count  int
	Record map[int]int
}



package main

import (
	"fmt"
	"goLearn/goroutine"
	"goLearn/model"
	"sync"
	"time"
)

var Ch = make(chan model.Task, 3)

var mutex sync.Mutex
var rwmutex sync.RWMutex

func main() {
	var wg sync.WaitGroup
	data := &model.Data{
		Count:  0,
		Record: make(map[int]int),
	}
	donetitle := make(chan struct{})
	for i := 0; i < 10; i++ {
		wg.Add(1)
		go goroutine.NewProducer(Ch, &wg, donetitle, i, &mutex, data, &rwmutex)
	}
	for i := 0; i < 3; i++ {
		wg.Add(1)
		go goroutine.NewConsumer(Ch, &wg, donetitle, i, data, &rwmutex)
	}
	time.Sleep(10 * time.Second)
	close(donetitle)
	wg.Wait()
	fmt.Println("任务完成")
}


package goroutine

import (
	"fmt"
	"goLearn/model"
	"math/rand"
	"strconv"
	"sync"
	"time"
)

func NewProducer(ch chan model.Task, wg *sync.WaitGroup, done chan struct{}, num int, mutex *sync.Mutex, data *model.Data, rwmutex *sync.RWMutex) {
	defer wg.Done()
	timech := time.Tick(1 * time.Second)
	rand.Seed(time.Now().UnixNano())
	for {
		select {
		case <-timech:
			fmt.Printf("生产者%d号上锁\n", num+1)
			randnum := rand.Intn(100)
			task := model.Task{
				Id:      strconv.Itoa(randnum),
				Content: "访问数据库",
			}
			mutex.Lock()
			data.Count += 1
			mutex.Unlock()

			rwmutex.Lock()
			data.Record[randnum] += 1
			rwmutex.Unlock()

			fmt.Printf("生产者%d号注入编号为%s,内容为:%s\n", num+1, task.Id, task.Content)
			ch <- task
			fmt.Printf("生产者%d号释放锁\n", num+1)

		case <-done:
			fmt.Printf("生产者%d号退出任务\n", num+1)
			return
		}

	}

}


package goroutine

import (
	"fmt"
	"goLearn/model"
	"sync"
)

func NewConsumer(ch chan model.Task, wg *sync.WaitGroup, done chan struct{}, num int, data *model.Data, rwmutex *sync.RWMutex) {
	defer wg.Done()
	for {
		select {
		case <-ch:
			rwmutex.RLock()
			fmt.Printf("消费者%d号处理消息统计消息总数为%d\n", num+1, data.Count)
			fmt.Printf("消费者%d号当前记录详情如下\n", num+1)
			for key, value := range data.Record {
				fmt.Printf("  键:%d%d\n", key, value)
			}
			rwmutex.RUnlock()
		case <-done:
			fmt.Printf("消费者%d号退出接收\n", num+1)
			return
		}
	}

}