@@ -13,7 +13,7 @@ import (
1313
1414 "github.com/ydb-platform/ydb-go-genproto/protos/Ydb_Operations"
1515
16- "github.com/ydb-platform/ydb-go-sdk/v3/internal/db "
16+ "github.com/ydb-platform/ydb-go-sdk/v3/internal/database "
1717 "github.com/ydb-platform/ydb-go-sdk/v3/internal/errors"
1818)
1919
@@ -158,7 +158,7 @@ func getField(name string, src, dst interface{}) bool {
158158 return fn (reflect .ValueOf (src ).Elem (), strings .Split (name , "." )... )
159159}
160160
161- type database struct {
161+ type cluster struct {
162162 onInvoke func (
163163 ctx context.Context ,
164164 method string ,
@@ -177,59 +177,59 @@ type database struct {
177177 ) error
178178}
179179
180- func (db * database ) Invoke (
180+ func (c * cluster ) Invoke (
181181 ctx context.Context ,
182182 method string ,
183183 args interface {},
184184 reply interface {},
185185 opts ... grpc.CallOption ,
186186) (err error ) {
187- if db .onInvoke == nil {
187+ if c .onInvoke == nil {
188188 return fmt .Errorf ("database.onInvoke() not implemented" )
189189 }
190- return db .onInvoke (ctx , method , args , reply , opts ... )
190+ return c .onInvoke (ctx , method , args , reply , opts ... )
191191}
192192
193- func (db * database ) NewStream (
193+ func (c * cluster ) NewStream (
194194 ctx context.Context ,
195195 desc * grpc.StreamDesc ,
196196 method string ,
197197 opts ... grpc.CallOption ,
198198) (_ grpc.ClientStream , err error ) {
199- if db .onNewStream == nil {
199+ if c .onNewStream == nil {
200200 return nil , fmt .Errorf ("database.onNewStream() not implemented" )
201201 }
202- return db .onNewStream (ctx , desc , method , opts ... )
202+ return c .onNewStream (ctx , desc , method , opts ... )
203203}
204204
205- func (db * database ) Get (context.Context ) (conn grpc.ClientConnInterface , err error ) {
205+ func (c * cluster ) Get (context.Context ) (conn grpc.ClientConnInterface , err error ) {
206206 cc := & clientConn {
207- onInvoke : db .onInvoke ,
208- onNewStream : db .onNewStream ,
207+ onInvoke : c .onInvoke ,
208+ onNewStream : c .onNewStream ,
209209 }
210210 return cc , nil
211211}
212212
213- func (db * database ) Name () string {
213+ func (c * cluster ) Name () string {
214214 return "testutil.database"
215215}
216216
217- func (db * database ) Close (ctx context.Context ) error {
218- if db .onClose == nil {
217+ func (c * cluster ) Close (ctx context.Context ) error {
218+ if c .onClose == nil {
219219 return fmt .Errorf ("database.Close() not implemented" )
220220 }
221- return db .onClose (ctx )
221+ return c .onClose (ctx )
222222}
223223
224224type (
225225 InvokeHandlers map [MethodCode ]func (request interface {}) (result proto.Message , err error )
226226 NewStreamHandlers map [MethodCode ]func (desc * grpc.StreamDesc ) (grpc.ClientStream , error )
227227)
228228
229- type dbOption func (c * database )
229+ type clusterOption func (c * cluster )
230230
231- func WithInvokeHandlers (invokeHandlers InvokeHandlers ) dbOption {
232- return func (db * database ) {
231+ func WithInvokeHandlers (invokeHandlers InvokeHandlers ) clusterOption {
232+ return func (db * cluster ) {
233233 db .onInvoke = func (
234234 ctx context.Context ,
235235 method string ,
@@ -260,8 +260,8 @@ func WithInvokeHandlers(invokeHandlers InvokeHandlers) dbOption {
260260 }
261261}
262262
263- func WithNewStreamHandlers (newStreamHandlers NewStreamHandlers ) dbOption {
264- return func (db * database ) {
263+ func WithNewStreamHandlers (newStreamHandlers NewStreamHandlers ) clusterOption {
264+ return func (db * cluster ) {
265265 db .onNewStream = func (
266266 ctx context.Context ,
267267 desc * grpc.StreamDesc ,
@@ -276,14 +276,14 @@ func WithNewStreamHandlers(newStreamHandlers NewStreamHandlers) dbOption {
276276 }
277277}
278278
279- func WithClose (onClose func (ctx context.Context ) error ) dbOption {
280- return func (c * database ) {
279+ func WithClose (onClose func (ctx context.Context ) error ) clusterOption {
280+ return func (c * cluster ) {
281281 c .onClose = onClose
282282 }
283283}
284284
285- func NewDB (opts ... dbOption ) db .Cluster {
286- c := & database {}
285+ func NewDB (opts ... clusterOption ) database .Cluster {
286+ c := & cluster {}
287287 for _ , opt := range opts {
288288 opt (c )
289289 }
0 commit comments