From be12cc27f1d774154b17763c071e1dc6f91eab8c Mon Sep 17 00:00:00 2001
From: xuri <xuri.me@gmail.com>
Date: Mon, 10 May 2021 00:09:24 +0800
Subject: This closes #652, new SetColWidth API, support set column width in
stream writing mode, and export error message
---
stream.go | 40 +++++++++++++++++++++++++++++++++++++---
1 file changed, 37 insertions(+), 3 deletions(-)
(limited to 'stream.go')
diff --git a/stream.go b/stream.go
index 57bf4a2..f12b201 100644
--- a/stream.go
+++ b/stream.go
@@ -29,6 +29,8 @@ type StreamWriter struct {
File *File
Sheet string
SheetID int
+ sheetWritten bool
+ cols string
worksheet *xlsxWorksheet
rawData bufferedWriter
mergeCellsCount int
@@ -104,8 +106,7 @@ func (f *File) NewStreamWriter(sheet string) (*StreamWriter, error) {
f.streams[sheetXML] = sw
_, _ = sw.rawData.WriteString(XMLHeader + `<worksheet` + templateNamespaceIDMap)
- bulkAppendFields(&sw.rawData, sw.worksheet, 2, 6)
- _, _ = sw.rawData.WriteString(`<sheetData>`)
+ bulkAppendFields(&sw.rawData, sw.worksheet, 2, 5)
return sw, err
}
@@ -298,7 +299,13 @@ func (sw *StreamWriter) SetRow(axis string, values []interface{}) error {
if err != nil {
return err
}
-
+ if !sw.sheetWritten {
+ if len(sw.cols) > 0 {
+ sw.rawData.WriteString("<cols>" + sw.cols + "</cols>")
+ }
+ _, _ = sw.rawData.WriteString(`<sheetData>`)
+ sw.sheetWritten = true
+ }
fmt.Fprintf(&sw.rawData, `<row r="%d">`, row)
for i, val := range values {
axis, err := CoordinatesToCellName(col+i, row)
@@ -325,6 +332,33 @@ func (sw *StreamWriter) SetRow(axis string, values []interface{}) error {
return sw.rawData.Sync()
}
+// SetColWidth provides a function to set the width of a single column or
+// multiple columns for the the StreamWriter. Note that you must call
+// the 'SetColWidth' function before the 'SetRow' function. For example set
+// the width column B:C as 20:
+//
+// err := streamWriter.SetColWidth(2, 3, 20)
+//
+func (sw *StreamWriter) SetColWidth(min, max int, width float64) error {
+ if sw.sheetWritten {
+ return ErrStreamSetColWidth
+ }
+ if min > TotalColumns || max > TotalColumns {
+ return ErrColumnNumber
+ }
+ if min < 1 || max < 1 {
+ return ErrColumnNumber
+ }
+ if width > MaxColumnWidth {
+ return ErrColumnWidth
+ }
+ if min > max {
+ min, max = max, min
+ }
+ sw.cols += fmt.Sprintf(`<col min="%d" max="%d" width="%f" customWidth="1"/>`, min, max, width)
+ return nil
+}
+
// MergeCell provides a function to merge cells by a given coordinate area for
// the StreamWriter. Don't create a merged cell that overlaps with another
// existing merged cell.
--
cgit v1.2.1