root/golib
1
0
Fork 0
mirror of https://github.com/therootcompany/golib.git synced 2026-11-04 06:55:00 +00:00
Code Issues Projects Releases Wiki Activity

feat(gsheet2csv): add context and custom client options

Browse Source
This commit is contained in:
AJ ONeal 2026-09-08 22:15:42 -06:00
parent 6d60dae5e6
commit d55380ce48
No known key found for this signature in database
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
io/transform/gsheet2csv/gsheet2csv.go
View File

@ -51,8 +51,13 @@ type Reader struct {
// NewReaderFromContext is NewReaderFrom with request cancellation. // NewReaderFromContext is NewReaderFrom with request cancellation.
func NewReaderFromContext(ctx context.Context, urlOrPath string) *Reader { func NewReaderFromContext(ctx context.Context, urlOrPath string) *Reader {
return NewReaderFromContextWith(ctx, httpClient, urlOrPath)
}
// NewReaderFromContextWith is NewReaderFromContext with a custom HTTP client.
func NewReaderFromContextWith(ctx context.Context, client *http.Client, urlOrPath string) *Reader {
if strings.HasPrefix(urlOrPath, "https://") || strings.HasPrefix(urlOrPath, "http://") { if strings.HasPrefix(urlOrPath, "https://") || strings.HasPrefix(urlOrPath, "http://") {
return NewReaderFromURLContext(ctx, urlOrPath) return NewReaderFromURLWith(ctx, client, urlOrPath)
} }
return NewReaderFrom(urlOrPath) return NewReaderFrom(urlOrPath)
} }
@ -74,8 +79,13 @@ func NewReaderFrom(urlOrPath string) *Reader {
} }
func NewReaderFromURLContext(ctx context.Context, url string) *Reader { func NewReaderFromURLContext(ctx context.Context, url string) *Reader {
return NewReaderFromURLWith(ctx, httpClient, url)
}
// NewReaderFromURLWith is NewReaderFromURLContext with a custom HTTP client.
func NewReaderFromURLWith(ctx context.Context, client *http.Client, url string) *Reader {
docid, gid := ParseIDs(url) docid, gid := ParseIDs(url)
return NewReaderFromIDsContext(ctx, httpClient, docid, gid) return NewReaderFromIDsContext(ctx, client, docid, gid)
} }
func NewReaderFromURL(url string) *Reader { func NewReaderFromURL(url string) *Reader {