Golang gorutine间通信

package main

import (
	"encoding/csv"
	"fmt"
	"os"
	"strconv"
	"sync"
	"time"
)

type Server struct {
   
}

func (server *Server) Connect() chan []string {
   
	session := make(chan []string, 128)
	go func(c chan []string) {
   
		file, _ := os.OpenFile("./test.csv", os.O_CREATE|os.O_TRUNC, 0666)
		defer file.Close()
		writer := csv.NewWriter(file)
		defer file.Close()
		for {
   
			strs := <-c
			if strs[0] == "CLOSE" {
   
				writer.Write(strs)
				writer.Flush()
				fmt.Println("结束通信")
				break
			}
			writer.Write(strs)
			writer.Flush()
		}

	}(session)

	fmt.Println("建立连接成功!")
	return session
}
func NewServer() *Server {
   
	return &Server{
   }
}

type Client struct {
   
	conn chan []string
}

func NewClient(server *Server) *Client {
   
	c := server.Connect()
	return &Client{
   c}
}

func (client *Client) Send(strs []string) {
   
	client.conn <- strs
}

var wp sync.WaitGroup

func main() {
   

	server := NewServer()
	client := NewClient(server)
	for i := 0; i < 10000; i++ {
   
		wp.Add(1)
		go func() {
   
			defer wp.Done()
			client.Send([]string{
   "hello", strconv.Itoa(i)})
		}()
	}

	//time.Sleep(time.Duration(1) * time.Second)
	wp.Wait()
	go client.Send([]string{
   "CLOSE"})
	time.Sleep(time.Duration(1) * time.Second)

}

GO语言快速入门 文章被收录于专栏

帮助GO初学者快速入门Golang

全部评论

相关推荐

仁者伍敌:服务员还要脱颖而出,这是五星级酒店吗
点赞 评论 收藏
分享
Southyeung:我说一下我的看法(有冒犯实属抱歉):(1)简历不太美观,给我一种看都不想看的感觉,感觉字体还是排版问题;(2)numpy就一个基础包,机器学习算法是什么鬼?我感觉你把svm那些写上去都要好一点。(2)课程不要写,没人看,换成获奖经历;(3)项目太少了,至少2-3个,是在不行把网上学习的也写上去。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务