整合es

Author: lidehua

.gitignore

@@ -0,0 +1,19 @@
+# Created by .ignore support plugin (hsz.mobi)
+### Go template
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+
+.idea/

api/syn.go

@@ -0,0 +1,30 @@
+package api
+
+import (
+	"github.com/gin-gonic/gin"
+	"my/models"
+	"net/http"
+)
+
+func Syn(c *gin.Context) {
+	err := models.BookMysqlToEs()
+	error := models.SectionMysqlToEs()
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"code": "500",
+			"msg":  err,
+		})
+		return
+	}
+	if error != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"code": "500",
+			"msg":  error,
+		})
+		return
+	}
+	c.JSON(http.StatusOK, gin.H{
+		"code": "200",
+		"msg":  "操作成功",
+	})
+}

models/book.go

@@ -1,6 +1,7 @@
 package models
 
 import (
+	"context"
 	"fmt"
 	"github.com/jinzhu/gorm"
 	orm "my/database"
@@ -23,14 +24,18 @@ func Books() ([]*Book, error) {
 	return books, nil
 }
 
-func BookMysqlToEs() {
+func BookMysqlToEs() (err error) {
 	var books []*Book
 	error := orm.DB.Find(&books).Error
 	if error != nil && error != gorm.ErrRecordNotFound {
 		fmt.Errorf("BookMysqlToEs（） 查询返回失败")
-		return
+		return error
 	}
 	for i := range books {
-		orm.Es.Index().Index("book").Id(strconv.Itoa(books[i].Id)).BodyJson(books[i])
+		_, err := orm.Es.Index().Index("book").Id(strconv.Itoa(books[i].Id)).BodyJson(books[i]).Do(context.Background())
+		if err != nil {
+			fmt.Sprintf(" BookMysqlToEs Index err:%s", err.Error())
+		}
 	}
+	return nil
 }

models/section.go

@@ -1,6 +1,7 @@
 package models
 
 import (
+	"context"
 	"fmt"
 	"github.com/jinzhu/gorm"
 	orm "my/database"
@@ -11,7 +12,7 @@ type Section struct {
 	Id             int    `gorm:"size:20;primary_key;AUTO_INCREMENT;not null" json:"id"`
 	BookId         int    `gorm:"type:bigint(20);"column:book_id" json:"bookId"`
 	SectionTitle   string `gorm:"type:varchar(255);"column:section_title" json:"sectionTitle"`
-	SectionSeq     string `gorm:"type:varchar(255);"column:section_seq" json:"sectionSeq"`
+	SectionSeq     int    `gorm:"type:bigint(20);"column:section_seq" json:"sectionSeq"`
 	SectionContent string `gorm:"type:text;"column:section_content" json:"sectionContent"`
 }
 
@@ -24,14 +25,18 @@ func Sections() ([]*Section, error) {
 	return s, nil
 }
 
-func SectionMysqlToEs() {
+func SectionMysqlToEs() (err error) {
 	var s []*Section
 	error := orm.DB.Find(&s).Error
 	if error != nil && error != gorm.ErrRecordNotFound {
 		fmt.Errorf("SectionMysqlToEs（） 查询返回失败")
-		return
+		return error
 	}
 	for i := range s {
-		orm.Es.Index().Index("section").Id(strconv.Itoa(s[i].Id)).BodyJson(s[i])
+		_, err := orm.Es.Index().Index("section").Id(strconv.Itoa(s[i].Id)).BodyJson(s[i]).Do(context.Background())
+		if err != nil {
+			fmt.Sprintf(" SectionMysqlToEs Index err:%s", err.Error())
+		}
 	}
+	return nil
 }

router/router.go

@@ -12,5 +12,6 @@ func InnitRouter() *gin.Engine {
 	Engine.GET("/book/section/:num", BookSectionByNum)
 	Engine.GET("/section/list", Books)
 	Engine.GET("/section/{}", Books)
+	Engine.GET("/syn", Syn)
 	return Engine
 }