Stowage/go
2
0
Fork 0
mirror of https://github.com/golang/go.git synced 2025-12-08 06:10:04 +00:00
Code Issues Projects Releases Packages Wiki Activity

clean up the decode loop and fix a couple of bad prints

Browse source 
R=rsc
DELTA=15  (8 added, 2 deleted, 5 changed)
OCL=31738
CL=31738
This commit is contained in:
Rob Pike 2009-07-16 13:05:46 -07:00
parent 1902283053
commit 04ae91cf71
2 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/pkg/gob/decoder.go
View file

@ -19,6 +19,7 @@ type Decoder struct {
seen map[TypeId] *wireType; // which types we've already seen described
state *decodeState; // reads data from in-memory buffer
countState *decodeState; // reads counts from wire
buf []byte;
oneByte []byte;
}
@ -63,10 +64,15 @@ func (dec *Decoder) Decode(e interface{}) os.Error {
return err;
}
// Allocate the buffer.
if nbytes > uint64(len(dec.buf)) {
dec.buf = make([]byte, nbytes + 1000);
}
dec.state.b = bytes.NewBuffer(dec.buf[0:nbytes]);
// Read the data
buf := make([]byte, nbytes); // TODO(r): avoid repeated allocation
var n int;
n, err = dec.r.Read(buf);
n, err = dec.r.Read(dec.buf[0:nbytes]);
if err != nil {
return err;
}
@ -74,13 +80,13 @@ func (dec *Decoder) Decode(e interface{}) os.Error {
return os.ErrorString("gob decode: short read");
}
dec.state.b = bytes.NewBuffer(buf); // TODO(r): avoid repeated allocation
// Receive a type id.
id := TypeId(decodeInt(dec.state));
if dec.state.err != nil {
return dec.state.err
}
// Is it a type?
if id < 0 { // 0 is the error state, handled above
// If the id is negative, we have a type.
dec.recvType(-id);
@ -90,7 +96,7 @@ func (dec *Decoder) Decode(e interface{}) os.Error {
continue;
}
// we have a value
// No, it's a value.
info := getTypeInfo(rt);
// Check type compatibility.

6
src/pkg/gob/encoder_test.go
View file

@ -171,7 +171,7 @@ func TestEncoderDecoder(t *testing.T) {
newEt1 := new(ET1);
dec.Decode(newEt1);
if dec.state.err != nil {
t.Fatalf("error decoding ET1:", dec.state.err);
t.Fatal("error decoding ET1:", dec.state.err);
}
if !reflect.DeepEqual(et1, newEt1) {
@ -185,7 +185,7 @@ func TestEncoderDecoder(t *testing.T) {
newEt1 = new(ET1);
dec.Decode(newEt1);
if dec.state.err != nil {
t.Fatalf("round 2: error decoding ET1:", dec.state.err);
t.Fatal("round 2: error decoding ET1:", dec.state.err);
}
if !reflect.DeepEqual(et1, newEt1) {
t.Fatalf("round 2: invalid data for et1: expected %+v; got %+v\n", *et1, *newEt1);
@ -202,7 +202,7 @@ func TestEncoderDecoder(t *testing.T) {
newEt2 := new(ET2);
dec.Decode(newEt2);
if dec.state.err == nil {
t.Fatalf("round 3: expected `bad type' error decoding ET2");
t.Fatal("round 3: expected `bad type' error decoding ET2");
}
}