frist commit

.gitignore

@@ -0,0 +1,2 @@
+*.exe
+*.exe~

conf/app.conf

@@ -0,0 +1,18 @@
+appname = h5-2019
+httpport = 8080
+runmode = dev
+autorender = false
+copyrequestbody = true
+EnableDocs = true
+sessionon = true
+
+
+[wechat]
+appid = wx776f26b70b03902b
+secret = 36ad3e2017a77cb4005770bfa405d7b1
+wxid = gh_1ec36e12f0a9
+token = yansenisashit
+aeskey =
+
+# 最后要加 /
+service = http://localhost:9000/

conf/db.conf

@@ -0,0 +1,29 @@
+; 数据库类型，目前只支持mysql
+db_type      = mysql
+
+; 连接协议
+con_protocol = tcp
+
+; 数据库地址，可以使用IP
+db_addr      = rm-uf6z3z6jq11x653d77o.mysql.rds.aliyuncs.com
+
+; 端口
+db_port      = 3306
+
+; 用户名
+username     = h5-2019
+
+; 密码
+password     = 8YTxJn2zgljIBikv
+
+; 数据库名或者schema
+database     = h5-2019
+
+; 前缀，目前尚未使用
+dbprefix     = 
+
+; 模式，目前尚未使用
+db_debug     = false
+
+; 字符集
+char_set     = utf8

controllers/base.go

@@ -0,0 +1,72 @@
+package controllers
+
+import (
+	"h5-2019/utils"
+	"reflect"
+
+	"github.com/astaxie/beego"
+
+	"net/http"
+)
+
+//Message api统一的返回消息格式
+type Message struct {
+	Code    int         `json:"code"`
+	Message interface{} `json:"message"`
+}
+
+//BaseController 基础controller
+type BaseController struct {
+	beego.Controller
+}
+
+const (
+	SESSION_USER = "h5.user"
+	PAGENUM      = 10
+	LISTPAGENUM  = 20
+)
+
+// BaseControllerInterface 项目约定的 Controller 须满足的接口
+type BaseControllerInterface interface {
+	Constructor()
+}
+
+//Prepare 继承beego的
+func (c *BaseController) Prepare() {
+	// 微信初始化
+	utils.WechatInit()
+
+	// 路由对应的实际 Controller 初始化
+	c.initController()
+}
+
+// ResponseJson 自定义的JSON返回
+func (c *BaseController) ResponseJson(msg interface{}, code ...int) {
+	status := http.StatusOK
+	if len(code) > 0 {
+		status = code[0]
+	}
+
+	if reflect.ValueOf(msg).Type().String() == "*errors.errorString" {
+		err := msg.(error)
+		c.Data["json"] = Message{status, err.Error()}
+	} else {
+		c.Data["json"] = Message{status, msg}
+	}
+
+	c.ServeJSON()
+	c.StopRun()
+}
+
+// initAppController 执行当前路由请求对应的 Controller 初始化
+func (c *BaseController) initController() {
+	if ctrl, ok := c.AppController.(BaseControllerInterface); ok {
+		ctrl.Constructor()
+	}
+}
+
+func (c *BaseController) WechatAppID() {
+	c.ResponseJson(map[string]interface{}{
+		"appid": beego.AppConfig.String("wechat::appid"),
+	})
+}

controllers/h5.go

@@ -0,0 +1,107 @@
+package controllers
+
+import (
+	"errors"
+	"h5-2019/models"
+	"h5-2019/models/model"
+	"h5-2019/utils"
+	"net/http"
+
+	"github.com/astaxie/beego"
+)
+
+// H5Controller 基础controller
+type H5Controller struct {
+	dao *models.H5Model
+	BaseController
+}
+
+// Constructor 初始化
+func (c *H5Controller) Constructor() {
+	c.dao = new(models.H5Model)
+}
+
+// GetCustomer 获取用户及分享信息
+func (c *H5Controller) GetCustomer() {
+	code := c.GetString("code")
+	link := c.GetString("link")
+
+	// 已经存在的用户直接从数据库拿
+	userID := 0
+	userSesn := c.GetSession(SESSION_USER)
+	if userSesn != nil {
+		userID, _ = userSesn.(int)
+	}
+
+	var err error
+	user := new(model.TaCustomer)
+	if userID > 0 {
+		user, err = c.dao.GetCustomer(userID)
+		if err != nil {
+			beego.Error("获取用户信息失败:", err.Error())
+			c.ResponseJson(errors.New("获取用户信息失败"), http.StatusInternalServerError)
+		}
+	} else {
+		// 不存在的用户，则保存入库
+		if res, err := utils.GetUserInfo(code); err != nil {
+			beego.Error("获取微信用户失败:", err.Error())
+			c.ResponseJson(errors.New("获取微信用户失败"), http.StatusInternalServerError)
+		} else if user, err = c.dao.SaveCustomer(res.(map[string]interface{})); err != nil {
+			beego.Error("校验微信用户失败:", err.Error())
+			c.ResponseJson(errors.New("校验微信用户失败"), http.StatusInternalServerError)
+		}
+	}
+
+	// 获取分享 ticket
+	res, err := utils.GetJSTicket(link)
+	if err != nil {
+		c.ResponseJson(errors.New("微信分享设置校验失败"), http.StatusInternalServerError)
+	}
+
+	c.SetSession(SESSION_USER, user.Id)
+
+	c.ResponseJson(map[string]interface{}{
+		"user":  user,
+		"share": res,
+	})
+}
+
+// PutShare 更新分享记录
+func (c *H5Controller) PutShare() {
+	userSesn := c.GetSession(SESSION_USER)
+	if userSesn == nil {
+		c.ResponseJson(errors.New("记录分享失败, 请重新获取用户信息"), http.StatusUnauthorized)
+	}
+
+	userID, _ := userSesn.(int)
+	if userID == 0 {
+		c.ResponseJson(errors.New("未知错误, 请重新获取用户信息"), http.StatusUnauthorized)
+	}
+
+	if err := c.dao.SharePage(userID); err != nil {
+		beego.Error("记录分享失败:", err.Error())
+		c.ResponseJson(errors.New("记录分享失败"), http.StatusInternalServerError)
+	}
+
+	c.ResponseJson("ok")
+}
+
+// GetTotalShare 获取分享总数
+func (c *H5Controller) GetTotalShare() {
+	total, err := c.dao.TotalShareNum()
+	if err != nil {
+		beego.Error("获取分享总数失败:", err.Error())
+		c.ResponseJson(errors.New("获取分享总数失败"), http.StatusInternalServerError)
+	}
+
+	custnum, err := c.dao.TotalCustomer()
+	if err != nil {
+		beego.Error("获取人数失败:", err.Error())
+		c.ResponseJson(errors.New("获取人数失败"), http.StatusInternalServerError)
+	}
+
+	c.ResponseJson(map[string]interface{}{
+		"total":   total,
+		"custnum": custnum,
+	})
+}

main.go

@@ -0,0 +1,15 @@
+package main
+
+import (
+	_ "h5-2019/routers"
+
+	"github.com/astaxie/beego"
+)
+
+func main() {
+	if beego.BConfig.RunMode == "dev" {
+		beego.BConfig.WebConfig.DirectoryIndex = true
+		beego.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
+	}
+	beego.Run()
+}

models/h5.go

@@ -0,0 +1,97 @@
+package models
+
+import (
+	"h5-2019/models/model"
+	"time"
+
+	"github.com/astaxie/beego"
+)
+
+// H5Model model
+type H5Model struct{}
+
+// GetCustomer 获取
+func (m *H5Model) GetCustomer(id int) (*model.TaCustomer, error) {
+	cust := model.TaCustomer{}
+	_, err := Dao.Where("id=?", id).Get(&cust)
+	if err != nil {
+		return nil, err
+	}
+
+	return &cust, nil
+}
+
+// SaveCustomer 新增
+func (m *H5Model) SaveCustomer(mem map[string]interface{}) (*model.TaCustomer, error) {
+	cust := model.TaCustomer{}
+	openid := mem["openid"].(string)
+
+	if _, err := Dao.Where("openid=?", openid).Get(&cust); err != nil {
+		beego.Error("查询错误->", err)
+		return nil, err
+	}
+
+	beego.Info("获取到用户信息, ", cust)
+
+	cust.Nickyname = mem["nickname"].(string)
+	cust.Headimgurl = mem["headimgurl"].(string)
+
+	if cust.Openid != "" {
+		// 如果用户已经存在, 则更新
+		if _, err := Dao.Where("openid=?", openid).
+			Cols([]string{"nickyname", "headimgurl"}...).
+			Update(&cust); err != nil {
+			beego.Error("更新错误->", err)
+			return nil, err
+		}
+		return &cust, nil
+	}
+
+	cust.Openid = openid
+	cust.Unionid, _ = mem["unionid"].(string)
+	cust.CreateDate = time.Now().Local()
+	cust.Sharedpage = 0
+
+	_, err := Dao.Insert(&cust)
+	if err != nil {
+		return nil, err
+	}
+
+	return &cust, nil
+}
+
+// SharePage 分享
+func (m *H5Model) SharePage(id int) error {
+	sql := `
+		update ta_customer
+		set sharedpage = IFNULL(sharedpage,0) + 1
+		where id = ?
+	`
+
+	_, err := Dao.Exec(sql, id)
+	return err
+}
+
+// TotalShareNum 获取总次数
+func (m *H5Model) TotalShareNum() (int64, error) {
+	query := `select sum(IFNULL(sharedpage,0)) as total from ta_customer`
+
+	var total []int64
+	if err := Dao.SQL(query).Find(&total); err != nil {
+		return 0, err
+	}
+
+	return total[0], nil
+}
+
+// TotalCustomer 获取总人数
+func (m *H5Model) TotalCustomer() (int64, error) {
+	query := `select count(1) as total from ta_customer`
+
+	var total []int64
+	if err := Dao.SQL(query).Find(&total); err != nil {
+		return 0, err
+	}
+
+	return total[0], nil
+}

models/model/ta_customer.go

@@ -0,0 +1,13 @@
+package model
+
+import "time"
+
+type TaCustomer struct {
+	Id         int       `xorm:"not null pk autoincr INT(11)"`
+	Openid     string    `xorm:"VARCHAR(128)"`
+	Nickyname  string    `xorm:"VARCHAR(255)"`
+	Headimgurl string    `xorm:"TEXT"`
+	Unionid    string    `xorm:"VARCHAR(128)"`
+	CreateDate time.Time `xorm:"DATETIME"`
+	Sharedpage int       `xorm:"SMALLINT(6)"`
+}

models/models.go

@@ -0,0 +1,57 @@
+package models
+
+import (
+	"github.com/astaxie/beego/config"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/go-xorm/xorm"
+)
+
+var (
+	Dao *xorm.Engine
+)
+
+func init() {
+	Dao = NewDAO()
+}
+
+// NewDAO 初始化数据库连接
+func NewDAO() *xorm.Engine {
+	dbType := "mysql"
+	dns := getMysqlDns()
+
+	dao, err := xorm.NewEngine(dbType, dns)
+	// dao.ShowSQL()
+
+	if err != nil {
+		panic(err)
+		return nil
+	}
+
+	Dao = dao
+	return dao
+}
+
+func getMysqlDns() string {
+	dbconf, _ := config.NewConfig("ini", "conf/db.conf")
+
+	// db_type  := dbconf.DefaultString("db_type", "mysql")
+	conProt := dbconf.DefaultString("con_protocol", "tcp")
+	dbAddr := dbconf.DefaultString("db_addr", "localhost")
+	dbPort := dbconf.DefaultString("db_port", "3306")
+	userName := dbconf.DefaultString("username", "root")
+	password := dbconf.String("password")
+	database := dbconf.String("database")
+	// dbprefix := dbconf.String("dbprefix")
+	// db_debug := dbconf.DefaultBool("db_debug", false)
+	charSet := dbconf.DefaultString("char_set", "utf8")
+
+	dns := userName
+
+	if len(password) > 0 {
+		dns += ":" + password
+	}
+
+	dns += "@" + conProt + "(" + dbAddr + ":" + dbPort + ")" + "/" + database + "?charset=" + charSet
+
+	return dns
+}

routers/router.go

@@ -0,0 +1,24 @@
+// @APIVersion 1.0.0
+// @Title beego Test API
+// @Description beego has a very cool tools to autogenerate documents for your API
+// @Contact astaxie@gmail.com
+// @TermsOfServiceUrl http://beego.me/
+// @License Apache 2.0
+// @LicenseUrl http://www.apache.org/licenses/LICENSE-2.0.html
+package routers
+
+import (
+	"h5-2019/controllers"
+
+	"github.com/astaxie/beego"
+)
+
+func init() {
+	ns := beego.NewNamespace("/api",
+		beego.NSRouter("/appid", &controllers.BaseController{}, "get:WechatAppID"),
+		beego.NSRouter("/user", &controllers.H5Controller{}, "get:GetCustomer"),
+		beego.NSRouter("/share", &controllers.H5Controller{}, "get:GetTotalShare"),
+		beego.NSRouter("/share", &controllers.H5Controller{}, "put:PutShare"),
+	)
+	beego.AddNamespace(ns)
+}

tests/default_test.go

@@ -0,0 +1,38 @@
+package test
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"runtime"
+	"path/filepath"
+	_ "h5-2019/routers"
+
+	"github.com/astaxie/beego"
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func init() {
+	_, file, _, _ := runtime.Caller(1)
+	apppath, _ := filepath.Abs(filepath.Dir(filepath.Join(file, ".." + string(filepath.Separator))))
+	beego.TestBeegoInit(apppath)
+}
+
+// TestGet is a sample to run an endpoint test
+func TestGet(t *testing.T) {
+	r, _ := http.NewRequest("GET", "/v1/object", nil)
+	w := httptest.NewRecorder()
+	beego.BeeApp.Handlers.ServeHTTP(w, r)
+
+	beego.Trace("testing", "TestGet", "Code[%d]\n%s", w.Code, w.Body.String())
+
+	Convey("Subject: Test Station Endpoint\n", t, func() {
+	        Convey("Status Code Should Be 200", func() {
+	                So(w.Code, ShouldEqual, 200)
+	        })
+	        Convey("The Result Should Not Be Empty", func() {
+	                So(w.Body.Len(), ShouldBeGreaterThan, 0)
+	        })
+	})
+}
+

utils/wechat.go

@@ -0,0 +1,116 @@
+package utils
+
+import (
+	"encoding/json"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net/http"
+
+	"github.com/astaxie/beego"
+	"github.com/zjxpcyc/request"
+)
+
+type WxResult struct {
+	Code    int
+	Message string
+	Result  interface{}
+}
+
+var wechatInited = false
+
+// WechatInit 微信初始化
+func WechatInit() error {
+	if wechatInited {
+		return nil
+	}
+
+	wxconf, err := beego.AppConfig.GetSection("wechat")
+	if err != nil {
+		beego.Error("获取微信配置失败: ", err.Error())
+		return err
+	}
+
+	config := request.NewConfig(wxconf["service"] + "wx/register")
+	config.AddParam("appid", wxconf["appid"])
+	config.AddParam("secret", wxconf["secret"])
+	config.AddParam("token", wxconf["token"])
+	config.AddParam("aeskey", wxconf["aeskey"])
+	config.Method = http.MethodPost
+
+	_, err = Request(config)
+	if err == nil {
+		wechatInited = true
+	}
+
+	return err
+}
+
+// GetUserInfo 获取个人信息
+func GetUserInfo(code string) (interface{}, error) {
+	wxconf, err := beego.AppConfig.GetSection("wechat")
+	if err != nil {
+		beego.Error("获取微信配置失败: ", err.Error())
+		return nil, err
+	}
+
+	config := request.NewConfig(wxconf["service"] + "wx/userinfo")
+	config.AddParam("appid", wxconf["appid"])
+	config.AddParam("code", code)
+	config.Method = http.MethodGet
+
+	return Request(config)
+}
+
+// GetJSTicket 获取分享 ticket
+func GetJSTicket(link string) (interface{}, error) {
+	wxconf, err := beego.AppConfig.GetSection("wechat")
+	if err != nil {
+		beego.Error("获取微信配置失败: ", err.Error())
+		return nil, err
+	}
+
+	config := request.NewConfig(wxconf["service"] + "wx/jssdk")
+	config.AddParam("appid", wxconf["appid"])
+	config.AddParam("url", link)
+	config.Method = http.MethodGet
+
+	return Request(config)
+}
+
+func Request(config *request.Config) (interface{}, error) {
+	body, _, err := request.Do(config)
+	if err != nil {
+		beego.Error("请求微信服务失败: ", err.Error())
+		return nil, err
+	}
+
+	res, err := getResult(body)
+	if err != nil {
+		return nil, err
+	}
+
+	if res.Code != http.StatusOK {
+		return nil, fmt.Errorf("请求微信服务失败: %s", res.Message)
+	}
+
+	return res.Result.(interface{}), nil
+}
+
+func getResult(body io.ReadCloser) (*WxResult, error) {
+	defer body.Close()
+
+	bs, err := ioutil.ReadAll(body)
+	if err != nil {
+		beego.Error("读取微信服务结果失败: ", err.Error())
+		return nil, err
+	}
+
+	res := WxResult{}
+	if err := json.Unmarshal(bs, &res); err != nil {
+		beego.Error("转换微信服务结果失败: ", err.Error())
+		return nil, err
+	}
+
+	return &res, nil
+}