1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

os: change f.write and f.writeln to return options too

This commit is contained in:
Delyan Angelov 2020-09-12 08:31:15 +03:00
parent 332f3a924c
commit 60ecb7e4b6
2 changed files with 24 additions and 13 deletions

View File

@ -925,7 +925,7 @@ A defer statement defers the execution of a block of statements until the surrou
```v ```v
fn read_log() { fn read_log() {
f := os.open('log.txt') f := os.open('log.txt') or { panic(err) }
defer { f.close() } defer { f.close() }
... ...
if !ok { if !ok {

View File

@ -3,9 +3,9 @@ module os
import strings import strings
pub struct File { pub struct File {
cfile voidptr // Using void* instead of FILE* cfile voidptr // Using void* instead of FILE*
pub: pub:
fd int fd int
pub mut: pub mut:
is_opened bool is_opened bool
} }
@ -22,9 +22,9 @@ pub fn (f File) is_opened() bool {
} }
// **************************** Write ops *************************** // **************************** Write ops ***************************
pub fn (mut f File) write(s string) { pub fn (mut f File) write(s string) ?int {
if !f.is_opened { if !f.is_opened {
return return error('file is not opened')
} }
/* /*
$if linux { $if linux {
@ -34,12 +34,16 @@ pub fn (mut f File) write(s string) {
} }
} }
*/ */
C.fwrite(s.str, s.len, 1, f.cfile) written := C.fwrite(s.str, s.len, 1, f.cfile)
if written == 0 && s.len != 0 {
return error('0 bytes written')
}
return written
} }
pub fn (mut f File) writeln(s string) { pub fn (mut f File) writeln(s string) ?int {
if !f.is_opened { if !f.is_opened {
return return error('file is not opened')
} }
/* /*
$if linux { $if linux {
@ -51,8 +55,15 @@ pub fn (mut f File) writeln(s string) {
} }
*/ */
// TODO perf // TODO perf
C.fwrite(s.str, s.len, 1, f.cfile) written := C.fwrite(s.str, s.len, 1, f.cfile)
C.fputs('\n', f.cfile) if written == 0 && s.len != 0 {
return error('0 bytes written')
}
x := C.fputs('\n', f.cfile)
if x < 0 {
return error('could not add newline')
}
return (written + 1)
} }
pub fn (mut f File) write_bytes(data voidptr, size int) int { pub fn (mut f File) write_bytes(data voidptr, size int) int {
@ -74,8 +85,8 @@ pub fn (f &File) read_bytes(size int) []byte {
// read_bytes_at reads an amount of bytes at the given position in the file // read_bytes_at reads an amount of bytes at the given position in the file
pub fn (f &File) read_bytes_at(size, pos int) []byte { pub fn (f &File) read_bytes_at(size, pos int) []byte {
//mut arr := [`0`].repeat(size) // mut arr := [`0`].repeat(size)
mut arr := []byte{ len:size } mut arr := []byte{len: size}
C.fseek(f.cfile, pos, C.SEEK_SET) C.fseek(f.cfile, pos, C.SEEK_SET)
nreadbytes := C.fread(arr.data, 1, size, f.cfile) nreadbytes := C.fread(arr.data, 1, size, f.cfile)
C.fseek(f.cfile, 0, C.SEEK_SET) C.fseek(f.cfile, 0, C.SEEK_SET)
@ -143,7 +154,7 @@ pub fn (mut f File) get_line() ?string {
mut blen := vstrlen(bufbp) mut blen := vstrlen(bufbp)
res.write_bytes(bufbp, blen) res.write_bytes(bufbp, blen)
unsafe { unsafe {
if blen == 0 || bufbp[blen-1] == `\n` || bufbp[blen-1] == `\r` { if blen == 0 || bufbp[blen - 1] == `\n` || bufbp[blen - 1] == `\r` {
break break
} }
} }