2015-06-19 3 views
1

Я пытаюсь вывести первую строку файла и, таким образом, уменьшить строки файла по одному. Моя реализация для удаления первой линии заключается в следующийУдалить первую строку из текстового файла в Golang

type FS struct { 
    ... 
    File  *os.File 
} 

//File creation ok... 


func (fs *Fs) pop() []byte { 
    var buf []string 
    scanner := bufio.NewScanner(fs.File) 
    //Reading lines 
    for scanner.Scan() { 
     line := scanner.Text() 
     buf = append(buf, line) 
    } 
    //Writing from second line on the same file 
    for s := 1; s < len(buf); s++ { 
     fs.File.WriteString(fmt.Println(buf[s])) 
    } 
    //Commit changes 
    fs.File.Sync() 
    fs.File.Close() 

    return []byte(buf[0]) 
} 

я получаю возвращенные [] байты с ожидаемой строкой, но файл не изменяется. Что мне здесь не хватает?

+3

Во-первых, не игнорировать ошибки. Во-вторых, вы не пишете файл в том месте, которое, как вы думаете. – JimB

+0

Hi Jim, спасибо за это, я, возможно, пропустил 'fs.File.Seek (0,0)', но все равно я бы не получил никаких ошибок ... –

+3

Вы игнорируете ошибку, возвращаемую 'File.WriteString '. (также функция 'ScanLines' переносит символы новой строки, но если файл не изменяется, вы все равно не записываете его) – JimB

ответ

2

Например,

package main 

import (
    "bytes" 
    "fmt" 
    "io" 
    "os" 
) 

func popLine(f *os.File) ([]byte, error) { 
    fi, err := f.Stat() 
    if err != nil { 
     return nil, err 
    } 
    buf := bytes.NewBuffer(make([]byte, 0, fi.Size())) 

    _, err = f.Seek(0, os.SEEK_SET) 
    if err != nil { 
     return nil, err 
    } 
    _, err = io.Copy(buf, f) 
    if err != nil { 
     return nil, err 
    } 
    line, err := buf.ReadString('\n') 
    if err != nil && err != io.EOF { 
     return nil, err 
    } 

    _, err = f.Seek(0, os.SEEK_SET) 
    if err != nil { 
     return nil, err 
    } 
    nw, err := io.Copy(f, buf) 
    if err != nil { 
     return nil, err 
    } 
    err = f.Truncate(nw) 
    if err != nil { 
     return nil, err 
    } 
    err = f.Sync() 
    if err != nil { 
     return nil, err 
    } 

    _, err = f.Seek(0, os.SEEK_SET) 
    if err != nil { 
     return nil, err 
    } 
    return []byte(line), nil 
} 

func main() { 
    fname := `popline.txt` 
    f, err := os.OpenFile(fname, os.O_RDWR|os.O_CREATE, 0666) 
    if err != nil { 
     fmt.Println(err) 
     return 
    } 
    defer f.Close() 
    line, err := popLine(f) 
    if err != nil { 
     fmt.Println(err) 
     return 
    } 
    fmt.Println("pop:", string(line)) 
} 
 
$ cat popline.txt 
Lorem ipsum dolor sit amet, consectetur adipiscing elit, 
sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. 
Ut enim ad minim veniam, 
quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. 
Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. 
Excepteur sint occaecat cupidatat non proident, 
sunt in culpa qui officia deserunt mollit anim id est laborum. 

$ go run popline.go 
pop: Lorem ipsum dolor sit amet, consectetur adipiscing elit, 

$ cat popline.txt 
sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. 
Ut enim ad minim veniam, 
quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. 
Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. 
Excepteur sint occaecat cupidatat non proident, 
sunt in culpa qui officia deserunt mollit anim id est laborum. 

$ 
+0

Привет, Питер, приведенный выше код делает то, что я ожидал, спасибо! Читая и тестируя это, я считаю, что неправильно понял документацию os.Seek и os.Truncate. Я попробовал другую версию с усечением, но не смог заставить ее работать. Еще раз спасибо за то, что указали мне в правильном направлении :) –

Смежные вопросы