// StreamExecuteKeyRanges2 is the RPC version of // vtgateservice.VTGateService method func (vtg *VTGate) StreamExecuteKeyRanges2(ctx context.Context, request *gorpcvtgatecommon.KeyRangeQuery, sendReply func(interface{}) error) (err error) { defer vtg.server.HandlePanic(&err) ctx = callerid.NewContext(ctx, gorpccallerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.StreamExecuteKeyRanges(ctx, request.Sql, request.BindVariables, request.Keyspace, request.KeyRanges, request.TabletType, func(value *sqltypes.Result) error { return sendReply(&gorpcvtgatecommon.QueryResult{ Result: value, }) }) if vtgErr == nil { return nil } // If there was an app error, send a QueryResult back with it. return sendReply(&gorpcvtgatecommon.QueryResult{ Err: vterrors.RPCErrFromVtError(vtgErr), }) }
// Rollback2 is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Rollback2(ctx context.Context, request *gorpcvtgatecommon.RollbackRequest, reply *gorpcvtgatecommon.RollbackResponse) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, gorpccallerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) sessionFromRPC(request.Session) vtgErr := vtg.server.Rollback(ctx, request.Session) reply.Err = vterrors.RPCErrFromVtError(vtgErr) return nil }
// Begin2 is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Begin2(ctx context.Context, request *gorpcvtgatecommon.BeginRequest, reply *gorpcvtgatecommon.BeginResponse) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, gorpccallerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) // Don't pass in a nil pointer session, vtgErr := vtg.server.Begin(ctx) reply.Session = sessionToRPC(session) reply.Err = vterrors.RPCErrFromVtError(vtgErr) return nil }
// StreamExecute is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) StreamExecute(ctx context.Context, request *gorpcvtgatecommon.Query, sendReply func(interface{}) error) (err error) { defer vtg.server.HandlePanic(&err) ctx = callerid.NewContext(ctx, gorpccallerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) return vtg.server.StreamExecute(ctx, request.Sql, request.BindVariables, request.TabletType, func(value *sqltypes.Result) error { return sendReply(&gorpcvtgatecommon.QueryResult{ Result: value, }) }) }
// SplitQuery is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) SplitQuery(ctx context.Context, request *gorpcvtgatecommon.SplitQueryRequest, reply *gorpcvtgatecommon.SplitQueryResult) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, gorpccallerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) var vtgErr error reply.Splits, vtgErr = vtg.server.SplitQuery(ctx, request.Keyspace, request.Query.Sql, request.Query.BindVariables, request.SplitColumn, request.SplitCount) reply.Err = vterrors.RPCErrFromVtError(vtgErr) return nil }
// Execute is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Execute(ctx context.Context, request *gorpcvtgatecommon.Query, reply *gorpcvtgatecommon.QueryResult) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, gorpccallerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) sessionFromRPC(request.Session) var vtgErr error reply.Result, vtgErr = vtg.server.Execute(ctx, request.Sql, request.BindVariables, request.TabletType, request.Session, request.NotInTransaction) reply.Session = sessionToRPC(request.Session) reply.Err = vterrors.RPCErrFromVtError(vtgErr) return nil }
// ExecuteBatchKeyspaceIds is the RPC version of // vtgateservice.VTGateService method func (vtg *VTGate) ExecuteBatchKeyspaceIds(ctx context.Context, request *gorpcvtgatecommon.KeyspaceIdBatchQuery, reply *gorpcvtgatecommon.QueryResultList) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, gorpccallerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) sessionFromRPC(request.Session) qs, err := gorpcvtgatecommon.BoundKeyspaceIdQueriesToProto(request.Queries) if err != nil { return err } var vtgErr error reply.List, vtgErr = vtg.server.ExecuteBatchKeyspaceIds(ctx, qs, request.TabletType, request.AsTransaction, request.Session) reply.Session = sessionToRPC(request.Session) reply.Err = vterrors.RPCErrFromVtError(vtgErr) return nil }