csv文件操作（go语言）

go语言自带的有csv文件读取模块，看起来好像不错，今天玩玩，也算是系统学习go语言的一部分……^_^

一、写csv文件

函数：

func
NewWriter(w io.Writer) *Writer

func
(w *Writer) Flush()

func
(w *Writer) Write(record []string) (err os.Error)

func
(w *Writer) WriteAll(records [][]string) (err os.Error)

具体见官网：http://golang.org/pkg/csv/#NewWriter

其实写csv文件大可不必用这些函数，直接按照规则写入文件就行，不过这里有这些函数，还是写个demo，也方便我以后使用：

package main

import (
"csv"
"os"
"bytes"
"fmt"
)

func main(){
fileName := "2.csv"
buf := new(bytes.Buffer)
r2 := csv.NewWriter(buf)
for i:=0;i<10;i++ {
s := make([]string,3)
s[0] = "user id"
s[1] = "name"
s[2] = "depart"
r2.Write(s)
r2.Flush()
}
fmt.Println(buf)
fout,err := os.Create(fileName)
defer fout.Close()
if err != nil {
fmt.Println(fileName,err)
return
}
fout.WriteString(buf.String())
}

二、读csv文件

函数：

func
NewReader(r io.Reader) *Reader

func
(r *Reader) Read() (record []string, err os.Error)

func
(r *Reader) ReadAll() (records [][]string, err os.Error)

具体见官网：http://golang.org/pkg/csv/#Reader.Read

1、首先要解决的是读取文件的所有内容并存成字符串，这里用的是ioutil模块，具体代码如下：

package main

import (
"io/ioutil"
"fmt"
)

func main(){
b,err := ioutil.ReadFile("1.csv")
if err != nil {
panic(err.String())
}
//fmt.Println(b)
fmt.Printf("%s",string(b))
}

2、读取csv文件内容：

package main

import (
"fmt"
"csv"
"strings"
"io/ioutil"
)

func main(){
//fileName := "1.csv"
fmt.Printf("Input file name : ")
var fileName string
fmt.Scanf("%s",&fileName)
cntb,err := ioutil.ReadFile(fileName)
if err != nil {
panic(err.String())
}
r2 := csv.NewReader(strings.NewReader(string(cntb)))
ss,_ := r2.ReadAll()
//fmt.Println(ss)
sz := len(ss)
for i:=0;i<sz;i++{
fmt.Println(ss[i])
}
}

好，就这些了，希望对你有帮助