atom feed 订阅

studygolang · polaris1119 · Oct 13, 2017 · Oct 5, 2017 · Oct 5, 2017 · Oct 5, 2017
commit 792f5e0a65a7abf2cec2ee9c1acf84f3e56b28b1
diff --git a/src/http/controller/feed.go b/src/http/controller/feed.go
@@ -7,6 +7,7 @@
 package controller
 
 import (
+	"fmt"
 	"logic"
 	"model"
 	"net/http"
@@ -22,14 +23,20 @@ type FeedController struct{}
 
 // 注册路由
 func (self FeedController) RegisterRoute(g *echo.Group) {
+	g.Get("/feed.html", self.Atom)
 	g.Get("/feed.xml", self.List)
 }
 
+func (self FeedController) Atom(ctx echo.Context) error {
+	return Render(ctx, "atom.html", map[string]interface{}{})
+}
+
 func (self FeedController) List(ctx echo.Context) error {
-	link := "http://" + logic.WebsiteSetting.Domain
-	isHttps := CheckIsHttps(ctx)
-	if isHttps {
-		link = "https://" + logic.WebsiteSetting.Domain
+	link := logic.WebsiteSetting.Domain
+	if logic.WebsiteSetting.OnlyHttps {
+		link = "https://" + link
+	} else {
+		link = "http://" + link
 	}
 
 	now := time.Now()
@@ -43,32 +50,37 @@ func (self FeedController) List(ctx echo.Context) error {
 		Updated:     now,
 	}
 
-	siteFeeds := logic.DefaultFeed.FindRecent(ctx, 50)
+	respBody, err := logic.DefaultSearcher.FindAtomFeeds(50)
+	if err != nil {
+		return err
+	}
+
+	feed.Items = make([]*feeds.Item, len(respBody.Docs))
 
-	feed.Items = make([]*feeds.Item, len(siteFeeds))
+	for i, doc := range respBody.Docs {
+		url := ""
 
-	for i, siteFeed := range siteFeeds {
-		strObjtype := ""
-		switch siteFeed.Objtype {
+		switch doc.Objtype {
 		case model.TypeTopic:
-			strObjtype = "主题"
-		case model.TypeResource:
-			strObjtype = "资源"
+			url = fmt.Sprintf("%s/topics/%d", link, doc.Objid)
 		case model.TypeArticle:
-			strObjtype = "文章"
+			url = fmt.Sprintf("%s/articles/%d", link, doc.Objid)
+		case model.TypeResource:
+			url = fmt.Sprintf("%s/resources/%d", link, doc.Objid)
 		case model.TypeProject:
-			strObjtype = "开源项目"
-		case model.TypeBook:
-			strObjtype = "图书"
+			url = fmt.Sprintf("%s/p/%d", link, doc.Objid)
 		case model.TypeWiki:
-			strObjtype = "Wiki"
+			url = fmt.Sprintf("%s/wiki/%d", link, doc.Objid)
+		case model.TypeBook:
+			url = fmt.Sprintf("%s/book/%d", link, doc.Objid)
 		}
 		feed.Items[i] = &feeds.Item{
-			Title:       siteFeed.Title,
-			Link:        &feeds.Link{Href: link + siteFeed.Uri},
-			Description: "这是" + strObjtype,
-			Created:     time.Time(siteFeed.CreatedAt),
-			Updated:     time.Time(siteFeed.UpdatedAt),
+			Title:       doc.Title,
+			Link:        &feeds.Link{Href: url},
+			Author:      &feeds.Author{Name: doc.Author},
+			Description: doc.Content,
+			Created:     time.Time(doc.SortTime),
+			Updated:     time.Time(doc.UpdatedAt),
 		}
 	}
 

diff --git a/src/logic/searcher.go b/src/logic/searcher.go
@@ -417,6 +417,39 @@ func (this *SearcherLogic) SearchByField(field, value string, start, rows int, s
 	return searchResponse.RespBody, nil
 }
 
+func (this *SearcherLogic) FindAtomFeeds(rows int) (*model.ResponseBody, error) {
+	selectUrl := this.engineUrl + "/select?"
+
+	var values = url.Values{
+		"q":     []string{"*:*"},
+		"sort":  []string{"sort_time desc"},
+		"wt":    []string{"json"},
+		"start": []string{"0"},
+		"rows":  []string{strconv.Itoa(rows)},
+	}
+
+	resp, err := http.Get(selectUrl + values.Encode())
+	if err != nil {
+		logger.Errorln("search error:", err)
+		return &model.ResponseBody{}, err
+	}
+
+	defer resp.Body.Close()
+
+	var searchResponse model.SearchResponse
+	err = json.NewDecoder(resp.Body).Decode(&searchResponse)
+	if err != nil {
+		logger.Errorln("parse response error:", err)
+		return &model.ResponseBody{}, err
+	}
+
+	if searchResponse.RespBody == nil {
+		searchResponse.RespBody = &model.ResponseBody{}
+	}
+
+	return searchResponse.RespBody, nil
+}
+
 func (this *SearcherLogic) FillNodeAndUser(ctx context.Context, respBody *model.ResponseBody) (map[int]*model.User, map[int]*model.TopicNode) {
 	if respBody.NumFound == 0 {
 		return nil, nil

diff --git a/template/atom.html b/template/atom.html
@@ -0,0 +1,30 @@
+{{define "title"}}Feed订阅 {{end}}
+{{define "content"}}
+<div class="row">
+    <div class="col-md-9 col-sm-6">
+        <div class="sep20"></div>
+        <div>
+            <ol class="breadcrumb">
+                <li><a href="/">首页</a></li>
+                <li class="active">Feed订阅</li>
+            </ol>
+        </div>
+
+        <div class="page box_white">
+            <div class="cell" style="font-size: 1.2em;">
+                <p>欢迎订阅下面的Feed，您可以及时跟踪我的更新：</p>
+                <p><img src="https://static.studygolang.com/171006/a7c3d8e5e57aab26c01d965f3bb47870.png" alt=""></p>
+
+                <ul>
+                    <li><a href="/feed.xml">{{if .is_https}}https://{{else}}http://{{end}}{{.setting.Domain}}/feed.xml</a></li>
+                </ul>
+            </div>
+        </div>
+    </div>
+    <div class="col-md-3 col-sm-6">
+        <div class="sep20"></div>
+
+        {{include "common/my_info.html" .}}
+    </div>
+</div>
+{{end}}