code: /*
   * 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.
   *
   */
  
  package org.apache.skywalking.oap.server.core.profiling.trace.analyze;
  
  import lombok.Data;
  import org.apache.skywalking.oap.server.core.query.type.ProfileStackElement;
  import org.apache.skywalking.oap.server.core.query.type.ProfileStackTree;
  import org.apache.skywalking.oap.server.library.util.CollectionUtils;
  import org.junit.jupiter.api.Assertions;
  
  import java.util.Collections;
  import java.util.Comparator;
  import java.util.List;
  import java.util.regex.Matcher;
  import java.util.regex.Pattern;
  import java.util.stream.Collectors;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  
  @Data
  public class ProfileStackElementMatcher {
  
      private static final Pattern DURATION_PATTERN = Pattern.compile("(\\d+)\\:(\\d+)");
  
      private String code;
      private String duration;
      private int count;
      private int id = 0;
      private List<ProfileStackElementMatcher> children;
  
      public void verify(ProfileStackTree tree) {
          // assert root
          List<ProfileStackElement> fromParent = getFromParentId(tree, 0);
          assertEquals(fromParent.size(), 1);
          assertCurrentNode(fromParent.get(0));
  
          setId(fromParent.get(0).getId());
          assertChildren(tree, fromParent.get(0));
      }
  
      private void assertChildren(ProfileStackTree tree, ProfileStackElement parent) {
          List<ProfileStackElement> analyzedChildren = getFromParentId(tree, parent.getId());
  
          if (CollectionUtils.isEmpty(analyzedChildren)) {
              analyzedChildren = Collections.emptyList();
          }
          if (CollectionUtils.isEmpty(getChildren())) {
              setChildren(Collections.emptyList());
          }
          assertEquals(analyzedChildren.size(), children.size());
  
          // children code signature not sorted, need sort it, then verify
          Collections.sort(children, Comparator.comparing(c -> c.code));
          Collections.sort(analyzedChildren, Comparator.comparing(ProfileStackElement::getCodeSignature));
  
          for (int i = 0; i < children.size(); i++) {
              children.get(i).setId(analyzedChildren.get(i).getId());
              children.get(i).assertCurrentNode(analyzedChildren.get(i));
              children.get(i).assertChildren(tree, analyzedChildren.get(i));
          }
      }
  
      private List<ProfileStackElement> getFromParentId(ProfileStackTree tree, int fromParentId) {
          return tree.getElements().stream().filter(e -> e.getParentId() == fromParentId).collect(Collectors.toList());
      }
  
      private void assertCurrentNode(ProfileStackElement element) {
          // analyze duration
          Matcher durationInfo = DURATION_PATTERN.matcher(duration);
          Assertions.assertTrue(durationInfo.find(), "duration field pattern not match");
          int duration = Integer.parseInt(durationInfo.group(1));
          int durationExcludeChild = Integer.parseInt(durationInfo.group(2));
  
          assertEquals(code, element.getCodeSignature());
          assertEquals(duration, element.getDuration());
          assertEquals(durationExcludeChild, element.getDurationChildExcluded());
          assertEquals(count, element.getCount());
      }
  
  }
  