diff options
author | Rob Pike <r@golang.org> | 2009-12-02 22:02:14 -0800 |
---|---|---|
committer | Rob Pike <r@golang.org> | 2009-12-02 22:02:14 -0800 |
commit | 9599e35e75f3bf5223edd9b0a0854536c23cdef2 (patch) | |
tree | c8ce48aeddfe892be3754b0aa74217b4eb6d1e57 /src/pkg/io/ioutil/ioutil.go | |
parent | 0247d93d819700b0bfc6814dc7b1c2afec4648a0 (diff) | |
download | golang-9599e35e75f3bf5223edd9b0a0854536c23cdef2.tar.gz |
move ReadFile, WriteFile, and ReadDir into a separate io/ioutil package.
this breaks the dependency of package io on package bytes.
R=rsc
CC=golang-dev
http://codereview.appspot.com/163085
Diffstat (limited to 'src/pkg/io/ioutil/ioutil.go')
-rw-r--r-- | src/pkg/io/ioutil/ioutil.go | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/src/pkg/io/ioutil/ioutil.go b/src/pkg/io/ioutil/ioutil.go new file mode 100644 index 000000000..a38e48811 --- /dev/null +++ b/src/pkg/io/ioutil/ioutil.go @@ -0,0 +1,90 @@ +// Copyright 2009 The Go Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +// Utility functions. + +package ioutil + +import ( + "bytes"; + "io"; + "os"; + "sort"; +) + +// ReadAll reads from r until an error or EOF and returns the data it read. +func ReadAll(r io.Reader) ([]byte, os.Error) { + var buf bytes.Buffer; + _, err := io.Copy(&buf, r); + return buf.Bytes(), err; +} + +// ReadFile reads the file named by filename and returns the contents. +func ReadFile(filename string) ([]byte, os.Error) { + f, err := os.Open(filename, os.O_RDONLY, 0); + if err != nil { + return nil, err + } + defer f.Close(); + // It's a good but not certain bet that Stat will tell us exactly how much to + // read, so let's try it but be prepared for the answer to be wrong. + dir, err := f.Stat(); + var n uint64; + if err != nil && dir.Size < 2e9 { // Don't preallocate a huge buffer, just in case. + n = dir.Size + } + if n == 0 { + n = 1024 // No idea what's right, but zero isn't. + } + // Pre-allocate the correct size of buffer, then set its size to zero. The + // Buffer will read into the allocated space cheaply. If the size was wrong, + // we'll either waste some space off the end or reallocate as needed, but + // in the overwhelmingly common case we'll get it just right. + buf := bytes.NewBuffer(make([]byte, n)[0:0]); + _, err = io.Copy(buf, f); + return buf.Bytes(), err; +} + +// WriteFile writes data to a file named by filename. +// If the file does not exist, WriteFile creates it with permissions perm; +// otherwise WriteFile truncates it before writing. +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); + f.Close(); + if err == nil && n < len(data) { + err = io.ErrShortWrite + } + return err; +} + +// A dirList implements sort.Interface. +type dirList []*os.Dir + +func (d dirList) Len() int { return len(d) } +func (d dirList) Less(i, j int) bool { return d[i].Name < d[j].Name } +func (d dirList) Swap(i, j int) { d[i], d[j] = d[j], d[i] } + +// ReadDir reads the directory named by dirname and returns +// a list of sorted directory entries. +func ReadDir(dirname string) ([]*os.Dir, os.Error) { + f, err := os.Open(dirname, os.O_RDONLY, 0); + if err != nil { + return nil, err + } + list, err := f.Readdir(-1); + f.Close(); + if err != nil { + return nil, err + } + dirs := make(dirList, len(list)); + for i := range list { + dirs[i] = &list[i] + } + sort.Sort(dirs); + return dirs, nil; +} |