PsiExpression (PsiLambdaExpression lambda, PsiExpression[] args, BulkMethodInfo info) { PsiParameterList parameterList = lambda.getParameterList(); PsiParameter[] parameters = parameterList.getParameters(); int lambdaParametersCount = parameterList.getParametersCount(); if (info.getClassName().equals(CommonClassNames.JAVA_UTIL_MAP)) { if (lambdaParametersCount == 1) { if (!isGetKeyAndGetValue(args, parameters[0])) return null; } else if (lambdaParametersCount == 2) { if (!ExpressionUtils.isReferenceTo(args[0], parameters[0]) || !ExpressionUtils.isReferenceTo(args[1], parameters[1])) return null; } else return null; } else if (lambdaParametersCount != 1 || !ExpressionUtils.isReferenceTo(args[0], parameters[0])) return null; return findIterableForFunction(lambda); }