mirror of
https://github.com/golang/go.git
synced 2024-10-01 15:27:13 +00:00
WriteFile util function
R=rsc DELTA=41 (41 added, 0 deleted, 0 changed) OCL=31349 CL=31358
This commit is contained in:
parent
764b6ec1aa
commit
ff9e657fb3
@ -28,3 +28,19 @@ func ReadFile(filename string) ([]byte, os.Error) {
|
|||||||
defer f.Close();
|
defer f.Close();
|
||||||
return ReadAll(f);
|
return ReadAll(f);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// WriteFile writes data to a file named by filename.
|
||||||
|
// If the file does not exist, WriteFile creates it with permissions perm.
|
||||||
|
//
|
||||||
|
func WriteFile(filename string, data []byte, perm int) os.Error {
|
||||||
|
f, err := os.Open(filename, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, perm);
|
||||||
|
if err != nil {
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
n, err := f.Write(data);
|
||||||
|
if err == nil && n < len(data) {
|
||||||
|
err = ErrShortWrite;
|
||||||
|
}
|
||||||
|
f.Close();
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
@ -7,6 +7,7 @@ package io
|
|||||||
import (
|
import (
|
||||||
"io";
|
"io";
|
||||||
"os";
|
"os";
|
||||||
|
"strings";
|
||||||
"testing";
|
"testing";
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -35,3 +36,27 @@ func TestReadFile(t *testing.T) {
|
|||||||
|
|
||||||
checkSize(t, filename, uint64(len(contents)));
|
checkSize(t, filename, uint64(len(contents)));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestWriteFile(t *testing.T) {
|
||||||
|
filename := "_obj/rumpelstilzchen";
|
||||||
|
data :=
|
||||||
|
"Programming today is a race between software engineers striving to "
|
||||||
|
"build bigger and better idiot-proof programs, and the Universe trying "
|
||||||
|
"to produce bigger and better idiots. So far, the Universe is winning.";
|
||||||
|
|
||||||
|
if err := WriteFile(filename, strings.Bytes(data), 0644); err != nil {
|
||||||
|
t.Fatalf("WriteFile %s: %v", filename, err);
|
||||||
|
}
|
||||||
|
|
||||||
|
contents, err := ReadFile(filename);
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ReadFile %s: %v", filename, err);
|
||||||
|
}
|
||||||
|
|
||||||
|
if string(contents) != data {
|
||||||
|
t.Fatalf("contents = %q\nexpected = %q", string(contents), data);
|
||||||
|
}
|
||||||
|
|
||||||
|
// cleanup
|
||||||
|
os.Remove(filename); // ignore error
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user