Golang 发送和接收数据公共类

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 CreateRequestWithConn(conn net.Conn) *QRequest {
	request := new(QRequest)
	//GC析构前调用
	runtime.SetFinalizer(request,func(q *QRequest) {
		q.Close()
	})
	request.conn = conn
	return request
}

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

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

}

//接收内容
func (q *QRequest) ReceiveData(timeout time.Duration) []byte {
	if !q.CheckReq() {
		return nil
	}
	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()
}

func (q *QRequest) CheckReq() bool {
	if q == nil || q.conn == nil {
		return false
	}
	return true
}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!