前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >IIncrementalGenerator 增量 Source Generator 生成代码入门 判断程序集的引用关系

IIncrementalGenerator 增量 Source Generator 生成代码入门 判断程序集的引用关系

作者头像
林德熙
发布2023-11-28 13:10:32
1400
发布2023-11-28 13:10:32
举报
文章被收录于专栏:林德熙的博客林德熙的博客

本文将告诉大家如何在 IIncrementalGenerator 增量 Source Generator 生成代码里面,在 Roslyn 分析器里面判断两个程序集是否存在引用关系

先上核心代码实现,核心代码就是调用以下代码进行判断程序集的引用关系

代码语言:javascript
复制
        private static bool IsReference(IAssemblySymbol currentAssemblySymbol, IAssemblySymbol requiredAssemblySymbol,
            Dictionary<IAssemblySymbol, bool /*是否引用*/> visited)
        {
            if (SymbolEqualityComparer.Default.Equals(currentAssemblySymbol, requiredAssemblySymbol))
            {
                // 这个就看业务了,如果两个程序集是相同的,是否判断为引用关系
                return true;
            }

            foreach (var moduleSymbol in currentAssemblySymbol.Modules)
            {
                foreach (var referencedAssemblySymbol in moduleSymbol.ReferencedAssemblySymbols)
                {
                    if (SymbolEqualityComparer.Default.Equals(referencedAssemblySymbol, requiredAssemblySymbol))
                    {
                        // 记录当前程序集存在引用关系
                        visited[currentAssemblySymbol] = true;
                        return true;
                    }
                    else
                    {
                        if (visited.TryGetValue(referencedAssemblySymbol, out var isReference))
                        {
                            // 这个是访问过的,那就从字典获取缓存,不需要再访问一次
                            // 同时也能解决程序集循环引用问题
                        }
                        else
                        {
                            // 没有访问过的,获取引用的程序集是否存在引用关系
                            isReference = IsReference(referencedAssemblySymbol, requiredAssemblySymbol, visited);
                            visited[referencedAssemblySymbol] = isReference;
                        }

                        if (isReference)
                        {
                            // 如果这个程序集有引用,那也算上
                            return true;
                        }
                    }
                }
            }

            return false;
        }

核心判断逻辑就是获取当前程序集的 Modules 从而获取到 ReferencedAssemblySymbols 也就是当前程序集所引用的所有程序集,再判断引用的程序集里面是否包含期望被引用的程序集

对于间接引用的程序集来说,需要进行遍历才可以判断到,但是此时需要处理程序集循环引用问题。本文这里采用传入一个字典的方法解决这个问题

获取当前分析器所分析的项目的引用程序集,可以通过 Compilation 的 SourceModule 的 ReferencedAssemblySymbols 拿到,如以下代码

代码语言:javascript
复制
            context.CompilationProvider.Select((compilation, _) =>
            {
                var sourceModuleReferencedAssemblySymbols = compilation.SourceModule.ReferencedAssemblySymbols;
                ... // 忽略代码
            });

本文所有代码放在 githubgitee 上,可以通过以下方式获取整个项目的代码

先创建一个空文件夹,接着使用命令行 cd 命令进入此空文件夹,在命令行里面输入以下代码,即可获取到本文的代码

代码语言:javascript
复制
git init
git remote add origin https://gitee.com/lindexi/lindexi_gd.git
git pull origin c87c7e78b068956f3f8f4640a85c04a3af9e07fc

以上使用的是 gitee 的源,如果 gitee 不能访问,请替换为 github 的源。请在命令行继续输入以下代码

代码语言:javascript
复制
git remote remove origin
git remote add origin https://github.com/lindexi/lindexi_gd.git
git pull origin c87c7e78b068956f3f8f4640a85c04a3af9e07fc

获取代码之后,进入 HulanucerbeljuChaijacemjarga 文件夹

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档