Skip to content

Commit 83e9765

Browse files
refactor: linters
1 parent 3208ead commit 83e9765

File tree

4 files changed

+29
-13
lines changed

4 files changed

+29
-13
lines changed

tests/slo/database/sql/storage.go

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ func (s *Storage) Write(ctx context.Context, e generator.Row) (attempts int, err
169169
ctx, cancel := context.WithTimeout(ctx, time.Duration(s.cfg.WriteTimeout)*time.Millisecond)
170170
defer cancel()
171171

172-
return attempts, retry.Do(ydb.WithTxControl(ctx, writeTx), s.db,
172+
err = retry.Do(ydb.WithTxControl(ctx, writeTx), s.db,
173173
func(ctx context.Context, cc *sql.Conn) (err error) {
174174
if err = ctx.Err(); err != nil {
175175
return err
@@ -199,6 +199,8 @@ func (s *Storage) Write(ctx context.Context, e generator.Row) (attempts int, err
199199
),
200200
),
201201
)
202+
203+
return attempts, err
202204
}
203205

204206
func (s *Storage) createTable(ctx context.Context) error {
@@ -239,10 +241,16 @@ func (s *Storage) close(ctx context.Context) error {
239241
}
240242

241243
if err := s.db.Close(); err != nil {
242-
return err
244+
return fmt.Errorf("error close database/sql driver: %w", err)
243245
}
246+
244247
if err := s.c.Close(); err != nil {
245-
return err
248+
return fmt.Errorf("error close connector: %w", err)
246249
}
247-
return s.cc.Close(ctx)
250+
251+
if err := s.cc.Close(ctx); err != nil {
252+
return fmt.Errorf("error close ydb driver: %w", err)
253+
}
254+
255+
return nil
248256
}

tests/slo/gorm/storage.go

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ func (s *Storage) Read(ctx context.Context, id generator.RowID) (r generator.Row
100100
return generator.Row{}, attempts, err
101101
}
102102

103-
return r, attempts, retry.Do(ydbSDK.WithTxControl(ctx, readTx), db,
103+
err = retry.Do(ydbSDK.WithTxControl(ctx, readTx), db,
104104
func(ctx context.Context, cc *sql.Conn) (err error) {
105105
if err = ctx.Err(); err != nil {
106106
return err
@@ -135,6 +135,8 @@ func (s *Storage) Read(ctx context.Context, id generator.RowID) (r generator.Row
135135
),
136136
),
137137
)
138+
139+
return r, attempts, err
138140
}
139141

140142
func (s *Storage) Write(ctx context.Context, row generator.Row) (attempts int, err error) {
@@ -150,7 +152,7 @@ func (s *Storage) Write(ctx context.Context, row generator.Row) (attempts int, e
150152
return attempts, err
151153
}
152154

153-
return attempts, retry.Do(ydbSDK.WithTxControl(ctx, writeTx), db,
155+
err = retry.Do(ydbSDK.WithTxControl(ctx, writeTx), db,
154156
func(ctx context.Context, cc *sql.Conn) (err error) {
155157
if err = ctx.Err(); err != nil {
156158
return err
@@ -183,6 +185,8 @@ func (s *Storage) Write(ctx context.Context, row generator.Row) (attempts int, e
183185
),
184186
),
185187
)
188+
189+
return attempts, err
186190
}
187191

188192
func (s *Storage) createTable(ctx context.Context) error {

tests/slo/native/storage.go

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ func (s *Storage) Read(ctx context.Context, entryID generator.RowID) (_ generato
106106

107107
e := generator.Row{}
108108

109-
return e, attempts, s.db.Table().Do(ctx,
109+
err = s.db.Table().Do(ctx,
110110
func(ctx context.Context, session table.Session) (err error) {
111111
if err = ctx.Err(); err != nil {
112112
return err
@@ -157,17 +157,19 @@ func (s *Storage) Read(ctx context.Context, entryID generator.RowID) (_ generato
157157
},
158158
}),
159159
)
160+
161+
return e, attempts, err
160162
}
161163

162-
func (s *Storage) Write(ctx context.Context, e generator.Row) (attempts int, err error) {
164+
func (s *Storage) Write(ctx context.Context, e generator.Row) (attempts int, _ error) {
163165
if err := ctx.Err(); err != nil {
164166
return attempts, err
165167
}
166168

167169
ctx, cancel := context.WithTimeout(ctx, time.Duration(s.cfg.WriteTimeout)*time.Millisecond)
168170
defer cancel()
169171

170-
return attempts, s.db.Table().Do(ctx,
172+
err := s.db.Table().Do(ctx,
171173
func(ctx context.Context, session table.Session) error {
172174
if err := ctx.Err(); err != nil {
173175
return err
@@ -203,6 +205,8 @@ func (s *Storage) Write(ctx context.Context, e generator.Row) (attempts int, err
203205
},
204206
}),
205207
)
208+
209+
return attempts, err
206210
}
207211

208212
func (s *Storage) createTable(ctx context.Context) error {

tests/slo/xorm/storage.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -217,19 +217,19 @@ func (s *Storage) close(ctx context.Context) error {
217217
}
218218

219219
if err := s.x.Context(ctx).Close(); err != nil {
220-
return fmt.Errorf("close sessions pool error: %w", err)
220+
return fmt.Errorf("error close sessions pool: %w", err)
221221
}
222222

223223
if err := s.db.Close(); err != nil {
224-
return fmt.Errorf("close database/sql driver error: %w", err)
224+
return fmt.Errorf("error close database/sql driver: %w", err)
225225
}
226226

227227
if err := s.c.Close(); err != nil {
228-
return fmt.Errorf("close connector error: %w", err)
228+
return fmt.Errorf("error close connector: %w", err)
229229
}
230230

231231
if err := s.cc.Close(ctx); err != nil {
232-
return fmt.Errorf("close ydb driver error: %w", err)
232+
return fmt.Errorf("error close ydb driver: %w", err)
233233
}
234234

235235
return nil

0 commit comments

Comments
 (0)