Golang 发送和接收数据公共类

KelvinQ · · 1832 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

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()
}

有疑问加站长微信联系（非本文作者）