cc-officialweb/controllers/contact.go

package controllers

import (
	"cc-officialweb/models"
	"cc-officialweb/unity"
	"encoding/json"
	beego "github.com/beego/beego/v2/server/web"
	"github.com/go-playground/validator/v10"
	"strconv"
)

type ContactController struct {
	beego.Controller
}

func (c *ContactController) Get() {
	c.Data["IsContact"] = true
	c.TplName = "contact.html"
}

// AddContact 提交联系我们信息
func (c *ContactController) AddContact() {
	var contact models.Contact
	name := c.GetString("name")
	phone := c.GetString("phone")
	message := c.GetString("message")
	validate := validator.New()
	err2 := validate.Var(name, "required")
	err2 = validate.Var(phone, "required")
	err2 = validate.Var(message, "required")
	if err2 != nil {
		c.Data["json"] = &JSON{Code: 102, Msg: "添加失败"}
		c.ServeJSON()
		return
	}
	contact.Name = name
	contact.Phone = phone
	contact.Message = message
	add, err := unity.Add(&contact)
	if err != nil {
		c.Data["json"] = &JSON{Code: 102, Msg: "添加失败"}
		c.ServeJSON()
		return
	} else {
		c.Data["json"] = &JSON{Code: 200, Msg: "添加成功", Data: add}
		c.ServeJSON()
	}
}

// DeleteContactById 根据id删除联系消息
func (c *ContactController) DeleteContactById() {
	id := c.GetString("id")
	atoi, _ := strconv.Atoi(id)
	validate := validator.New()
	err2 := validate.Var(atoi, "required")
	if err2 != nil {
		c.Data["json"] = &JSON{Code: 103, Msg: "id不能为空"}
		c.ServeJSON()
		return
	}
	_, err := unity.DeleteById(atoi, models.Contact{})
	if err != nil {
		c.Data["json"] = map[string]interface{}{"code": 101, "msg": "删除失败"}
		c.ServeJSON()
		return
	} else {
		c.Data["json"] = map[string]interface{}{"code": 200, "msg": "删除成功"}
		c.ServeJSON()
	}
}

// GetAllContact 获取所有联系方式
func (c *ContactController) GetAllContact() {
	var page unity.PageParams
	err := json.Unmarshal(c.Ctx.Input.RequestBody, &page)
	if err != nil {
		c.Data["json"] = &JSON{Code: 101, Msg: "参数错误"}
		c.ServeJSON()
		return
	}
	result, total, current, err := unity.Paginate(page, models.Contact{})
	if err != nil {
		c.Data["json"] = &JSON{Code: 101, Msg: "查询失败"}
		c.ServeJSON()
		return
	} else {
		c.Data["json"] = &JSON{Code: 200, Msg: "查询成功", Data: &JSONS{
			Current: current,
			Size:    total,
			Data:    result,
		}}
		c.ServeJSON()
		return
	}
}