C++实现的代码行数统计器

叶茂林

发布于 2024-05-26 08:28:18

400

发布于 2024-05-26 08:28:18

文章被收录于专栏：叶子的开发者社区叶子的开发者社区

代码在GitHubMaolinYe/CodeCounter: C++20实现的代码统计器，代码量小于100行，可以统计目录下所有代码文件的行数 (github.com)

前段时间到处面试找实习，有技术负责人的负责人问我C++写过多少行，5万还是10万，用来评估熟练度，有点难顶，于是写个代码统计器吧，输入文件夹目录或者代码文件，可以统计所有代码的行数

可以直接编译代码运行程序，在控制台输入目录的路径按下回车即可，例如输入

C:\Users\Yezi\Desktop\C++\CodeCounter

也可以在终端命令行直接运行编译好的程序，带上参数运行，例如输入

.\CodeCounter.exe C:\Users\Yezi\Desktop\C++\CodeCounter

思路比较简单，主要是用到了C++17的filesystem库用来解析目录和提取文件后缀，如果路径是个目录就提取子目录项逐个分析，如果子目录项是目录就递归调用本身继续解析目录，如果是代码文件就开始计数行数

//
// Created by YEZI on 2024/5/20.
//

#ifndef CODECOUNTER_H
#define CODECOUNTER_H
#include<vector>
#include<string>
#include<filesystem>
#include <fstream>
#include <iostream>

class CodeCounter {
    int lines = 0;
    // 检查是否是代码文件
    static bool isCodeFile(const std::filesystem::path &path) {
        // 常见代码文件后缀
        static const std::vector<std::string> extensions = {
            ".cpp", ".h", ".java", ".py", ".cs", ".js", ".go", ".c", ".cc", ".hh"
        };
        // 检查路径是否存在
        if (std::filesystem::exists(path) == false) {
            std::cerr << "There is no file " << path << std::endl;
            return false;
        }
        // 检查是否是文件
        if (is_regular_file(path) == false) {
            std::cerr << path << " is no a file." << std::endl;
            return false;
        }
        std::string extension = path.extension().string();
        for (const auto &e: extensions) {
            if (e == extension) {
                return true;
            }
        }
        return false;
    }

    void countCodeFile(const std::filesystem::path &filePath) {
        // 检查是否是代码文件
        if (isCodeFile(filePath) == false)
            return;
        std::ifstream file(filePath);
        // 检查文件是否可以打开
        if (file.is_open() == false) {
            std::cerr << "Error opening file: " << filePath << std::endl;
            return;
        }
        std::string trash;
        int count=0;
        while (std::getline(file, trash)) {
            ++count;
        }
        lines+=count;
        std::cout<<filePath<<" Lines: "<<count<<std::endl;
    }

    void countDirectory(const std::filesystem::path &path) {
        // 检查是否是目录
        if (is_directory(path) == false)
            return;
        for (const auto &entry: std::filesystem::directory_iterator(path)) {
            if (entry.is_directory())
                countDirectory(entry.path());
            else
                countCodeFile(entry.path());
        }
    }

public:
    void countThis(const std::filesystem::path &path) {
        if (is_directory(path))
            countDirectory(path);
        else
            countCodeFile(path);
        std::cout << "Code Lines: " << lines;
    }
};
#endif //CODECOUNTER_H

从命令行参数读取目录或者从控制台输入读取目录

#include <iostream>
#include"CodeCounter.h"

int main(int argc, char *argv[]) {
    CodeCounter code_counter;
    std::string path;
    if (argc == 2)
        path.assign(argv[1]);
    else
        std::getline(std::cin, path);
    code_counter.countThis(path);
    return 0;
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2024-05-25，如有侵权请联系 cloudcommunity@tencent.com 删除

统计