Update TraceSegment

src/SkyWalking.Core/Context/Trace/TraceSegment.cs

+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+using System.Collections.Generic;
+using System.Linq;
+using Google.Protobuf;
+using SkyWalking.Config;
+using SkyWalking.Context.Ids;
+using SkyWalking.NetworkProtocol;
+
+namespace SkyWalking.Context.Trace
+{
+    public class TraceSegment : ITraceSegment
+    {
+        private readonly IList<ITraceSegmentRef> _refs;
+        private readonly IList<AbstractTracingSpan> _spans;
+        private readonly DistributedTraceIdCollection _relatedGlobalTraces;
+        private bool _isSizeLimited = false;
+
+
+        public int ApplicationId => RemoteDownstreamConfig.Agent.ApplicationId;
+
+        public int ApplicationInstanceId => RemoteDownstreamConfig.Agent.ApplicationInstanceId;
+
+        public IEnumerable<ITraceSegmentRef> Refs => _refs;
+
+        public IEnumerable<DistributedTraceId> RelatedGlobalTraces { get; }
+
+        public ID TraceSegmentId { get; }
+
+        public bool HasRef => _refs.Count > 0;
+
+        public bool IsIgnore { get; set; }
+
+        public bool IsSingleSpanSegment => _spans.Count == 1;
+
+        public TraceSegment()
+        {
+            TraceSegmentId = GlobalIdGenerator.Generate();
+            _spans = new List<AbstractTracingSpan>();
+            _relatedGlobalTraces = new DistributedTraceIdCollection();
+            _relatedGlobalTraces.Append(new NewDistributedTraceId());
+            _refs = new List<ITraceSegmentRef>();
+        }
+
+        public void Archive(AbstractTracingSpan finishedSpan)
+        {
+            _spans.Add(finishedSpan);
+        }
+
+        public ITraceSegment Finish(bool isSizeLimited)
+        {
+            _isSizeLimited = isSizeLimited;
+            return this;
+        }
+
+        /// <summary>
+        /// Establish the link between this segment and its parents.
+        /// </summary>
+        public void Ref(ITraceSegmentRef refSegment)
+        {
+            if (!_refs.Contains(refSegment))
+            {
+                _refs.Add(refSegment);
+            }
+        }
+
+
+        public void RelatedGlobalTrace(DistributedTraceId distributedTraceId)
+        {
+            _relatedGlobalTraces.Append(distributedTraceId);
+        }
+
+        public UpstreamSegment Transform()
+        {
+            var upstreamSegment = new UpstreamSegment();
+            upstreamSegment.GlobalTraceIds.AddRange(_relatedGlobalTraces.GetRelatedGlobalTraces()
+                .Select(x => x.ToUniqueId()));
+
+            var traceSegment = new TraceSegmentObject {TraceSegmentId = TraceSegmentId.Transform()};
+
+            traceSegment.Spans.AddRange(_spans.Select(x => x.Transform()));
+
+            traceSegment.ApplicationId = ApplicationId;
+            traceSegment.ApplicationInstanceId = ApplicationInstanceId;
+            traceSegment.IsSizeLimited = _isSizeLimited;
+
+            upstreamSegment.Segment = traceSegment.ToByteString();
+
+            return upstreamSegment;
+        }
+
+        public override string ToString()
+        {
+            return "TraceSegment{"
+                   +
+                   $"traceSegmentId='{TraceSegmentId}', refs={_refs}, spans={_spans}, relatedGlobalTraces={_relatedGlobalTraces}"
+                   + "}";
+        }
+    }
+}
\ No newline at end of file