Introduce RegisterReaderHandlerWithEncoderInfo

AUTHORS

@@ -52,6 +52,7 @@ Jeffrey Charles <jeffreycharles at gmail.com>
 Jerome Meyer <jxmeyer at gmail.com>
 Jiajia Zhong <zhong2plus at gmail.com>
 Jian Zhen <zhenjl at gmail.com>
+Jille Timmermans <jille at quis.cx>
 Joshua Prunier <joshua.prunier at gmail.com>
 Julien Lefevre <julien.lefevr at gmail.com>
 Julien Schmidt <go-sql-driver at julienschmidt.com>

infile.go

@@ -14,15 +14,27 @@ import (
 	"os"
 	"strings"
 	"sync"
+	"time"
 )
 
 var (
 	fileRegister       map[string]bool
 	fileRegisterLock   sync.RWMutex
-	readerRegister     map[string]func() io.Reader
+	readerRegister     map[string]func(EncoderInfo) io.Reader
 	readerRegisterLock sync.RWMutex
 )
 
+// EncoderInfo contains the settings needed to encode a TSV file for LOAD DATA INFILE.
+type EncoderInfo struct {
+	Location *time.Location
+}
+
+func (c *Config) encoderInfo() EncoderInfo {
+	return EncoderInfo{
+		Location: c.Loc,
+	}
+}
+
 // RegisterLocalFile adds the given file to the file allowlist,
 // so that it can be used by "LOAD DATA LOCAL INFILE <filepath>".
 // Alternatively you can allow the use of all local files with
@@ -66,10 +78,18 @@ func DeregisterLocalFile(filePath string) {
 //	if err != nil {
 //	...
 func RegisterReaderHandler(name string, handler func() io.Reader) {
+	RegisterReaderHandlerWithEncoderInfo(name, func(EncoderInfo) io.Reader {
+		return handler()
+	})
+}
+
+// RegisterReaderHandlerWithEncoderInfo is like RegisterReaderHandler but the
+// callback receives the information needed to correctly encode a TSV file.
+func RegisterReaderHandlerWithEncoderInfo(name string, handler func(EncoderInfo) io.Reader) {
 	readerRegisterLock.Lock()
 	// lazy map init
 	if readerRegister == nil {
-		readerRegister = make(map[string]func() io.Reader)
+		readerRegister = make(map[string]func(EncoderInfo) io.Reader)
 	}
 
 	readerRegister[name] = handler
@@ -110,7 +130,7 @@ func (mc *mysqlConn) handleInFileRequest(name string) (err error) {
 		readerRegisterLock.RUnlock()
 
 		if inMap {
-			rdr = handler()
+			rdr = handler(mc.cfg.encoderInfo())
 			if rdr != nil {
 				if cl, ok := rdr.(io.Closer); ok {
 					defer deferredClose(&err, cl)