Golang 发送和接收数据公共类

KelvinQ · · 1457 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

package RequestCenter

import (
	"bytes"
	"io"
	"net"
	"runtime"
	"strings"
	"time"
)

//请求结构体
type QRequest struct {
	conn net.Conn
}

//构造函数
func CreateRequest(typename string, host string) *QRequest {
	request := new(QRequest)
	//GC析构前调用
	runtime.SetFinalizer(request, func(q *QRequest) {
		q.Close()
	})
	conn, err := net.Dial(typename, host)
	if err == nil {
		request.conn = conn

		return request
	}
	return nil
}

//析构函数
func (q *QRequest) Close() {
	q.conn.SetDeadline(time.Now())
	q.conn.Close()
}

//发送内容
func (q *QRequest) SendData(heads []string, body string) bool {
	headsStr := strings.Join(heads, "\r\n") + body
	count, err := q.conn.Write([]byte(headsStr))
	if err != nil || count == 0 {
		return false
	}
	return true
}

//接收内容
func (q *QRequest) ReceiveData(timeout time.Duration) []byte {
	if timeout > 0 {
		q.conn.SetReadDeadline(time.Now().Add(time.Millisecond * timeout))
	}
	var buf bytes.Buffer
	buffer := make([]byte, 8192)
	for {
		sizenew, err := q.conn.Read(buffer)
		buf.Write(buffer[:sizenew])
		if err == io.EOF || sizenew < 8192 {
			break
		}
	}
	return buf.Bytes()
}

 


有疑问加站长微信联系(非本文作者)

本文来自:开源中国博客

感谢作者:KelvinQ

查看原文:Golang 发送和接收数据公共类

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1457 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传