现在的位置: 首页 > 综合 > 正文

golang 解析cvs文件

2017年12月21日 ⁄ 综合 ⁄ 共 1486字 ⁄ 字号 评论关闭

//本程序是为了根据cvs的内容,自动生实时效果的报文,第七列为最后值

//用法: 程序名+文件名
package main
import (
    "encoding/csv"
    "flag"
    "fmt"
    "io"
    "log"
    "os"
    "strings"
)
func main() {
    //参数
    flag.Parse()
    fmt.Println(flag.NArg())
    if flag.NArg() != 1 {
        log.Printf("usage: command + filename")
        return
    }
    //只取第1个参数做为文件名
    file, err := os.Open(flag.Arg(0))
    if err != nil {
        fmt.Println("Error:", err)
        return
    }
    defer file.Close()
    reader := csv.NewReader(file)
    reader.Comment = '#' //可以设置读入文件中的注释符
    reader.Comma = ',' //默认是逗号,也可以自己设置
    //还可以设置以下信息
    //FieldsPerRecord  int  // Number of expected fields per record
    //LazyQuotes       bool // Allow lazy quotes
    //TrailingComma    bool // Allow trailing comma
    //TrimLeadingSpace bool // Trim leading space
    //line             int
    //column           int
    fout, err := os.OpenFile("out.txt", os.O_RDWR|os.O_APPEND|os.O_CREATE, 0666)
    if err != nil {
        fmt.Println("Error:", err)
        return
    }
    defer fout.Close()
    content := ""
    k := 0 //第一行是字段名,不需要
    for {
        record, err := reader.Read()
        if err == io.EOF {
            break
        } else if err != nil {
            fmt.Println("Error:", err)
            return
        }
        if k > 0 { //record是[]strings, 怎样直接获得域值
            for _, v := range record {
                tmp := strings.Split(v, "|")
                //fmt.Print("<" + tmp[1] + ">" + tmp[7] + "</" + tmp[1] + ">")
                content = content + "<" + tmp[1] + ">" + tmp[7] + "</" + tmp[1] + ">"
            }
        }
        k = k + 1
    }
    fmt.Printf("\n")
    fout.WriteString(content + "\n")
    fmt.Printf("\n")
}

抱歉!评论已关闭.