This closes #2024, add TempDir field in the Options data type (#2024) (#2163)

Author: mengpromax

excelize.go

@@ -98,12 +98,16 @@ type charsetTranscoderFn func(charset string, input io.Reader) (rdr io.Reader, e
 //
 // CultureInfo specifies the country code for applying built-in language number
 // format code these effect by the system's local language settings.
+//
+// TmpDir specifies the temporary directory for creating temporary files, if the
+// value is empty, the system default temporary directory will be used.
 type Options struct {
 	MaxCalcIterations uint
 	Password          string
 	RawCellValue      bool
 	UnzipSizeLimit    int64
 	UnzipXMLSizeLimit int64
+	TmpDir            string
 	ShortDatePattern  string
 	LongDatePattern   string
 	LongTimePattern   string

lib.go

@@ -80,7 +80,7 @@ func (f *File) ReadZipReader(r *zip.Reader) (map[string][]byte, int, error) {
 // unzipToTemp unzip the zip entity to the system temporary directory and
 // returned the unzipped file path.
 func (f *File) unzipToTemp(zipFile *zip.File) (string, error) {
-	tmp, err := os.CreateTemp("", "excelize-")
+	tmp, err := os.CreateTemp(f.options.TmpDir, "excelize-")
 	if err != nil {
 		return "", err
 	}

rows.go

@@ -371,7 +371,7 @@ func (f *File) getFromStringItem(index int) string {
 		}()
 	}
 	f.sharedStringItem = [][]uint{}
-	f.sharedStringTemp, _ = os.CreateTemp("", "excelize-")
+	f.sharedStringTemp, _ = os.CreateTemp(f.options.TmpDir, "excelize-")
 	f.tempFiles.Store(defaultTempFileSST, f.sharedStringTemp.Name())
 	var (
 		inElement string

stream.go

@@ -123,6 +123,7 @@ func (f *File) NewStreamWriter(sheet string) (*StreamWriter, error) {
 		file:    f,
 		Sheet:   sheet,
 		SheetID: sheetID,
+		rawData: bufferedWriter{tmpDir: f.options.TmpDir},
 	}
 	var err error
 	sw.worksheet, err = f.workSheetReader(sheet)
@@ -737,8 +738,9 @@ func bulkAppendFields(w io.Writer, ws *xlsxWorksheet, from, to int) {
 // is written to the temp file with Sync, which may return an error.
 // Therefore, Sync should be periodically called and the error checked.
 type bufferedWriter struct {
-	tmp *os.File
-	buf bytes.Buffer
+	tmpDir string
+	tmp    *os.File
+	buf    bytes.Buffer
 }
 
 // Write to the in-memory buffer. The error is always nil.
@@ -775,7 +777,7 @@ func (bw *bufferedWriter) Sync() (err error) {
 		return nil
 	}
 	if bw.tmp == nil {
-		bw.tmp, err = os.CreateTemp("", "excelize-")
+		bw.tmp, err = os.CreateTemp(bw.tmpDir, "excelize-")
 		if err != nil {
 			// can not use local storage
 			return nil

stream_test.go

@@ -97,7 +97,7 @@ func TestStreamWriter(t *testing.T) {
 	assert.NoError(t, file.Close())
 
 	// Test close temporary file error
-	file = NewFile()
+	file = NewFile(Options{TmpDir: os.TempDir()})
 	streamWriter, err = file.NewStreamWriter("Sheet1")
 	assert.NoError(t, err)
 	for rowID := 10; rowID <= 25600; rowID++ {