Skip to content

Commit 59c3d62

Browse files
committed
refactor: mysql datasource param
1 parent 624ae12 commit 59c3d62

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

datasource/mysql/mysql.go

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,12 @@ func (m *MySQL) QueryData(ctx context.Context, query interface{}) ([]models.Data
127127
return nil, fmt.Errorf("valueKey is required")
128128
}
129129

130-
timeoutCtx, cancel := context.WithTimeout(ctx, time.Duration(m.Shards[0].Timeout)*time.Second)
130+
timeout := m.Shards[0].Timeout
131+
if timeout == 0 {
132+
timeout = 60
133+
}
134+
135+
timeoutCtx, cancel := context.WithTimeout(ctx, time.Duration(timeout)*time.Second)
131136
defer cancel()
132137

133138
items, err := m.QueryTimeseries(timeoutCtx, &sqlbase.QueryParam{
@@ -169,7 +174,12 @@ func (m *MySQL) QueryLog(ctx context.Context, query interface{}) ([]interface{},
169174
}
170175
}
171176

172-
timeoutCtx, cancel := context.WithTimeout(ctx, time.Duration(m.Shards[0].Timeout)*time.Second)
177+
timeout := m.Shards[0].Timeout
178+
if timeout == 0 {
179+
timeout = 60
180+
}
181+
182+
timeoutCtx, cancel := context.WithTimeout(ctx, time.Duration(timeout)*time.Second)
173183
defer cancel()
174184

175185
items, err := m.Query(timeoutCtx, &sqlbase.QueryParam{

0 commit comments

Comments
 (0)