You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
849 B
37 lines
849 B
package util |
|
|
|
import ( |
|
"fmt" |
|
"io" |
|
) |
|
|
|
// Shortcut function to read n bytes from reader. The general idea peeked from here: https://github.com/dhowden/tag/blob/master/util.go |
|
func Read(rs io.Reader, n uint64) ([]byte, error) { |
|
read := make([]byte, n) |
|
_, err := rs.Read(read) |
|
if err != nil { |
|
return nil, fmt.Errorf("could not read from reader: %s", err) |
|
} |
|
|
|
return read, nil |
|
} |
|
|
|
// Shortcut function to read n bytes and convert them into string. |
|
// If encountered zero-byte - converts to string only previously read bytes |
|
func ReadToString(rs io.Reader, n int) (string, error) { |
|
read := make([]byte, n) |
|
_, err := rs.Read(read) |
|
if err != nil { |
|
return "", fmt.Errorf("could not read from reader: %s", err) |
|
} |
|
|
|
var readString string |
|
for _, b := range read { |
|
if b == 0 { |
|
break |
|
} |
|
readString += string(b) |
|
} |
|
|
|
return readString, nil |
|
}
|
|
|