@Override
  public void visitYieldStatement(@NotNull PsiYieldStatement statement) {
    super.visitYieldStatement(statement);
    if (!hasErrorResults()) add(HighlightUtil.checkYieldOutsideSwitchExpression(statement));
    if (!hasErrorResults()) {
      PsiExpression expression = statement.getExpression();
      if (expression != null) {
        add(HighlightUtil.checkYieldExpressionType(expression));
      }
    }
  }
@Override
  public void visitYieldStatement(@NotNull PsiYieldStatement statement)