Ejemplo n.º 1
0
        public static IApplicationBuilder UserIdentity(this IApplicationBuilder builder)
        {
            return(builder.Use(async(httpContext, next) =>
            {
                int userid; DateTime time;

                string token = httpContext.Request.Cookies["token"] ?? "";
                try
                {
                    if (UserUtility.GetIdByToken(token, out userid, out time))
                    {
                        UserInfo user = null;
                        IMemoryCache cache = httpContext.RequestServices.GetService(typeof(IMemoryCache)) as IMemoryCache;

                        user = cache.GetUser(userid);
                        if (user == null)
                        {
                            UserInfoRepository repository = httpContext.RequestServices.GetService(typeof(UserInfoRepository)) as UserInfoRepository;
                            if (repository != null)
                            {
                                user = repository.FindFirst(n => n.Id == userid);
                                cache.SetUser(user);
                            }
                        }
                        if (user != null)
                        {
                            httpContext.Items.Add("user", user);
                        }
                    }
                }
                catch (Exception ex)
                {
                    Console.WriteLine(ex.Message + ex.StackTrace);
                }

                await next.Invoke();
            }));
        }