src/tools/clippy/clippy_lints/src/integer_division.rs

   1 use clippy_utils::diagnostics::span_lint_and_help;
   2 use if_chain::if_chain;
   3 use rustc_hir as hir;
   4 use rustc_lint::{LateContext, LateLintPass};
   5 use rustc_session::{declare_lint_pass, declare_tool_lint};
   6
   7 declare_clippy_lint! {
   8     /// ### What it does
   9     /// Checks for division of integers
  10     ///
  11     /// ### Why is this bad?
  12     /// When outside of some very specific algorithms,
  13     /// integer division is very often a mistake because it discards the
  14     /// remainder.
  15     ///
  16     /// ### Example
  17     /// ```rust
  18     /// // Bad
  19     /// let x = 3 / 2;
  20     /// println!("{}", x);
  21     ///
  22     /// // Good
  23     /// let x = 3f32 / 2f32;
  24     /// println!("{}", x);
  25     /// ```
  26     #[clippy::version = "1.37.0"]
  27     pub INTEGER_DIVISION,
  28     restriction,
  29     "integer division may cause loss of precision"
  30 }
  31
  32 declare_lint_pass!(IntegerDivision => [INTEGER_DIVISION]);
  33
  34 impl<'tcx> LateLintPass<'tcx> for IntegerDivision {
  35     fn check_expr(&mut self, cx: &LateContext<'tcx>, expr: &'tcx hir::Expr<'_>) {
  36         if is_integer_division(cx, expr) {
  37             span_lint_and_help(
  38                 cx,
  39                 INTEGER_DIVISION,
  40                 expr.span,
  41                 "integer division",
  42                 None,
  43                 "division of integers may cause loss of precision. consider using floats",
  44             );
  45         }
  46     }
  47 }
  48
  49 fn is_integer_division<'tcx>(cx: &LateContext<'tcx>, expr: &'tcx hir::Expr<'_>) -> bool {
  50     if_chain! {
  51         if let hir::ExprKind::Binary(binop, left, right) = &expr.kind;
  52         if binop.node == hir::BinOpKind::Div;
  53         then {
  54             let (left_ty, right_ty) = (cx.typeck_results().expr_ty(left), cx.typeck_results().expr_ty(right));
  55             return left_ty.is_integral() && right_ty.is_integral();
  56         }
  57     }
  58
  59     false
  60 }