diff --git a/controllers/BlogController.go b/controllers/BlogController.go index 7d97cf4..a66151f 100644 --- a/controllers/BlogController.go +++ b/controllers/BlogController.go @@ -22,7 +22,7 @@ func (this *BlogController) Save() { if err == nil { this.Data["json"] = blog } else { - this.Data["json"] = models.ReurnError("保存失败") + this.Data["json"] = models.ReurnError(500,"保存失败") } this.ServeJSON() } diff --git a/controllers/NoteController.go b/controllers/NoteController.go index ed6655e..d932850 100644 --- a/controllers/NoteController.go +++ b/controllers/NoteController.go @@ -23,7 +23,30 @@ func (this *NoteController) Save() { if err == nil { this.Data["json"] = note } else { - this.Data["json"] = models.ReurnError("保存失败") + this.Data["json"] = models.ReurnError(500,"保存失败") + } + this.ServeJSON() +} +func (this *NoteController) Edit() { + idStr := this.Ctx.Input.Param(":id") + noteHtml := this.GetString("noteHtml") + id, _ := strconv.ParseInt(idStr, 10, 64) + uid := this.GetSession("userid").(int64) + note := &models.Note{Id:id} + err1 := service.GetNote(note) + if err1 != nil { + fmt.Print(err1) + this.Data["json"] = models.ReurnError(500,"保存失败") + } + if uid != note.UserId { + this.Data["json"] = models.ReurnError(403,"保存失败") + } + note.NoteHtml = noteHtml + err := service.SaveNote(note) + if err == nil { + this.Data["json"] = note + } else { + this.Data["json"] = models.ReurnError(500,"保存失败") } this.ServeJSON() } @@ -37,7 +60,7 @@ func (this *NoteController) SaveNoteColl() { if err == nil { this.Data["json"] = note } else { - this.Data["json"] = models.ReurnError("保存失败") + this.Data["json"] = models.ReurnError(500,"保存失败") } this.ServeJSON() } @@ -60,11 +83,11 @@ func (this *NoteController) Note() { this.Redirect("/login", 302) } noteColls, err := service.GetNoteColl(uid.(int64)) - if err != nil { + if err == nil { if len(noteColls) > 0 { for i := 0; i < len(noteColls); i++ { notes, err1 := service.GetNoteByPid(noteColls[i].Id) - if err1 != nil { + if err1 == nil { noteColls[i].Notes = notes } } @@ -73,6 +96,7 @@ func (this *NoteController) Note() { noteColls = make([]*models.NoteColl, 0) } fmt.Println(noteColls) + fmt.Println(len(noteColls)) this.Data["NoteColls"] = noteColls this.TplName = "note.html" } diff --git a/controllers/UserController.go b/controllers/UserController.go index 571246d..775599d 100644 --- a/controllers/UserController.go +++ b/controllers/UserController.go @@ -37,19 +37,19 @@ func (this *UserController) Login() { username := this.GetString("username") userpwd := this.GetString("userpwd") if username == "" { - this.Data["json"] = models.ReurnError("用户名为空") + this.Data["json"] = models.ReurnError(1,"用户名为空") this.ServeJSON() } if len(username) < 4 { - this.Data["json"] = models.ReurnError("用户名最低4位") + this.Data["json"] = models.ReurnError(1,"用户名最低4位") this.ServeJSON() } if userpwd == "" { - this.Data["json"] = models.ReurnError("密码为空") + this.Data["json"] = models.ReurnError(1,"密码为空") this.ServeJSON() } if len(userpwd) < 6 { - this.Data["json"] = models.ReurnError("密码最低6位") + this.Data["json"] = models.ReurnError(1,"密码最低6位") this.ServeJSON() } user, error := service.FindByUserName(username) @@ -62,10 +62,10 @@ func (this *UserController) Login() { this.SetSession("userid", user.Id) fmt.Println(this.CruSession) } else { - this.Data["json"] = models.ReurnError("用户名或密码错误") + this.Data["json"] = models.ReurnError(1,"用户名或密码错误") } } else { - this.Data["json"] = models.ReurnError("用户名不存在") + this.Data["json"] = models.ReurnError(1,"用户名不存在") } this.ServeJSON() } @@ -76,24 +76,24 @@ func (this *UserController) Regist() { username = strings.Replace(username, " ", "", -1) userpwd = strings.Replace(userpwd, " ", "", -1) if username == "" { - this.Data["json"] = models.ReurnError("用户名为空") + this.Data["json"] = models.ReurnError(1,"用户名为空") this.ServeJSON() } if len(username) < 4 { - this.Data["json"] = models.ReurnError("用户名最低4位") + this.Data["json"] = models.ReurnError(1,"用户名最低4位") this.ServeJSON() } if userpwd == "" { - this.Data["json"] = models.ReurnError("密码为空") + this.Data["json"] = models.ReurnError(1,"密码为空") this.ServeJSON() } if len(userpwd) < 6 { - this.Data["json"] = models.ReurnError("密码最低6位") + this.Data["json"] = models.ReurnError(1,"密码最低6位") this.ServeJSON() } user, _ := service.FindByUserName(username) if user != nil { - this.Data["json"] = models.ReurnError("用户已经存在") + this.Data["json"] = models.ReurnError(1,"用户已经存在") this.ServeJSON() } h := md5.New() @@ -107,7 +107,7 @@ func (this *UserController) Regist() { if err == nil { this.Data["json"] = models.ReurnSuccess("") } else { - this.Data["json"] = models.ReurnError("注册失败") + this.Data["json"] = models.ReurnError(1,"注册失败") } this.ServeJSON() } diff --git a/models/Result.go b/models/Result.go index 3c22302..7215e13 100644 --- a/models/Result.go +++ b/models/Result.go @@ -5,8 +5,11 @@ type Error struct { Msg string } -func ReurnError(msg string) *Error { - return &Error{Status: byte(1), Msg: msg} +func ReurnError(status int,msg string) *Error { + if msg == "" { + msg = "error" + } + return &Error{Status: byte(status), Msg: msg} } type Success struct { diff --git a/routers/NoteRouter.go b/routers/NoteRouter.go index 7b5385a..fd70989 100644 --- a/routers/NoteRouter.go +++ b/routers/NoteRouter.go @@ -10,4 +10,5 @@ func init() { beego.Router("/note", &controllers.NoteController{}, "get:Note") beego.Router("/notecoll/save", &controllers.NoteController{}, "post:SaveNoteColl") beego.Router("/note/save", &controllers.NoteController{}, "post:Save") + beego.Router("/note/edit/:id([0-9]+)", &controllers.NoteController{}, "post:Edit") } diff --git a/service/NoteService.go b/service/NoteService.go index 7970113..58b892b 100644 --- a/service/NoteService.go +++ b/service/NoteService.go @@ -27,10 +27,9 @@ func GetNoteByPid(pid int64) ([]*models.Note, error) { o := orm.NewOrm() qs := o.QueryTable(models.Note{}) _, err := qs.Filter("Pid", pid).All(¬es) - return notes,err + return notes, err } - func SaveNoteColl(note *models.NoteColl) error { o := orm.NewOrm() id, err := o.Insert(note) @@ -45,5 +44,5 @@ func GetNoteColl(uid int64) ([]*models.NoteColl, error) { o := orm.NewOrm() qs := o.QueryTable(models.NoteColl{}) _, err := qs.Filter("UserId", uid).All(¬es) - return notes,err -} \ No newline at end of file + return notes, err +} diff --git a/views/login.html b/views/login.html index 34e1d13..dbce38e 100644 --- a/views/login.html +++ b/views/login.html @@ -53,6 +53,7 @@ layer.msg(data.Msg, {icon: 5}); }else{ layer.msg("登录成功", {icon: 6}); + window.location.href = "/" } }, 'json') }) diff --git a/views/note.html b/views/note.html index 77ed0c3..9370ba0 100644 --- a/views/note.html +++ b/views/note.html @@ -58,25 +58,16 @@ @@ -86,6 +77,7 @@