diff --git a/controller/blog.go b/controller/blog.go index bb783b5..6c3f6f2 100644 --- a/controller/blog.go +++ b/controller/blog.go @@ -5,6 +5,8 @@ import ( "background/db" "background/logs" "background/model" + "time" + "fmt" "strconv" "strings" @@ -61,7 +63,12 @@ func InitDocCache() { } for _, v := range articles { if v.Father != 0 { - gDocMapCache[int(v.Father)].Children = append(gDocMapCache[int(v.Father)].Children, gDocMapCache[int(v.ID)]) + if gDocMapCache[int(v.Father)] != nil { + gDocMapCache[int(v.Father)].Children = append(gDocMapCache[int(v.Father)].Children, gDocMapCache[int(v.ID)]) + + } else { + log.Print(v.Father) + } } } @@ -109,6 +116,7 @@ func GetArticlesTree(c *gin.Context) { rsp.Msg = "OK" rsp.Status = 0 } + func GetArticles(c *gin.Context) { type ReqArticles struct { Name string `json:"name"` @@ -495,46 +503,6 @@ func SearchArticleES(c *gin.Context) { rsp.Status = 0 } -func AddArticle(c *gin.Context) { - rsp := RespBase{Msg: "FAIL", Status: 210} - type ReqAddArticle struct { - ID int64 `json:"id"` - Title string `json:"title"` - Content string `json:"content"` - Author string `json:"author"` - Type int64 `json:"type"` - Ispublic int `json:"is_public"` - } - var req ReqAddArticle - defer func() { - c.JSON(200, rsp) - }() - er := c.BindJSON(&req) - if nil != er { - logs.Error(er.Error()) - return - } - if req.Title == "" { - rsp.Msg = "title required" - return - } - e := model.CreateDoc( - model.Doc{ - Type: int32(req.Type), - Title: req.Title, - Content: req.Content, - Author: req.Author, - IsPublic: int32(req.Ispublic), - }, - ) - if nil != e { - logs.Error(e.Error()) - return - } - rsp.Msg = "OK" - rsp.Status = 0 -} - func AddArticleTree(c *gin.Context) { rsp := RespBase{Msg: "FAIL", Status: 210} type ReqAddArticle struct { @@ -560,7 +528,7 @@ func AddArticleTree(c *gin.Context) { return } log.Print(req.Level) - + // req.Content = strings.Replace(req.Content, "\\", "\\\\", -1) e := model.CreateDocTree( model.DocTree{ Type: int32(req.Type), @@ -576,13 +544,71 @@ func AddArticleTree(c *gin.Context) { logs.Error(e.Error()) return } + + e = model.InsertCrawDocToElaticSearch(model.DocTree{ + Type: int32(req.Type), + Title: req.Title, + Content: req.Content, + Author: req.Author, + IsPublic: int32(req.Ispublic), + Father: int32(req.Father), + Level: int32(req.Level), + CreateTime: time.Now().Format("2006-01-02"), + UpdateTime: time.Now(), + }) + if nil != e { + log.Print(e.Error()) + } + config.RedisOne().Del("artilcetree") + rsp.Msg = "OK" + rsp.Status = 0 + rsp.Data = model.GetIdFromTitle(req.Title) +} + +func AddKnowledge(c *gin.Context) { + rsp := RespBase{Msg: "FAIL", Status: 210} + type ReqAddArticle struct { + Title string `json:"title"` + Content string `json:"content"` + Author string `json:"author"` + Type int64 `json:"type"` + Ispublic int `json:"is_public"` + Father int `json:"father"` + Level int `json:"level"` + } + var req ReqAddArticle + defer func() { + c.JSON(200, rsp) + }() + er := c.BindJSON(&req) + if nil != er { + logs.Error(er.Error()) + return + } + if req.Title == "" { + rsp.Msg = "title required" + return + } + log.Print(req.Level) + + e := model.CreateCrawDoc( + model.DocTree{ + Type: int32(req.Type), + Title: req.Title, + Content: req.Content, + Author: req.Author, + IsPublic: int32(req.Ispublic), + Father: int32(req.Father), + Level: int32(req.Level), + }, + ) + if nil != e { + logs.Error(e.Error()) + return + } rsp.Msg = "OK" rsp.Status = 0 - rsp.Data = model.GetIdFromTitle(req.Title) - config.RedisOne().Del("artilcetree") - } - func ArticlesType(c *gin.Context) { rsp := RespBase{Msg: "FAIL", Status: 210} defer func() { @@ -962,6 +988,7 @@ func GetDocTypeGroup(c *gin.Context) { rsp.Status = 0 rsp.Msg = "OK" } + func GetDoGroupcType(c *gin.Context) { type DocType struct { Id int32 `json:"id"` diff --git a/controller/file.go b/controller/file.go index 556880d..dcd2d4c 100644 --- a/controller/file.go +++ b/controller/file.go @@ -7,7 +7,9 @@ import ( "fmt" "image" "image/color" + "image/gif" "image/jpeg" + "io" "io/ioutil" "log" "math" @@ -24,8 +26,12 @@ import ( // "github.com/nfnt/resize" // "image/jpeg" - "image/gif" // "strings" + + "bufio" + _ "image/gif" + _ "image/jpeg" + _ "image/png" ) type FileController struct { @@ -68,6 +74,49 @@ func (this *FileController) OnUploadDoc(c *gin.Context) { } +func DownloadImg(imgUrl, filePath string) { + + res, err := http.Get(imgUrl) + if err != nil { + fmt.Println("A error occurred!") + return + } + defer res.Body.Close() + // 获得get请求响应的reader对象 + reader := bufio.NewReaderSize(res.Body, 32*1024) + + file, err := os.Create(filePath) + if err != nil { + panic(err) + } + // 获得文件的writer对象 + writer := bufio.NewWriter(file) + written, _ := io.Copy(writer, reader) + fmt.Printf("Total length: %d", written) + file.Close() +} + +func (this *FileController) OnUpaloadPic(c *gin.Context) { + uid, e := uuid.NewV1() + if nil != e { + log.Print(e.Error()) + return + } + type Url struct { + Url string `json:"url"` + } + var url Url + + e = c.BindJSON(&url) + if nil != e { + log.Print(e.Error()) + return + } + DownloadImg(url.Url, "image/"+uid.String()+".png") + + c.JSON(200, map[string]interface{}{"url": uid.String() + ".png"}) +} + func (this *FileController) OnUpload(c *gin.Context) { uid, e := uuid.NewV1() if nil != e { diff --git a/db/elasticEngine.go b/db/elasticEngine.go index d57fa79..c324e25 100644 --- a/db/elasticEngine.go +++ b/db/elasticEngine.go @@ -26,7 +26,7 @@ type ElkEngine struct { func (p *ElkEngine) Create(index string, types string, id string, data interface{}) error { if nil != p { if (reflect.TypeOf(data).Kind() == reflect.String) || (reflect.TypeOf(data).Kind() == reflect.Struct) { - resp, err := p.cli.Index(). + _, err := p.cli.Index(). Index(index). BodyJson(data). Do(context.Background()) @@ -34,7 +34,6 @@ func (p *ElkEngine) Create(index string, types string, id string, data interface log.Print("create error", err) return err } - log.Print(resp) } else { log.Print(reflect.TypeOf(data).Kind()) return errors.New(INPUT_TYPE_ERROR) @@ -162,7 +161,6 @@ func (p *ElkEngine) Query(index string, query elastic.Query, v interface{}, print(err) return nil, err } - id := []string{} for _, vs := range res.Hits.Hits { id = append(id, vs.Id) diff --git a/db/sqlManager.go b/db/sqlManager.go index 1ca52e1..60f55d5 100644 --- a/db/sqlManager.go +++ b/db/sqlManager.go @@ -121,14 +121,13 @@ func InitELK() { elastic.SetBasicAuth(elkconf.User, elkconf.Password), ) - log.Print("init elk cli is ",gElkEngine.cli,elkconf) + log.Print("init elk cli is ", gElkEngine.cli, elkconf) if nil != e { logs.Error(e.Error()) gElkEngine.cli = nil } } - func initMysql(mysqlconf *config.MysqlConfig) *sql.DB { cnn := fmt.Sprintf("%s:%s@tcp(%s:3306)/%s?charset=utf8", mysqlconf.UserName, mysqlconf.Password, mysqlconf.Addr, mysqlconf.Db) diff --git a/main.go b/main.go index 17998a5..81b8dc7 100644 --- a/main.go +++ b/main.go @@ -39,7 +39,6 @@ func CORSMiddleware(c *gin.Context) { c.Writer.Header().Set("Access-Control-Allow-Origin", "https://testingcloud.club") } else { c.Writer.Header().Set("Access-Control-Allow-Origin", "https://testingcloud.club") - } c.Writer.Header().Set("Access-Control-Max-Age", "86400") c.Writer.Header().Set("Access-Control-Allow-Headers", @@ -163,10 +162,8 @@ func main() { api.POST("/articles", controller.GetArticles) // 获取所有文章 api.GET("/articles_tree", controller.GetArticlesTree) // 获取所有文章 - api.PUT("/article", controller.AddArticle) // 添加文章 - api.PUT("/article_file", controller.AddArticle) // 添加文章 - api.PUT("/article_tree", controller.AddArticleTree) // 添加文章 + api.PUT("/knowledge", controller.AddKnowledge) // 添加文章 api.PUT("/article_tree_history", controller.AddArticleHistory) // 添加文章 api.PUT("/article_search", controller.SearchArticles) // 添加文章 @@ -184,6 +181,7 @@ func main() { api.DELETE("/article_tree/:id", controller.DeleteArticleTree) ////删除文章 api.GET("/article_history", controller.GetArticleHistory) // 历史 + api.POST("/download_pic", fileController.OnUpaloadPic) // 获取类所在的组 api.POST("/image_upload", fileController.OnUpload) // 上传图片，如果图片太大自动裁减为原来的一半 api.POST("/image_upload_origin", fileController.OnUploadOrigin) // 上传图片 api.GET("/image_download/:file", fileController.OnDownLoad) // 下载图片 diff --git a/model/blog.go b/model/blog.go index a04fdc4..10d2cb1 100644 --- a/model/blog.go +++ b/model/blog.go @@ -11,6 +11,25 @@ import ( "qiniupkg.com/x/log.v7" ) +type CrawDoc struct { + Id int32 `json:"id" gorm:"column:id" sql:"id"` + Title string `json:"title" gorm:"column:title" sql:"title"` + Type int32 `json:"type" gorm:"column:type" sql:"type"` + Content []byte `json:"content" gorm:"column:content" sql:"content"` + Author string `json:"author" gorm:"column:author" sql:"author"` + CreateTime time.Time `json:"create_time" gorm:"column:create_time" sql:"create_time"` + UpdateTime time.Time `json:"update_time" gorm:"column:update_time" sql:"update_time"` + DeleteTime time.Time `json:"delete_time" gorm:"column:delete_time" sql:"delete_time"` + Version int32 `json:"version" gorm:"column:version" sql:"version"` + IsPublic int32 `json:"is_public" gorm:"column:is_public" sql:"is_public"` + Deleted int32 `json:"deleted" gorm:"column:deleted" sql:"deleted"` + OriginUrl string `json:"origin_url" gorm:"column:origin_url" sql:"origin_url"` + EsId string `json:"es_id" gorm:"column:es_id" sql:"es_id"` + DeletedAt time.Time `json:"deleted_at" gorm:"column:deleted_at" sql:"deleted_at"` + Father int32 `json:"father" gorm:"column:father" sql:"father"` + Level int32 `json:"level" gorm:"column:level" sql:"level"` +} + type Doc struct { ID int64 `json:"id" gorm:"column:id" sql:"id"` Title string `json:"title" gorm:"column:title" sql:"title"` @@ -36,7 +55,7 @@ type DocTree struct { Type int32 `json:"type" gorm:"column:type" sql:"type"` Content string `json:"content" gorm:"column:content" sql:"content"` Author string `json:"author" gorm:"column:author" sql:"author"` - CreateTime time.Time `json:"create_time" gorm:"column:create_time" sql:"create_time"` + CreateTime string `json:"create_time" gorm:"column:create_time" sql:"create_time"` UpdateTime time.Time `json:"update_time" gorm:"column:update_time" sql:"update_time"` DeleteTime time.Time `json:"delete_time" gorm:"column:delete_time" sql:"delete_time"` Version int32 `json:"version" gorm:"column:version" sql:"version"` @@ -54,6 +73,11 @@ func (DocTree) TableName() string { return "doc_copy1" } +// 设置User的表名为`profiles` +func (CrawDoc) TableName() string { + return "craw_doc" +} + type DocType struct { Id int32 `json:"id" gorm:"column:id" sql:"id"` TypeName string `json:"type_name" gorm:"column:type_name" sql:"type_name"` @@ -100,31 +124,6 @@ func GetArticlesType() []ArticleType { return ret } -/* -CreateDoc 新建文档 -*/ -func CreateDoc(doc Doc) error { - sql := fmt.Sprintf(`INSERT INTO doc ( doc.title, doc.content, doc.author, doc.type,doc.is_public,doc.create_time) SELECT - '%s', - '%s', - '%s', - %d , - %d, - '%s' - FROM - DUAL - WHERE - NOT EXISTS ( SELECT * FROM doc WHERE doc.title = '%s' );`, doc.Title, strings.Replace(doc.Content, "'", "\\'", -1), - doc.Author, doc.Type, doc.IsPublic, time.Now().Format("2006-01-02 15:04:05"), doc.Title) - _, e := db.GetMysqlClient().Query(sql) - if nil != e { - log.Print(sql) - logs.Error(e.Error()) - return e - } - return nil -} - /* CreateDoc 新建文档 */ @@ -154,6 +153,34 @@ func CreateDocTree(doc DocTree) error { return nil } +/* +CreateDoc 新建文档 +*/ +func CreateCrawDoc(doc DocTree) error { + sql := fmt.Sprintf(`INSERT INTO craw_doc ( craw_doc.title, craw_doc.content, + craw_doc.author, craw_doc.type,craw_doc.is_public,craw_doc.create_time,father,level) SELECT + '%s', + '%s', + '%s', + %d , + %d, + '%s', + %d, + %d + FROM + DUAL + WHERE + NOT EXISTS ( SELECT * FROM craw_doc WHERE craw_doc.title = '%s' );`, doc.Title, strings.Replace(doc.Content, "'", "\\'", -1), + doc.Author, doc.Type, doc.IsPublic, time.Now().Format("2006-01-02 15:04:05"), doc.Father, doc.Level, doc.Title) + _, e := db.GetMysqlClient().Query(sql) + log.Print(sql) + if nil != e { + log.Print(sql) + logs.Error(e.Error()) + return e + } + return nil +} func GetIdFromTitle(title string) int { motors := []DocTree{} @@ -286,9 +313,19 @@ func DeleteArticleType(id int32) error { return nil } -func GetAllDocs() ([]DocTree, error) { +func GetAllDocs(tbl string) ([]DocTree, error) { ret := []DocTree{} - sql := fmt.Sprintf("select * from doc_copy1") + sql := fmt.Sprintf("select * from " + tbl) + e := db.GetMysqlClient().Query2(sql, &ret) + if nil != e { + logs.Error(e.Error()) + return nil, e + } + return ret, nil +} +func GetAllCraw() ([]DocTree, error) { + ret := []DocTree{} + sql := fmt.Sprintf("select * from craw_doc") e := db.GetMysqlClient().Query2(sql, &ret) if nil != e { logs.Error(e.Error()) @@ -296,7 +333,6 @@ func GetAllDocs() ([]DocTree, error) { } return ret, nil } - func GetAllGroup() ([]DocGroup, error) { ret := []DocGroup{} sql := fmt.Sprintf("select * from doc_group") diff --git a/model/hardware.go b/model/hardware.go index 8f1c748..71c3ca6 100644 --- a/model/hardware.go +++ b/model/hardware.go @@ -91,6 +91,23 @@ func QueryHardwares(query elastic.Query, limit int, offset int) ([]Hardware, err return ret, nil } +func QueryDoc(query elastic.Query, limit int, offset int) ([]DocTree, error) { + ret := []DocTree{} + _, e := db.GetElastic().Query("doc", query, &ret, limit, offset) + if nil != e { + return nil, e + } + return ret, nil +} +func QueryKnowledge(query elastic.Query, limit int, offset int) ([]DocTree, error) { + ret := []DocTree{} + _, e := db.GetElastic().Query("doc", query, &ret, limit, offset) + if nil != e { + return nil, e + } + return ret, nil +} + func DeleteHardware(name string) error { query := elastic.NewTermQuery("name", name) err := db.GetElastic().Delete(query, "hardware_data") diff --git a/model/port.go b/model/port.go index fa5e409..85972ee 100644 --- a/model/port.go +++ b/model/port.go @@ -5,7 +5,6 @@ import ( "strings" "ubntgo/logger" - "github.com/go-openapi/errors" json "github.com/json-iterator/go" "gopkg.in/olivere/elastic.v3" "qiniupkg.com/x/log.v7" @@ -78,24 +77,47 @@ func QueryDocument(query elastic.Query, limit int, offset int) ([]Hardware, erro return ret, nil } func InsertDocToElaticSearch(doc DocTree) error { - matchPhraseQuery := elastic.NewMatchQuery("title", doc.Title) - existedHardware, e := QueryHardwares(matchPhraseQuery, 10, 0) - log.Print(e, existedHardware) - - for _, v := range existedHardware { - if v.Name == doc.Title { - log.Print(v.ID) - return errors.New(200, "existed title") - } - } - e = db.GetElastic().Create("doc", "0", "", doc) + matchPhraseQuery := elastic.NewTermQuery("title.keyword", doc.Title) + existedHardware, e := QueryDoc(matchPhraseQuery, 10, 0) if nil != e { log.Print(e.Error()) return e } + if len(existedHardware) == 0 { + log.Print("doc created ", doc.Title) + e = db.GetElastic().Create("doc", "0", "", doc) + if nil != e { + log.Print(e.Error()) + return e + } + return nil + } else { + log.Print("doc existed ", doc.Title) + + } return nil } +func InsertCrawDocToElaticSearch(doc DocTree) error { + matchPhraseQuery := elastic.NewTermQuery("title.keyword", doc.Title) + existedHardware, e := QueryDoc(matchPhraseQuery, 10, 0) + if nil != e { + log.Print(e.Error()) + return e + } + if len(existedHardware) == 0 { + log.Print("doc created ", doc.Title) + e = db.GetElastic().Create("doc", "0", "", doc) + if nil != e { + log.Print(e.Error()) + return e + } + return nil + } else { + log.Print("doc existed ", doc.Title) + } + return nil +} func CreateIndexFromMysqlTable(tblname string) error { columns := []Field{} e := db.GetMysqlClient().Query2("describe "+tblname, &columns) @@ -138,8 +160,8 @@ func CreateIndexFromMysqlTable(tblname string) error { return nil } -func PortDocumentToElasticsearch(tblname string) error { - ret, e := GetAllDocs() +func PortDocumentToElasticsearch(tblname string, indexname string) error { + ret, e := GetAllDocs(tblname) if nil != e { log.Print(e.Error()) } @@ -149,7 +171,7 @@ func PortDocumentToElasticsearch(tblname string) error { logger.Debug(e.Error()) return e } - if existed, _ := db.GetElastic().IndexExisted(tblname); !existed { + if existed, _ := db.GetElastic().IndexExisted(indexname); !existed { props := map[string]interface{}{} mapping := map[string]interface{}{ "settings": map[string]interface{}{ @@ -168,20 +190,36 @@ func PortDocumentToElasticsearch(tblname string) error { } for _, v := range columns { - props[v.Field] = map[string]string{"type": MysqlToElasticSearchMapping(v.Type, v.Key)} + props[v.Field] = map[string]interface{}{"type": MysqlToElasticSearchMapping(v.Type, v.Key)} + if MysqlToElasticSearchMapping(v.Type, v.Key) == "text" { + props[v.Field].(map[string]interface{})["fields"] = map[string]interface{}{ + "keyword": map[string]interface{}{ + "type": "keyword", + }, + } + } + } + dat, e := json.Marshal(mapping) if nil != e { log.Print(e.Error()) } log.Print(string(dat)) - e = db.GetElastic().CreateIndex(tblname, string(dat)) + e = db.GetElastic().CreateIndex(indexname, string(dat)) if nil != e { log.Print(e.Error()) } log.Print(string(dat)) for _, v := range ret { - e := InsertDocToElaticSearch(v) + e := InsertCrawDocToElaticSearch(v) + if nil != e { + log.Print(e.Error()) + } + } + } else { + for _, v := range ret { + e := InsertCrawDocToElaticSearch(v) if nil != e { log.Print(e.Error()) } diff --git a/test/portData_test.go b/test/portData_test.go index e02898f..7885dec 100644 --- a/test/portData_test.go +++ b/test/portData_test.go @@ -59,24 +59,22 @@ func TestPortDocToElastic(t *testing.T) { InitRedisConfig() InitMysql() db.InitELK() - e := model.PortDocumentToElasticsearch("doc_copy1") + e := model.PortDocumentToElasticsearch("doc_copy1", "doc") if nil != e { log.Print(e.Error()) } } - -func TestPortDocTreeToElastic(t *testing.T) { +func TestPortCrawToElastic(t *testing.T) { InitConfig() InitLogs() InitRedisConfig() InitMysql() db.InitELK() - e := model.PortDocumentToElasticsearch("doc_copy1") + e := model.PortDocumentToElasticsearch("craw_doc", "craw_doc") if nil != e { log.Print(e.Error()) } } - func TestReflect(t *testing.T) { type XX struct { A int16 `json:"bb"` @@ -111,12 +109,13 @@ func TestQueryDoc(t *testing.T) { InitMysql() db.InitELK() - query := elastic.NewTermQuery("title", "c") + query := elastic.NewTermQuery("title.keyword", "QT 下tableview的坑") x := []model.Doc{} _, e := db.GetElastic().Query("doc", query, &x, 10, 0) if nil != e { log.Print(e.Error()) } + t.Log(x) for _, v := range x { t.Logf(v.Title, " ", v.ID, " ", v.Type) } @@ -140,185 +139,3 @@ func TestChangeStructFieldThroughStruct(t *testing.T) { fmt.Println(cap(e0)) fmt.Println(cap(arr1)) } - -func TestPortDocVersion(t *testing.T) { - InitConfig() - InitLogs() - InitRedisConfig() - InitMysql() - - docs := []model.Doc{} - doc_groups := []model.DocGroup{} - doc_types := []model.DocType{} - e := db.GetMysqlClient().Query2("select * from doc_group", &doc_groups) - - if nil != e { - log.Print(e.Error()) - } - // for k,v := range doc_groups{ - // log.Print(k,v) - // e := db.GetMysqlClient().Query2("select * from doc_group",&doc_groups) - // if(nil != e){ - // log.Print(e.Error()) - - // } - // str := fmt.Sprintf("insert into doc_copy1 (title,type,content,author,create_time,version,is_public, deleted,origin_url,es_id,father,level) values ('%s',0,'','admin','%s', 1,1,0,'','',0,0)", - // v.Name,time.Now().Format("2006-01-02 15:04:05")) - // log.Print(str) - // _,e = db.GetMysqlClient().Query(str) - // if nil != e{ - // log.Print(e.Error()) - // } - // } - // e = db.GetMysqlClient().Query2("select * from doc_type",&doc_types) - // if(nil != e){ - // log.Print(e.Error()) - // } - // for k,v := range doc_types{ - // log.Print(k,v) - // doc_group := []model.DocGroup{} - // e = db.GetMysqlClient().Query2(fmt.Sprintf("select * from doc_group where doc_group.int = '%d'",v.Group),&doc_group) - // if nil != e{ - // log.Print(e.Error()) - // } - // log.Print(doc_group[0]) - // doc := []model.Doc{} - // e = db.GetMysqlClient().Query2(fmt.Sprintf("select * from doc_copy1 where doc_copy1.title = '%s' and doc_copy1.father=0",doc_group[0].Name),&doc) - // if nil != e{ - // log.Print(e.Error()) - // } - // log.Print(doc) - // str := fmt.Sprintf("insert into doc_copy1 (title,type,content,author,create_time,version,is_public, deleted,origin_url,es_id,father,level) values ('%s',0,'','admin','%s', 1,1,0,'','',%d,1)", - // v.TypeName,time.Now().Format("2006-01-02 15:04:05"),doc[0].ID) - // log.Print(str) - // _,e = db.GetMysqlClient().Query(str) - // if nil != e{ - // log.Print(e.Error()) - // } - - // } - - e = db.GetMysqlClient().Query2("select id,title,type from doc_copy1 where doc_copy1.father is NULL ", &docs) - if nil != e { - log.Print(e.Error()) - } - for k, v := range docs { - log.Print(k, v) - - e = db.GetMysqlClient().Query2(fmt.Sprintf("select * from doc_type where doc_type.id = %d ", v.Type), &doc_types) - if nil != e { - log.Print(e.Error()) - } - log.Print(doc_types) - if len(doc_types) > 0 { - docsfortype := []model.Doc{} - e = db.GetMysqlClient().Query2(fmt.Sprintf("select * from doc_copy1 where doc_copy1.title = '%s' and doc_copy1.level = 1", - doc_types[0].TypeName), &docsfortype) - if nil != e { - log.Print(e.Error()) - } - log.Print(docsfortype) - if len(docsfortype) > 0 { - str := fmt.Sprintf("update doc_copy1 set father = %d ,level = %d where doc_copy1.id = %d", - docsfortype[0].ID, docsfortype[0].Level+1, (v.ID)) - log.Print(str) - _, e = db.GetMysqlClient().Query(str) - if nil != e { - log.Printf(e.Error()) - } - } - } - } - -} - -func TestPortDocTree(t *testing.T) { - InitConfig() - InitLogs() - InitRedisConfig() - InitMysql() - - docs := []model.Doc{} - doc_groups := []model.DocGroup{} - doc_types := []model.DocType{} - e := db.GetMysqlClient().Query2("select * from doc_group", &doc_groups) - - if nil != e { - log.Print(e.Error()) - } - // for k,v := range doc_groups{ - // log.Print(k,v) - // e := db.GetMysqlClient().Query2("select * from doc_group",&doc_groups) - // if(nil != e){ - // log.Print(e.Error()) - - // } - // str := fmt.Sprintf("insert into doc_copy1 (title,type,content,author,create_time,version,is_public, deleted,origin_url,es_id,father,level) values ('%s',0,'','admin','%s', 1,1,0,'','',0,0)", - // v.Name,time.Now().Format("2006-01-02 15:04:05")) - // log.Print(str) - // _,e = db.GetMysqlClient().Query(str) - // if nil != e{ - // log.Print(e.Error()) - // } - // } - // e = db.GetMysqlClient().Query2("select * from doc_type",&doc_types) - // if(nil != e){ - // log.Print(e.Error()) - // } - // for k,v := range doc_types{ - // log.Print(k,v) - // doc_group := []model.DocGroup{} - // e = db.GetMysqlClient().Query2(fmt.Sprintf("select * from doc_group where doc_group.int = '%d'",v.Group),&doc_group) - // if nil != e{ - // log.Print(e.Error()) - // } - // log.Print(doc_group[0]) - // doc := []model.Doc{} - // e = db.GetMysqlClient().Query2(fmt.Sprintf("select * from doc_copy1 where doc_copy1.title = '%s' and doc_copy1.father=0",doc_group[0].Name),&doc) - // if nil != e{ - // log.Print(e.Error()) - // } - // log.Print(doc) - // str := fmt.Sprintf("insert into doc_copy1 (title,type,content,author,create_time,version,is_public, deleted,origin_url,es_id,father,level) values ('%s',0,'','admin','%s', 1,1,0,'','',%d,1)", - // v.TypeName,time.Now().Format("2006-01-02 15:04:05"),doc[0].ID) - // log.Print(str) - // _,e = db.GetMysqlClient().Query(str) - // if nil != e{ - // log.Print(e.Error()) - // } - - // } - - e = db.GetMysqlClient().Query2("select id,title,type from doc_copy1 where doc_copy1.father is NULL ", &docs) - if nil != e { - log.Print(e.Error()) - } - for k, v := range docs { - log.Print(k, v) - - e = db.GetMysqlClient().Query2(fmt.Sprintf("select * from doc_type where doc_type.id = %d ", v.Type), &doc_types) - if nil != e { - log.Print(e.Error()) - } - log.Print(doc_types) - if len(doc_types) > 0 { - docsfortype := []model.Doc{} - e = db.GetMysqlClient().Query2(fmt.Sprintf("select * from doc_copy1 where doc_copy1.title = '%s' and doc_copy1.level = 1", - doc_types[0].TypeName), &docsfortype) - if nil != e { - log.Print(e.Error()) - } - log.Print(docsfortype) - if len(docsfortype) > 0 { - str := fmt.Sprintf("update doc_copy1 set father = %d ,level = %d where doc_copy1.id = %d", - docsfortype[0].ID, docsfortype[0].Level+1, (v.ID)) - log.Print(str) - _, e = db.GetMysqlClient().Query(str) - if nil != e { - log.Printf(e.Error()) - } - } - } - } - -}