src/ZKEACMS.SpiderLog/Service/SpiderLogData.cs

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

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

using Microsoft.Extensions.Logging;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using ZKEACMS.SpiderLog.Models;
using ZKEACMS.Storage;

namespace ZKEACMS.SpiderLog.Service
{
    public class SpiderLogData : PluginData<SpiderLogPlug>, ISpiderLogData
    {
        public SpiderLogData(ILogger<SpiderLogPlug> logger) : base(logger)
        {
        }

        public void WriteLog(SearchEngineVisitLog visitLog)
        {
            var collection = GetCollection<SearchEngineVisitLog>(visitLog.Name);
            collection.EnsureIndex(m => m.Url);
            collection.EnsureIndex(m => m.Host);
            collection.EnsureIndex(m => m.VisitAt);
            collection.Insert(visitLog);            
        }
        public SearchEngineVisitLog GetLastVisit(string host, string name)
        {
            var lastVisit = GetCollection<SearchEngineVisitLog>(name).Query()
                .Where(m => m.Host == host)
                .OrderByDescending(m => m.VisitAt)
                .Limit(1)
                .ToArray();

            if (lastVisit.Length > 0) return lastVisit[0];

            return null;
        }
        public IEnumerable<SearchEngineVisitLog> GetAll(string host, string name)
        {
            return GetCollection<SearchEngineVisitLog>(name)
                .Query()
                .Where(m => m.Host == host)
                .OrderByDescending(m => m.VisitAt)
                .Limit(1000)
                .ToEnumerable()
                .ToArray();
        }
    }
}