LayIM项目之基础数据获取代码优化，Dapper取代ADO.NET

前言

　　最近在开发LayIM融云版，也在进行项目重构，现在在看之前的代码，简直不敢直视。不过不知道以后看现在的代码是不是也是糟糕的一批。LayIM有个接口，一般接触过的开发人员都不会生疏，就是init接口。接口返回的数据大概就是酱紫的：

{
    "code": 0,
    "msg": "",
    "data": {
        "mine": {
            "username": "纸飞机",
            "id": "100000",
            "status": "online",
            "sign": "在深邃的编码世界，做一枚轻盈的纸飞机",
            "avatar": "/images/default.jpg"
        },
        "friend": [
            {
                "groupname": "前端码屌",
                "id": 1,
                "online": 2,
                "list": [
                    {
                        "username": "贤心",
                        "id": "100001",
                        "avatar": "/images/default.jpg",
                        "sign": "这些都是测试数据，实际使用请严格按照该格式返回"
                    },
                    {
                        "username": "Z_子晴",
                        "id": "108101",
                        "avatar": "/images/default.jpg",
                        "sign": "微电商达人"
                    }
                ]
            },
            {
                "groupname": "网红",
                "id": 2,
                "online": 3,
                "list": [
                    {
                        "username": "罗玉凤",
                        "id": "121286",
                        "avatar": "/images/default.jpg",
                        "sign": "在自己实力不济的时候，不要去相信什么媒体和记者。他们不是善良的人，有时候候他们的采访对当事人而言就是陷阱"
                    },
                    {
                        "username": "长泽梓Azusa",
                        "id": "100001222",
                        "sign": "我是日本女艺人长泽あずさ",
                        "avatar": "/images/default.jpg"
                    }
                ]
            }
        ],
        "group": [
            {
                "groupname": "前端群",
                "id": "101",
                "avatar": "/images/default.jpg"
            },
            {
                "groupname": "Fly社区官方群",
                "id": "102",
                "avatar": "/images/default.jpg"
            }
        ]
    }
}

　　总之，里面嵌套了很多关系，比如，我的好友分组和好友的关系，还有其他的一些数据，其实，仔细分析一下，也就会各个击破了。今天的重点不是这个数据，而是关于获取这段数据代码的重构。

之前的代码

　　之前用的ADO.NET直接读取的DataSet，然后进行数据处理，主要麻烦的是，还需要手动写DataTable转Model的过程，还要处理关系，比较繁琐，并且需要知道其中的字段。（当然，dapper也需要对应）。先看一下之前的代码，总之这段代码就是很古老的一种形式。

 private  BaseListResult ToBaseListResult(DataSet ds)
        {
            if (ds.Tables.Count > 0)
            {
                if (ds.Tables[0].Rows.Count ==0) {
                    return new BaseListResult();
                }
                //当前用户的信息
                var rowMine = ds.Tables[0].Rows[0];
                //用户组信息
                var rowFriendDetails = ds.Tables[2].Rows.Cast<DataRow>().Select(x => new GroupUserEntity
                {
                    id = x["uid"].ToInt(),
                    avatar = x["avatar"].ToString(),
                    groupid = x["gid"].ToInt(),
                    remarkname = x["remarkname"].ToString(),
                    username = x["nickname"].ToString(),
                    sign = x["sign"].ToString(),
                    //status之前的字段是为空的，现在我们把他的在线状态加上，IsOnline方法接收一个userid参数，从Redis缓存中读取该用户是否在线并返回
                    status = LayIMCache.Instance.IsOnline(x["uid"].ToInt()) ? "online" : "hide"
                }).OrderByDescending(x => x.status);//这里要根据用户是否在线这个字段排序，保证在线用户都在好友列表最上边
                //用户组信息，执行分组
                var friend = ds.Tables[1].Rows.Cast<DataRow>().Select(x => new FriendGroupEntity
                {
                    id = x["id"].ToInt(),
                    groupname = x["name"].ToString(),
                    online = 0,
                    list = rowFriendDetails.Where(f => f.groupid == x["id"].ToInt())
                });
                //群组信息
                var group = ds.Tables[3].Rows.Cast<DataRow>().Select(x => new GroupEntity
                {
                    id = x["id"].ToInt(),
                    groupname = x["name"].ToString(),
                    avatar = x["avatar"].ToString(),
                    groupdesc = x["groupdesc"].ToString()
                });
                //用户皮肤，第一个是默认正在使用的
                List<string> skin = ds.Tables[4].Rows.Cast<DataRow>().Select(x => x[0].ToString()).ToList();

                BaseListResult result = new BaseListResult
                {
                    mine = new UserEntity
                    {
                        id = rowMine["id"].ToInt(),
                        avatar = rowMine["avatar"].ToString(),
                        sign = rowMine["sign"].ToString(),
                        username = rowMine["nickname"].ToString(),
                        status = "online"
                    },
                    friend = friend,
                    group = group,
                    skin = skin
                };
                return result;
            }
            return null;
        }

　　下面，我们改用Dapper试试。重构过程先不谈，看一下重构后的代码处理：

  public BaseListResult Handle(SqlMapper.GridReader reader)
        {
            var result = new BaseListResult();
            //用户本人数据
            result.mine = reader.ReadFirst<UserEntity>();
            //处理friend逻辑 start
            var friend = reader.Read<FriendGroupEntity>();
            var groupUsers = reader.Read<GroupUserEntity>();
            friend.ToList().ForEach(f =>
            {
                //每一组的人分配给各个组
                f.list = groupUsers?.Where(x => x.groupid == f.id);
            });
            result.friend = friend;
            //处理friend逻辑 end
            //读取用户所在群
            result.group = reader.Read<GroupEntity>();
            return result;
        }

　　Dapper相较于ADO.NET比起来，就清爽多了，首先，处理GridReader，然后直接调用Read<T>方法，直接将表值转换为Model，不过Model的值需要对应。然后转换完之后，在进行一步逻辑处理，就是将相应的用户跟好友分组的id对应上。最后一个返回，这样看起来就清爽了许多，也不用处理的很麻烦。下文记录一下我的开发思路。

代码思路

　　首先要解决一个问题就是，解耦的问题，Dapper中有一个方法就是 QueryMultiple ，他是返回一个GridReader 对象，然后进行处理。那么这个GridReader对象又必须在连接开着的时候使用，所以，不能直接返回然后关闭对象，所以，可以采用接口的形式，将处理方法提出去，或者，我刚才恰好想到的就是用Func<SqlMapper.GridReader,TResult> 方式来处理结果。我的代码如下：

 public static T QueryMultiple<T>(string sql, object param,CommandType commandType = CommandType.Text, IMultipleHandler<T> handler=null)
        {
            using (var connection = getConnection())
            {
                using (var multi = connection.QueryMultiple(sql, param,commandType: commandType))
                {
                    if (handler == null) {
                        return default(T);
                    }
                    return handler.Handle(multi);
                }
            }
        }

　　其中的IMultipleHandler 中就一个方法，Handle方法，参数为GridReader。然后返回 T类型的结果。所以上文中，我的调用方式很简单了，就是传入相应的处理类就可以了。

 public class UserBaseListHandler : IMultipleHandler<BaseListResult>

　　其实我刚才突然想到，或许用Func更方便。不用再定义类继承接口了。此篇，Over，重构代码确实是有意思的事情，后续我会将各种体会做总结，如果能和读者产生共鸣那是最好不过的了。