tcp的read方法阻塞 about magpie

[haifenghuang]

tcp的代码在net.go文件中，其中的Read方法如下：

func (t *TcpConnObject) Read(line string, args ...Object) Object {
	if len(args) != 0 {
		panic(NewError(line, ARGUMENTERROR, "0", len(args)))
	}

	bytes, err := ioutil.ReadAll(t.Conn)
	if err != nil {
		return NewNil(err.Error())
	}

	return NewString(string(bytes))
}

这里使用的是ioutil.ReadAll这个方法，它是阻塞的。

你可能是希望如下的代码吧：

func (t *TcpConnObject) Read(line string, args ...Object) Object {
	if len(args) != 1 {
		panic(NewError(line, ARGUMENTERROR, "1", len(args)))
	}

	length, ok := args[0].(*Integer) //The first parameter is the length you want to read
	if !ok {
		panic(NewError(line, PARAMTYPEERROR, "first", "read", "*Integer", args[0].Type()))
	}

	data := make([]byte, length.Int64);

	readTotal := 0
	readBytes := 0
	var err error
	for {
		if readTotal == len(data) {
			break
		}
		if readBytes, err = t.Conn.Read(data[readTotal:]); err != nil {
			break
		}
		readTotal += readBytes
	}

	if err != nil {
		return NewNil(err.Error())
	}

	return NewString(string(data))
}