Чтение текстового файла в массив строк (и запись)
Возможность считывания (и записи) текстового файла в массив строк и из него, я полагаю, является довольно распространенным требованием. Это также весьма полезно, когда начинается с языка, который устраняет необходимость изначально обращаться к базе данных. Существует ли один на Голанге?
например
func ReadLines(sFileName string, iMinLines int) ([]string, bool) {
а также
func WriteLines(saBuff[]string, sFilename string) (bool) {
Я предпочел бы использовать существующий, а не дублировать.
Ответы
Ответ 1
Начиная с версии Go1.1, есть API-интерфейс bufio.Scanner, который может легко читать строки из файла. Рассмотрим следующий пример сверху, переписанный с помощью сканера:
package main
import (
"bufio"
"fmt"
"log"
"os"
)
// readLines reads a whole file into memory
// and returns a slice of its lines.
func readLines(path string) ([]string, error) {
file, err := os.Open(path)
if err != nil {
return nil, err
}
defer file.Close()
var lines []string
scanner := bufio.NewScanner(file)
for scanner.Scan() {
lines = append(lines, scanner.Text())
}
return lines, scanner.Err()
}
// writeLines writes the lines to the given file.
func writeLines(lines []string, path string) error {
file, err := os.Create(path)
if err != nil {
return err
}
defer file.Close()
w := bufio.NewWriter(file)
for _, line := range lines {
fmt.Fprintln(w, line)
}
return w.Flush()
}
func main() {
lines, err := readLines("foo.in.txt")
if err != nil {
log.Fatalf("readLines: %s", err)
}
for i, line := range lines {
fmt.Println(i, line)
}
if err := writeLines(lines, "foo.out.txt"); err != nil {
log.Fatalf("writeLines: %s", err)
}
}
Ответ 2
Если файл не слишком велик, это можно сделать с помощью функций ioutil.ReadFile
и strings.Split
:
content, err := ioutil.ReadFile(filename)
if err != nil {
//Do something
}
lines := strings.Split(string(content), "\n")
Вы можете прочитать документацию на ioutil и strings.
Ответ 3
Не удается обновить первый ответ.
В любом случае, после выпуска Go1, есть некоторые нарушения, поэтому я обновил, как показано ниже:
package main
import (
"os"
"bufio"
"bytes"
"io"
"fmt"
"strings"
)
// Read a whole file into the memory and store it as array of lines
func readLines(path string) (lines []string, err error) {
var (
file *os.File
part []byte
prefix bool
)
if file, err = os.Open(path); err != nil {
return
}
defer file.Close()
reader := bufio.NewReader(file)
buffer := bytes.NewBuffer(make([]byte, 0))
for {
if part, prefix, err = reader.ReadLine(); err != nil {
break
}
buffer.Write(part)
if !prefix {
lines = append(lines, buffer.String())
buffer.Reset()
}
}
if err == io.EOF {
err = nil
}
return
}
func writeLines(lines []string, path string) (err error) {
var (
file *os.File
)
if file, err = os.Create(path); err != nil {
return
}
defer file.Close()
//writer := bufio.NewWriter(file)
for _,item := range lines {
//fmt.Println(item)
_, err := file.WriteString(strings.TrimSpace(item) + "\n");
//file.Write([]byte(item));
if err != nil {
//fmt.Println("debug")
fmt.Println(err)
break
}
}
/*content := strings.Join(lines, "\n")
_, err = writer.WriteString(content)*/
return
}
func main() {
lines, err := readLines("foo.txt")
if err != nil {
fmt.Println("Error: %s\n", err)
return
}
for _, line := range lines {
fmt.Println(line)
}
//array := []string{"7.0", "8.5", "9.1"}
err = writeLines(lines, "foo2.txt")
fmt.Println(err)
}
Ответ 4
Вы можете использовать os.File (который реализует io.Reader) с пакетом bufio. Тем не менее, эти пакеты создаются с учетом использования фиксированной памяти (независимо от размера файла) и довольно быстро.
К сожалению, это делает чтение всего файла в памяти немного сложнее. Вы можете использовать bytes.Buffer для присоединения к частям строки, если они превышают лимитный лимит. Во всяком случае, я рекомендую вам попробовать использовать линейный ридер непосредственно в вашем проекте (особенно, если вы не знаете, насколько большой текстовый файл!). Но если файл мал, для вас может быть достаточно следующего примера:
package main
import (
"os"
"bufio"
"bytes"
"fmt"
)
// Read a whole file into the memory and store it as array of lines
func readLines(path string) (lines []string, err os.Error) {
var (
file *os.File
part []byte
prefix bool
)
if file, err = os.Open(path); err != nil {
return
}
reader := bufio.NewReader(file)
buffer := bytes.NewBuffer(make([]byte, 1024))
for {
if part, prefix, err = reader.ReadLine(); err != nil {
break
}
buffer.Write(part)
if !prefix {
lines = append(lines, buffer.String())
buffer.Reset()
}
}
if err == os.EOF {
err = nil
}
return
}
func main() {
lines, err := readLines("foo.txt")
if err != nil {
fmt.Println("Error: %s\n", err)
return
}
for _, line := range lines {
fmt.Println(line)
}
}
Другой альтернативой может быть использование io.ioutil.ReadAll для немедленного чтения в полный файл и последующего нарезки по строке. Я не даю вам явного примера того, как писать строки обратно в файл, но в основном os.Create()
, за которым следует цикл, подобный этому в примере (см. main()
).
Ответ 5
func readToDisplayUsingFile1(f *os.File){
defer f.Close()
reader := bufio.NewReader(f)
contents, _ := ioutil.ReadAll(reader)
lines := strings.Split(string(contents), '\n')
}
или
func readToDisplayUsingFile1(f *os.File){
defer f.Close()
slice := make([]string,0)
reader := bufio.NewReader(f)
for{
str, err := reader.ReadString('\n')
if err == io.EOF{
break
}
slice = append(slice, str)
}