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)

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值