// 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 }
// Execute is exposing tabletserver.SqlQuery.Execute func (sq *SqlQuery) Execute(ctx context.Context, query *proto.Query, reply *mproto.QueryResult) (err error) { defer sq.server.HandlePanic(&err) tErr := sq.server.Execute(callinfo.RPCWrapCallInfo(ctx), nil, query, reply) tabletserver.AddTabletErrorToQueryResult(tErr, reply) if *tabletserver.RPCErrorOnlyInReply { return nil } return tErr }
// 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.AddTabletErrorToQueryResult(tErr, reply) 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.AddTabletErrorToQueryResult(tErr, qr) return sendReply(qr) }