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

全部评论

相关推荐

头像
点赞 评论 收藏
转发
点赞 评论 收藏
转发
头像
04-29 10:53
已编辑
东北大学 自动化类
点赞 评论 收藏
转发
点赞 收藏 评论
分享
牛客网
牛客企业服务