// 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, callerid.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), }) }
// StreamExecute2 should not be used by anything other than tests. // It will eventually replace Rollback, but it breaks compatibility with older clients. // Once all clients are upgraded, it can be replaced. func (sq *SqlQuery) StreamExecute2(ctx context.Context, req *proto.StreamExecuteRequest, sendReply func(reply interface{}) error) (err error) { defer sq.server.HandlePanic(&err) if req == nil || req.Query == nil { return nil } ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(req.EffectiveCallerID), callerid.GoRPCImmediateCallerID(req.ImmediateCallerID), ) tErr := sq.server.StreamExecute(callinfo.RPCWrapCallInfo(ctx), proto.TargetToProto3(req.Target), req.Query, func(reply *mproto.QueryResult) error { return sendReply(reply) }) if tErr == nil { return nil } if *tabletserver.RPCErrorOnlyInReply { // If there was an app error, send a QueryResult back with it. qr := new(mproto.QueryResult) tabletserver.AddTabletErrorToQueryResult(tErr, qr) // Sending back errors this way is not backwards compatible. If a (new) server sends an additional // QueryResult with an error, and the (old) client doesn't know how to read it, it will cause // problems where the client will get out of sync with the number of QueryResults sent. // That's why this the error is only sent this way when the --rpc_errors_only_in_reply flag is set // (signalling that all clients are able to handle new-style errors). return sendReply(qr) } return tErr }
// StreamExecuteKeyRanges2 is the RPC version of // vtgateservice.VTGateService method func (vtg *VTGate) StreamExecuteKeyRanges2(ctx context.Context, request *proto.KeyRangeQuery, sendReply func(interface{}) error) (err error) { defer vtg.server.HandlePanic(&err) ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.StreamExecuteKeyRanges(ctx, request.Sql, request.BindVariables, request.Keyspace, key.KeyRangesToProto(request.KeyRanges), topo.TabletTypeToProto(request.TabletType), func(value *proto.QueryResult) error { return sendReply(value) }) if vtgErr == nil { return nil } if *vtgate.RPCErrorOnlyInReply { // If there was an app error, send a QueryResult back with it. qr := new(proto.QueryResult) vtgate.AddVtGateErrorToQueryResult(vtgErr, qr) // Sending back errors this way is not backwards compatible. If a (new) server sends an additional // QueryResult with an error, and the (old) client doesn't know how to read it, it will cause // problems where the client will get out of sync with the number of QueryResults sent. // That's why this the error is only sent this way when the --rpc_errors_only_in_reply flag is set // (signalling that all clients are able to handle new-style errors). return sendReply(qr) } return vtgErr }
// StreamExecuteKeyspaceIds is the RPC version of // vtgateservice.VTGateService method func (vtg *VTGate) StreamExecuteKeyspaceIds(ctx context.Context, request *proto.KeyspaceIdQuery, sendReply func(interface{}) error) (err error) { defer vtg.server.HandlePanic(&err) ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) return vtg.server.StreamExecuteKeyspaceIds(ctx, request, func(value *proto.QueryResult) error { return sendReply(value) }) }
// Execute2 should not be used by anything other than tests // It will eventually replace Execute, but it breaks compatibility with older clients // Once all clients are upgraded, it can be replaced func (sq *SqlQuery) Execute2(ctx context.Context, executeRequest *proto.ExecuteRequest, reply *mproto.QueryResult) (err error) { defer sq.server.HandlePanic(&err) ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(executeRequest.EffectiveCallerID), callerid.GoRPCImmediateCallerID(executeRequest.ImmediateCallerID), ) tErr := sq.server.Execute(callinfo.RPCWrapCallInfo(ctx), proto.TargetToProto3(executeRequest.Target), &executeRequest.QueryRequest, reply) tabletserver.AddTabletError(tErr, &reply.Err) return nil }
// SplitQuery is exposing tabletserver.SqlQuery.SplitQuery func (sq *SqlQuery) SplitQuery(ctx context.Context, req *proto.SplitQueryRequest, reply *proto.SplitQueryResult) (err error) { defer sq.server.HandlePanic(&err) ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(req.EffectiveCallerID), callerid.GoRPCImmediateCallerID(req.ImmediateCallerID), ) tErr := sq.server.SplitQuery(callinfo.RPCWrapCallInfo(ctx), proto.TargetToProto3(req.Target), req, reply) tabletserver.AddTabletError(tErr, &reply.Err) return nil }
// Rollback2 is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Rollback2(ctx context.Context, request *proto.RollbackRequest, reply *proto.RollbackResponse) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.Rollback(ctx, request.Session) vtgate.AddVtGateError(vtgErr, &reply.Err) return nil }
// Commit2 is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Commit2(ctx context.Context, request *gorpcvtgatecommon.CommitRequest, reply *gorpcvtgatecommon.CommitResponse) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) sessionFromRPC(request.Session) vtgErr := vtg.server.Commit(ctx, request.Session) reply.Err = vterrors.RPCErrFromVtError(vtgErr) return nil }
// ExecuteBatch2 should not be used by anything other than tests // It will eventually replace ExecuteBatch, but it breaks compatibility with older clients. // Once all clients are upgraded, it can be replaced. func (sq *SqlQuery) ExecuteBatch2(ctx context.Context, req *proto.ExecuteBatchRequest, reply *proto.QueryResultList) (err error) { defer sq.server.HandlePanic(&err) if req == nil { return nil } ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(req.EffectiveCallerID), callerid.GoRPCImmediateCallerID(req.ImmediateCallerID), ) tErr := sq.server.ExecuteBatch(callinfo.RPCWrapCallInfo(ctx), proto.TargetToProto3(req.Target), &req.QueryBatch, reply) tabletserver.AddTabletError(tErr, &reply.Err) return nil }
// StreamExecute is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) StreamExecute(ctx context.Context, request *proto.Query, sendReply func(interface{}) error) (err error) { defer vtg.server.HandlePanic(&err) ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) return vtg.server.StreamExecute(ctx, request.Sql, request.BindVariables, topo.TabletTypeToProto(request.TabletType), func(value *proto.QueryResult) error { return sendReply(value) }) }
// 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, callerid.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 }
// Commit2 is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Commit2(ctx context.Context, request *proto.CommitRequest, reply *proto.CommitResponse) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.Commit(ctx, request.Session) vtgate.AddVtGateErrorToCommitResponse(vtgErr, reply) if *vtgate.RPCErrorOnlyInReply { return nil } return vtgErr }
// Commit2 should not be used by anything other than tests. // It will eventually replace Commit, but it breaks compatibility with older clients. // Once all clients are upgraded, it can be replaced. func (sq *SqlQuery) Commit2(ctx context.Context, commitRequest *proto.CommitRequest, commitResponse *proto.CommitResponse) (err error) { defer sq.server.HandlePanic(&err) session := &proto.Session{ SessionId: commitRequest.SessionId, TransactionId: commitRequest.TransactionId, } ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(commitRequest.EffectiveCallerID), callerid.GoRPCImmediateCallerID(commitRequest.ImmediateCallerID), ) tErr := sq.server.Commit(callinfo.RPCWrapCallInfo(ctx), proto.TargetToProto3(commitRequest.Target), session) tabletserver.AddTabletError(tErr, &commitResponse.Err) return nil }
// ExecuteEntityIds is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) ExecuteEntityIds(ctx context.Context, request *proto.EntityIdsQuery, reply *proto.QueryResult) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.ExecuteEntityIds(ctx, request, reply) vtgate.AddVtGateErrorToQueryResult(vtgErr, reply) if *vtgate.RPCErrorOnlyInReply { return nil } return vtgErr }
// 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, callerid.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, }) }) }
// Begin2 should not be used by anything other than tests. // It will eventually replace Begin, but it breaks compatibility with older clients. // Once all clients are upgraded, it can be replaced. func (sq *SqlQuery) Begin2(ctx context.Context, beginRequest *proto.BeginRequest, beginResponse *proto.BeginResponse) (err error) { defer sq.server.HandlePanic(&err) session := &proto.Session{ SessionId: beginRequest.SessionId, } txInfo := new(proto.TransactionInfo) ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(beginRequest.EffectiveCallerID), callerid.GoRPCImmediateCallerID(beginRequest.ImmediateCallerID), ) tErr := sq.server.Begin(callinfo.RPCWrapCallInfo(ctx), proto.TargetToProto3(beginRequest.Target), session, txInfo) // Convert from TxInfo => beginResponse for the output beginResponse.TransactionId = txInfo.TransactionId tabletserver.AddTabletError(tErr, &beginResponse.Err) return nil }
// Begin2 is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Begin2(ctx context.Context, request *proto.BeginRequest, reply *proto.BeginResponse) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) // Don't pass in a nil pointer reply.Session = &proto.Session{} vtgErr := vtg.server.Begin(ctx, reply.Session) vtgate.AddVtGateError(vtgErr, &reply.Err) if *vtgate.RPCErrorOnlyInReply { return nil } return vtgErr }
// ExecuteBatchKeyspaceIds is the RPC version of // vtgateservice.VTGateService method func (vtg *VTGate) ExecuteBatchKeyspaceIds(ctx context.Context, request *proto.KeyspaceIdBatchQuery, reply *proto.QueryResultList) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.ExecuteBatchKeyspaceIds(ctx, request.Queries, topo.TabletTypeToProto(request.TabletType), request.AsTransaction, request.Session, reply) vtgate.AddVtGateError(vtgErr, &reply.Err) return nil }
// SplitQuery is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) SplitQuery(ctx context.Context, request *proto.SplitQueryRequest, reply *proto.SplitQueryResult) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) splits, vtgErr := vtg.server.SplitQuery(ctx, request.Keyspace, request.Query.Sql, request.Query.BindVariables, request.SplitColumn, request.SplitCount) reply.Splits = splits vtgate.AddVtGateError(vtgErr, &reply.Err) return nil }
// Rollback2 should not be used by anything other than tests. // It will eventually replace Rollback, but it breaks compatibility with older clients. // Once all clients are upgraded, it can be replaced. func (sq *SqlQuery) Rollback2(ctx context.Context, rollbackRequest *proto.RollbackRequest, rollbackResponse *proto.RollbackResponse) (err error) { defer sq.server.HandlePanic(&err) session := &proto.Session{ SessionId: rollbackRequest.SessionId, TransactionId: rollbackRequest.TransactionId, } ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(rollbackRequest.EffectiveCallerID), callerid.GoRPCImmediateCallerID(rollbackRequest.ImmediateCallerID), ) tErr := sq.server.Rollback(callinfo.RPCWrapCallInfo(ctx), proto.TargetToProto3(rollbackRequest.Target), session) tabletserver.AddTabletErrorToRollbackResponse(tErr, rollbackResponse) if *tabletserver.RPCErrorOnlyInReply { return nil } return tErr }
// Execute is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Execute(ctx context.Context, request *proto.Query, reply *proto.QueryResult) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) sessionFromRPC(request.Session) vtgErr := vtg.server.Execute(ctx, request.Sql, request.BindVariables, request.TabletType, request.Session, request.NotInTransaction, reply) reply.Session = sessionToRPC(reply.Session) vtgate.AddVtGateError(vtgErr, &reply.Err) return nil }
// StreamExecute2 is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) StreamExecute2(ctx context.Context, request *proto.Query, sendReply func(interface{}) error) (err error) { defer vtg.server.HandlePanic(&err) ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.StreamExecute(ctx, request.Sql, request.BindVariables, request.TabletType, func(value *proto.QueryResult) error { return sendReply(value) }) if vtgErr == nil { return nil } // If there was an app error, send a QueryResult back with it. qr := new(proto.QueryResult) vtgate.AddVtGateError(vtgErr, &qr.Err) return sendReply(qr) }
// ExecuteEntityIds is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) ExecuteEntityIds(ctx context.Context, request *proto.EntityIdsQuery, reply *proto.QueryResult) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.ExecuteEntityIds(ctx, request.Sql, request.BindVariables, request.Keyspace, request.EntityColumnName, proto.EntityIdsToProto(request.EntityKeyspaceIDs), topo.TabletTypeToProto(request.TabletType), request.Session, request.NotInTransaction, reply) vtgate.AddVtGateError(vtgErr, &reply.Err) return nil }
// StreamExecute2 should not be used by anything other than tests. // It will eventually replace Rollback, but it breaks compatibility with older clients. // Once all clients are upgraded, it can be replaced. func (sq *SqlQuery) StreamExecute2(ctx context.Context, req *proto.StreamExecuteRequest, sendReply func(reply interface{}) error) (err error) { defer sq.server.HandlePanic(&err) if req == nil || req.Query == nil { return nil } ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(req.EffectiveCallerID), callerid.GoRPCImmediateCallerID(req.ImmediateCallerID), ) tErr := sq.server.StreamExecute(callinfo.RPCWrapCallInfo(ctx), proto.TargetToProto3(req.Target), req.Query, func(reply *mproto.QueryResult) error { return sendReply(reply) }) if tErr == nil { return nil } // If there was an app error, send a QueryResult back with it. qr := new(mproto.QueryResult) tabletserver.AddTabletError(tErr, &qr.Err) return sendReply(qr) }
// Execute is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) Execute(ctx context.Context, request *proto.Query, reply *proto.QueryResult) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) vtgErr := vtg.server.Execute(ctx, request.Sql, request.BindVariables, topo.TabletTypeToProto(request.TabletType), request.Session, request.NotInTransaction, reply) vtgate.AddVtGateErrorToQueryResult(vtgErr, reply) if *vtgate.RPCErrorOnlyInReply { return nil } return vtgErr }
// ExecuteShard is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) ExecuteShard(ctx context.Context, request *gorpcvtgatecommon.QueryShard, 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, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) sessionFromRPC(request.Session) var vtgErr error reply.Result, vtgErr = vtg.server.ExecuteShards(ctx, request.Sql, request.BindVariables, request.Keyspace, request.Shards, request.TabletType, request.Session, request.NotInTransaction) reply.Session = sessionToRPC(request.Session) reply.Err = vterrors.RPCErrFromVtError(vtgErr) return nil }
// ExecuteBatchShard is the RPC version of vtgateservice.VTGateService method func (vtg *VTGate) ExecuteBatchShard(ctx context.Context, request *proto.BatchQueryShard, reply *proto.QueryResultList) (err error) { defer vtg.server.HandlePanic(&err) ctx, cancel := context.WithDeadline(ctx, time.Now().Add(*rpcTimeout)) defer cancel() ctx = callerid.NewContext(ctx, callerid.GoRPCEffectiveCallerID(request.CallerID), callerid.NewImmediateCallerID("gorpc client")) sessionFromRPC(request.Session) qs, err := proto.BoundShardQueriesToProto(request.Queries) if err != nil { return err } vtgErr := vtg.server.ExecuteBatchShards(ctx, qs, request.TabletType, request.AsTransaction, request.Session, reply) reply.Session = sessionToRPC(reply.Session) vtgate.AddVtGateError(vtgErr, &reply.Err) 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, callerid.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 }