fix missing return, make code flatter

Author: sluongng

server/util/vtprotocodec/vtprotocodec.go

@@ -24,35 +24,38 @@ func (vtprotoCodecV2) Name() string {
 }
 
 func (c *vtprotoCodecV2) Marshal(v any) (mem.BufferSlice, error) {
-	if m, ok := v.(proto.VtprotoMessage); ok {
-		size := m.SizeVT()
-		if mem.IsBelowBufferPoolingThreshold(size) {
-			buf := make([]byte, 0, size)
-			if _, err := m.MarshalToSizedBufferVT(buf[:size]); err != nil {
-				return nil, err
-			}
-		}
+	m, ok := v.(proto.VtprotoMessage)
+	if !ok {
+		return c.fallback.Marshal(v)
+	}
 
-		pool := mem.DefaultBufferPool()
-		buf := pool.Get(size)
-		if _, err := m.MarshalToSizedBufferVT((*buf)[:size]); err != nil {
-			pool.Put(buf)
+	size := m.SizeVT()
+	if mem.IsBelowBufferPoolingThreshold(size) {
+		buf := make([]byte, 0, size)
+		if _, err := m.MarshalToSizedBufferVT(buf[:size]); err != nil {
 			return nil, err
 		}
+		return mem.BufferSlice{mem.SliceBuffer(buf)}, nil
+	}
 
-		return mem.BufferSlice{mem.NewBuffer(buf, pool)}, nil
+	pool := mem.DefaultBufferPool()
+	buf := pool.Get(size)
+	if _, err := m.MarshalToSizedBufferVT((*buf)[:size]); err != nil {
+		pool.Put(buf)
+		return nil, err
 	}
-	return c.fallback.Marshal(v)
+	return mem.BufferSlice{mem.NewBuffer(buf, pool)}, nil
 }
 
 func (c *vtprotoCodecV2) Unmarshal(data mem.BufferSlice, v any) error {
-	if m, ok := v.(proto.VtprotoMessage); ok {
-		buf := data.MaterializeToBuffer(mem.DefaultBufferPool())
-		defer buf.Free()
-
-		return m.UnmarshalVT(buf.ReadOnlyData())
+	m, ok := v.(proto.VtprotoMessage)
+	if !ok {
+		return c.fallback.Unmarshal(data, v)
 	}
-	return c.fallback.Unmarshal(data, v)
+
+	buf := data.MaterializeToBuffer(mem.DefaultBufferPool())
+	defer buf.Free()
+	return m.UnmarshalVT(buf.ReadOnlyData())
 }
 
 // RegisterCodec registers the vtprotoCodec to encode/decode proto messages with