fumiama/go-simple-protobuf
1
0
Fork 0
mirror of https://github.com/fumiama/go-simple-protobuf.git synced 2026-06-27 07:30:34 +08:00
Code Activity

feat: add iterator

Browse Source
This commit is contained in:
源文雨
2024-02-26 03:44:30 +09:00
parent 93032ad80a
commit 6c9102f75f
3 changed files with 115 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
spb.go
View File

@@ -2,7 +2,7 @@ package spb
import "io"
func ReadNum(r io.Reader) (n uint32, cnt int, err error) {
func ReadNum(r io.Reader) (n uint32, cnt uint32, err error) {
var buf [1]byte
for cnt < 5 {
_, err = r.Read(buf[:])
@@ -43,7 +43,7 @@ type SimplePB struct {
}
func NewSimplePB(r io.Reader) (s SimplePB, err error) {
cnt := 0
cnt := uint32(0)
s.StructLen, cnt, err = ReadNum(r)
if err != nil {
s.RealLen = uint32(cnt)
@@ -55,7 +55,7 @@ func NewSimplePB(r io.Reader) (s SimplePB, err error) {
return
}
var offset, datalen uint32
n := 0
n := uint32(0)
for i := uint32(0); i < s.StructLen; i += offset {
offset, n, err = ReadNum(r)
cnt += n
@@ -75,8 +75,9 @@ func NewSimplePB(r io.Reader) (s SimplePB, err error) {
break
}
t := make([]byte, datalen)
n, err = r.Read(t)
cnt += n
x := 0
x, err = r.Read(t)
cnt += uint32(x)
if err != nil {
break
}