src/ZKEACMS.Sitemap/Service/SiteUrlProviders/PageSiteUrlProvider.cs

微信扫码关注公众号，代码整理好后第一时间通知你。

/* http://www.zkea.net/ 
 * Copyright (c) ZKEASOFT. All rights reserved. 
 * http://www.zkea.net/licenses */

using Microsoft.AspNetCore.Http;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using ZKEACMS.Page;
using ZKEACMS.Sitemap.Models;
using ZKEACMS.Widget;
using Easy.Extend;
using Easy;

namespace ZKEACMS.Sitemap.Service.SiteUrlProviders
{
    public class PageSiteUrlProvider : ISiteUrlProvider
    {
        private readonly IPageService _pageService;
        private readonly SitemapDbContext _sitemapDbContext;
        private readonly IWidgetBasePartService _widgetBasePartService;
        public PageSiteUrlProvider(SitemapDbContext sitemapDbContext,
            IPageService pageService,
            IWidgetBasePartService widgetBasePartService)
        {
            _sitemapDbContext = sitemapDbContext;
            _pageService = pageService;
            _widgetBasePartService = widgetBasePartService;
        }
        public IEnumerable<SiteUrl> Get()
        {
            HashSet<string> distinct = new HashSet<string>();
            var widgetIds = _sitemapDbContext.ArticleDetailWidget.ToList().Select(m => m.ID).ToArray();
            widgetIds = widgetIds.Concat(_sitemapDbContext.ProductDetailWidget.ToList().Select(m => m.ID)).ToArray();
            var pageIds = _widgetBasePartService.Get(m => widgetIds.Contains(m.ID)).Select(m => m.PageId);
            var pages = _pageService.Get().Where(m => !pageIds.Contains(m.ID) && m.IsPublishedPage).OrderBy(m => m.ReferencePageID).ThenByDescending(m => m.PublishDate).ToList();
            foreach (var page in pages)
            {
                if (page.Url.IsNotNullAndWhiteSpace() && !distinct.Contains(page.ReferencePageID))
                {
                    distinct.Add(page.ReferencePageID);
                    yield return new SiteUrl
                    {
                        Url = Helper.Url.ToAbsolutePath(page.Url),
                        ModifyDate = page.PublishDate ?? DateTime.Now,
                        Changefreq = "daily",
                        Priority = 0.5F
                    };
                }

            }
        }
    }
}