1414// limitations under the License.
1515// </copyright>
1616
17- namespace OpenCensus . Trace
17+ namespace OpenCensus . Trace . Internal
1818{
1919 using OpenCensus . Internal ;
2020 using OpenCensus . Trace . Export ;
2121
22- public sealed class StartEndHandler : IStartEndHandler
22+ internal sealed class StartEndHandler : IStartEndHandler
2323 {
2424 private readonly ISpanExporter spanExporter ;
2525 private readonly IRunningSpanStore runningSpanStore ;
@@ -39,7 +39,7 @@ public StartEndHandler(ISpanExporter spanExporter, IRunningSpanStore runningSpan
3939 this . eventQueue = eventQueue ;
4040 }
4141
42- public void OnEnd ( SpanBase span )
42+ public void OnEnd ( ISpan span )
4343 {
4444 if ( ( span . Options . HasFlag ( SpanOptions . RecordEvents ) && this . enqueueEventForNonSampledSpans )
4545 || span . Context . TraceOptions . IsSampled )
@@ -48,7 +48,7 @@ public void OnEnd(SpanBase span)
4848 }
4949 }
5050
51- public void OnStart ( SpanBase span )
51+ public void OnStart ( ISpan span )
5252 {
5353 if ( span . Options . HasFlag ( SpanOptions . RecordEvents ) && this . enqueueEventForNonSampledSpans )
5454 {
@@ -58,10 +58,10 @@ public void OnStart(SpanBase span)
5858
5959 private sealed class SpanStartEvent : IEventQueueEntry
6060 {
61- private readonly SpanBase span ;
61+ private readonly ISpan span ;
6262 private readonly IRunningSpanStore activeSpansExporter ;
6363
64- public SpanStartEvent ( SpanBase span , IRunningSpanStore activeSpansExporter )
64+ public SpanStartEvent ( ISpan span , IRunningSpanStore activeSpansExporter )
6565 {
6666 this . span = span ;
6767 this . activeSpansExporter = activeSpansExporter ;
@@ -78,13 +78,13 @@ public void Process()
7878
7979 private sealed class SpanEndEvent : IEventQueueEntry
8080 {
81- private readonly SpanBase span ;
81+ private readonly ISpan span ;
8282 private readonly IRunningSpanStore runningSpanStore ;
8383 private readonly ISpanExporter spanExporter ;
8484 private readonly ISampledSpanStore sampledSpanStore ;
8585
8686 public SpanEndEvent (
87- SpanBase span ,
87+ ISpan span ,
8888 ISpanExporter spanExporter ,
8989 IRunningSpanStore runningSpanStore ,
9090 ISampledSpanStore sampledSpanStore )
0 commit comments