From bb833561aa74f02970aee13cdc75973b29716491 Mon Sep 17 00:00:00 2001 From: leshe4ka46 Date: Mon, 27 Oct 2025 20:36:28 +0300 Subject: # This is a combination of 2 commits. # This is the 1st commit message: unmarshal all formats, merge them in the single table, users are truly unique # This is the commit message #2: i --- pkg/csvWriter/csv.go | 62 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 62 insertions(+) create mode 100644 pkg/csvWriter/csv.go (limited to 'pkg/csvWriter') diff --git a/pkg/csvWriter/csv.go b/pkg/csvWriter/csv.go new file mode 100644 index 0000000..8709644 --- /dev/null +++ b/pkg/csvWriter/csv.go @@ -0,0 +1,62 @@ +package csvwriter + +import ( + "bytes" + "os" +) + +type CsvWriter struct { + f *os.File + buf bytes.Buffer +} + +func NewCsvWriter(path string) (*CsvWriter, error) { + f, err := os.Create(path) + if err != nil { + return nil, err + } + + w := &CsvWriter{ + f: f, + } + return w, nil +} + +func (c *CsvWriter) Close() error { + c.Sync() + return c.f.Close() +} + +func (c *CsvWriter) Write(strs []string) error { + + for i, str := range strs { + if i != 0 { + _, err := c.buf.Write([]byte(",")) + if err != nil { + return err + } + } + _, err := c.buf.Write([]byte(str)) + if err != nil { + return err + } + } + c.buf.WriteByte('\n') + + if c.buf.Len() > 1*1024*1024 { + err := c.Sync() + if err != nil { + return err + } + } + return nil +} + +func (c *CsvWriter) Sync() error { + _, err := c.f.Write(c.buf.Bytes()) + if err != nil { + return err + } + c.buf.Reset() + return nil +} -- cgit v1.2.3