-
Notifications
You must be signed in to change notification settings - Fork 3.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
28181: importccl: Preserve '\r\n' during CSV import r=benesch a=neeral See #25344. Co-authored-by: neeral <neeral@users.noreply.github.com>
- Loading branch information
Showing
12 changed files
with
1,279 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,3 +1,4 @@ | ||
| (*bufio.Writer).Flush | ||
| (*database/sql.DB).Close | ||
| (*database/sql.Rows).Close | ||
| (*database/sql.Stmt).Close | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,140 @@ | ||
| // Copyright 2015 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. | ||
|
|
||
| package csv_test | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/csv" | ||
| "fmt" | ||
| "io" | ||
| "os" | ||
| "strings" | ||
|
|
||
| "github.com/cockroachdb/cockroach/pkg/util/log" | ||
| ) | ||
|
|
||
| func ExampleReader() { | ||
| ctx := context.Background() | ||
| in := `first_name,last_name,username | ||
| "Rob","Pike",rob | ||
| Ken,Thompson,ken | ||
| "Robert","Griesemer","gri" | ||
| ` | ||
| r := csv.NewReader(strings.NewReader(in)) | ||
|
|
||
| for { | ||
| record, err := r.Read() | ||
| if err == io.EOF { | ||
| break | ||
| } | ||
| if err != nil { | ||
| log.Fatal(ctx, err) | ||
| } | ||
|
|
||
| fmt.Println(record) | ||
| } | ||
| // Output: | ||
| // [first_name last_name username] | ||
| // [Rob Pike rob] | ||
| // [Ken Thompson ken] | ||
| // [Robert Griesemer gri] | ||
| } | ||
|
|
||
| // This example shows how csv.Reader can be configured to handle other | ||
| // types of CSV files. | ||
| func ExampleReader_options() { | ||
| ctx := context.Background() | ||
| in := `first_name;last_name;username | ||
| "Rob";"Pike";rob | ||
| # lines beginning with a # character are ignored | ||
| Ken;Thompson;ken | ||
| "Robert";"Griesemer";"gri" | ||
| ` | ||
| r := csv.NewReader(strings.NewReader(in)) | ||
| r.Comma = ';' | ||
| r.Comment = '#' | ||
|
|
||
| records, err := r.ReadAll() | ||
| if err != nil { | ||
| log.Fatal(ctx, err) | ||
| } | ||
|
|
||
| fmt.Print(records) | ||
| // Output: | ||
| // [[first_name last_name username] [Rob Pike rob] [Ken Thompson ken] [Robert Griesemer gri]] | ||
| } | ||
|
|
||
| func ExampleReader_ReadAll() { | ||
| ctx := context.Background() | ||
| in := `first_name,last_name,username | ||
| "Rob","Pike",rob | ||
| Ken,Thompson,ken | ||
| "Robert","Griesemer","gri" | ||
| ` | ||
| r := csv.NewReader(strings.NewReader(in)) | ||
|
|
||
| records, err := r.ReadAll() | ||
| if err != nil { | ||
| log.Fatal(ctx, err) | ||
| } | ||
|
|
||
| fmt.Print(records) | ||
| // Output: | ||
| // [[first_name last_name username] [Rob Pike rob] [Ken Thompson ken] [Robert Griesemer gri]] | ||
| } | ||
|
|
||
| func ExampleWriter() { | ||
| ctx := context.Background() | ||
| records := [][]string{ | ||
| {"first_name", "last_name", "username"}, | ||
| {"Rob", "Pike", "rob"}, | ||
| {"Ken", "Thompson", "ken"}, | ||
| {"Robert", "Griesemer", "gri"}, | ||
| } | ||
|
|
||
| w := csv.NewWriter(os.Stdout) | ||
|
|
||
| for _, record := range records { | ||
| if err := w.Write(record); err != nil { | ||
| log.Fatalf(ctx, "error writing record to csv: %v\n", err) | ||
| } | ||
| } | ||
|
|
||
| // Write any buffered data to the underlying writer (standard output). | ||
| w.Flush() | ||
|
|
||
| if err := w.Error(); err != nil { | ||
| log.Fatal(ctx, err) | ||
| } | ||
| // Output: | ||
| // first_name,last_name,username | ||
| // Rob,Pike,rob | ||
| // Ken,Thompson,ken | ||
| // Robert,Griesemer,gri | ||
| } | ||
|
|
||
| func ExampleWriter_WriteAll() { | ||
| ctx := context.Background() | ||
| records := [][]string{ | ||
| {"first_name", "last_name", "username"}, | ||
| {"Rob", "Pike", "rob"}, | ||
| {"Ken", "Thompson", "ken"}, | ||
| {"Robert", "Griesemer", "gri"}, | ||
| } | ||
|
|
||
| w := csv.NewWriter(os.Stdout) | ||
| if err := w.WriteAll(records); err != nil { // calls Flush internally | ||
| log.Fatalf(ctx, "error writing csv: %v\n", err) | ||
| } | ||
|
|
||
| if err := w.Error(); err != nil { | ||
| log.Fatalf(ctx, "error writing csv: %v\n", err) | ||
| } | ||
| // Output: | ||
| // first_name,last_name,username | ||
| // Rob,Pike,rob | ||
| // Ken,Thompson,ken | ||
| // Robert,Griesemer,gri | ||
| } |
Oops, something went wrong.