first commit

.gitignore

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

conf/app.conf

@@ -0,0 +1,61 @@
+appname = voting
+httpport = 8080
+runmode = dev
+autorender = false
+copyrequestbody = true
+EnableDocs = true
+
+
+[wechat]
+appid = 
+secret = 
+token =
+aeskey = 
+wxid =
+
+[alioss]
+Endpoint=oss-cn-shanghai.aliyuncs.com
+AccessKeyId=LTAIkc75dpkJw8Lb
+AccessKeySecret=v4bvXCaix6vSDTCFfwSAdqV53iFEQw
+Bucket=jingcheng-h5temp
+
+[cros]
+allowMode = dev
+allowMethods = *
+allowOrigin = http://localhost:9528
+allowCredentials = true
+
+
+[db]
+; 数据库类型，目前只支持mysql
+db_type      = mysql
+
+; 连接协议
+con_protocol = tcp
+
+; 数据库地址，可以使用IP
+db_addr = localhost
+# db_addr      = rm-uf6z3z6jq11x653d77o.mysql.rds.aliyuncs.com
+
+; 端口
+db_port      = 3306
+
+; 用户名
+username     = spaceofcheng
+
+; 密码
+password     = spaceofcheng
+# password     = c;a|vK)Xv/=L@c[6Fx,v[ITD5;WpR}+Y
+
+; 数据库名或者schema
+database     = voting
+
+; 前缀，目前尚未使用
+dbprefix     = 
+
+; 模式，目前尚未使用
+db_debug     = false
+
+; 字符集
+char_set     = utf8mb4
+

controllers/base.go

@@ -0,0 +1,111 @@
+package controllers
+
+import (
+	"encoding/json"
+	"net/http"
+	"spaceofcheng/services/utils"
+
+	"github.com/astaxie/beego"
+)
+
+type ControllerInterface interface {
+	Constructor()
+}
+
+//BaseController 基础controller
+type BaseController struct {
+	beego.Controller
+}
+
+//Prepare 继承beego的
+func (c *BaseController) Prepare() {
+	c.initController()
+}
+
+// ResponseJSON 返回JSON数据
+func (c *BaseController) ResponseJSON(data interface{}) {
+	c.ResponseData(data, "", http.StatusOK)
+}
+
+// ResponseError 返回错误
+func (c *BaseController) ResponseError(err error, code ...int) {
+	if len(code) > 0 {
+		c.ResponseData(nil, err, code[0])
+	}
+
+	c.ResponseData(nil, err, http.StatusBadRequest)
+}
+
+// ResponseRaw 返回
+func (c *BaseController) ResponseRaw(msg []byte, stop ...bool) {
+	c.Ctx.ResponseWriter.Write(msg)
+
+	if stop != nil && len(stop) > 0 {
+		if !stop[0] {
+			return
+		}
+	}
+
+	c.StopRun()
+}
+
+// ResponseData 自定义的JSON返回
+func (c *BaseController) ResponseData(data interface{}, msg interface{}, code int) {
+	status := code
+
+	sendMessage := ""
+	switch msgVal := msg.(type) {
+	case error:
+		sendMessage = msgVal.Error()
+	case string:
+		sendMessage = msgVal
+	default:
+		sendMessage = ""
+	}
+
+	c.Data["json"] = map[string]interface{}{
+		"code":    status,
+		"message": sendMessage,
+		"result":  data,
+	}
+
+	c.Ctx.Output.Header("Access-Control-Expose-Headers", utils.TokenHeader)
+
+	c.crosPolicy()
+	c.ServeJSON()
+	c.StopRun()
+}
+
+// initAppController 执行当前路由请求对应的 Controller 初始化
+func (c *BaseController) initController() {
+	if ctrl, ok := c.AppController.(ControllerInterface); ok {
+		ctrl.Constructor()
+	}
+}
+
+func (c *BaseController) GetBodyData() (map[string]interface{}, error) {
+	var data map[string]interface{}
+	err := json.Unmarshal(c.Ctx.Input.RequestBody, &data)
+	return data, err
+}
+
+// Options 解决跨域先发送 options
+func (c *BaseController) Options() {
+	c.ResponseJSON("")
+}
+
+// crosPolicy 跨域策略
+func (c *BaseController) crosPolicy() {
+	runmode := beego.AppConfig.String("runmode")
+	allowMode := beego.AppConfig.String("cros::allowMode")
+
+	if runmode == allowMode {
+		allowMethods := beego.AppConfig.String("cros::allowMethods")
+		allowOrigin := beego.AppConfig.String("cros::allowOrigin")
+		allowCredentials := beego.AppConfig.String("cros::allowCredentials")
+
+		c.Ctx.Output.Header("Access-Control-Allow-Origin", allowOrigin)
+		c.Ctx.Output.Header("Access-Control-Allow-Methods", allowMethods)
+		c.Ctx.Output.Header("Access-Control-Allow-Credentials", allowCredentials)
+	}
+}

controllers/object.go

@@ -0,0 +1,31 @@
+package controllers
+
+import (
+	"strconv"
+	"time"
+	"voting/utils"
+)
+
+// Operations about object
+type ObjectController struct {
+	BaseController
+}
+
+// @Title Create
+// @Description create object
+// @Param	body		body 	models.Object	true		"The object content"
+// @Success 200 {string} object URL
+// @Failure 403 body is empty
+// @router / [post]
+func (c *ObjectController) Post() {
+	fNameExtra := strconv.FormatInt(time.Now().Unix(), 10)
+
+	fileURL, err := utils.UploadFileToBucket(c.Ctx.Request, "file", fNameExtra)
+	if err != nil {
+		c.ResponseError(err)
+	}
+
+	c.ResponseJSON(map[string]interface{}{
+		"url": fileURL,
+	})
+}

controllers/voting.go

@@ -0,0 +1,59 @@
+package controllers
+
+import (
+	"errors"
+	"strconv"
+	"voting/models"
+)
+
+// VotingController 是投票 controller
+type VotingController struct {
+	BaseController
+	db *models.VotingModel
+}
+
+// Constructor 初始化
+func (c *VotingController) Constructor() {
+	c.db = new(models.VotingModel)
+}
+
+// @Title List
+// @Description 获取报名人员列表
+// @Param   actid   path   string  true	"活动ID"
+// @Param   page   query   int  false	"分页"
+// @Param   q   query   string  false	"查询内容, 不填即为查询全部"
+// @Param   tab   query   int  false	"列表类型, 1 为最新上传, 2 为 Top50"
+// @Success 200 [object] models.MemInfo
+// @Failure 403 message of error
+// @router /members [get]
+func (c *VotingController) List() {
+	// 活动
+	actID, _ := c.GetInt(":actid")
+
+	// 分页
+	page, _ := c.GetInt("page")
+	if page < 1 {
+		page = 1
+	}
+	page -= 1
+
+	// tab 类型
+	tab, _ := c.GetInt("tab")
+
+	// 搜索条件
+	search := c.GetString("q")
+	name := search
+	memID, err := strconv.Atoi(search)
+	if err != nil {
+		memID = 0
+	} else {
+		name = ""
+	}
+
+	list, err := c.db.List(actID, name, memID, tab, page)
+	if err != nil {
+		c.ResponseError(errors.New("查询列表失败, 请重试"))
+	}
+
+	c.ResponseJSON(list)
+}

main.go

@@ -0,0 +1,15 @@
+package main
+
+import (
+	_ "voting/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/common.go

@@ -0,0 +1,20 @@
+package models
+
+const (
+	StatusReady = iota
+	StatusNormal
+)
+
+const (
+	PageNum = 10
+)
+
+// Insert datas
+func Insert(t ...interface{}) (int64, error) {
+	return DBEngine.Insert(t...)
+}
+
+// Update datas
+func Update(t interface{}, cols []string, query string, args ...interface{}) (int64, error) {
+	return DBEngine.Where(query, args...).Cols(cols...).Update(t)
+}

models/model/ta_voting_activities.go

@@ -0,0 +1,27 @@
+package model
+
+import (
+	"time"
+)
+
+type TaVotingActivities struct {
+	ActivityId int       `xorm:"not null pk autoincr SMALLINT(6)"`
+	RuleId     int       `xorm:"SMALLINT(6)"`
+	Music      string    `xorm:"TEXT"`
+	Name       string    `xorm:"VARCHAR(300)"`
+	Thumb      string    `xorm:"TEXT"`
+	Banner     string    `xorm:"TEXT"`
+	Desc       string    `xorm:"TEXT"`
+	Rules      string    `xorm:"TEXT"`
+	StartTime  time.Time `xorm:"DATETIME"`
+	EndTime    time.Time `xorm:"DATETIME"`
+	Remark     string    `xorm:"VARCHAR(500)"`
+	CreateDate time.Time `xorm:"DATETIME"`
+	Status     int       `xorm:"SMALLINT(6)"`
+	Members    int       `xorm:"INT(11)"`
+	StartId    int       `xorm:"INT(11)"`
+	AfterVote  string    `xorm:"TEXT"`
+	ShareTitle string    `xorm:"VARCHAR(100)"`
+	ShareIcon  string    `xorm:"TEXT"`
+	ShareDesc  string    `xorm:"VARCHAR(200)"`
+}

models/model/ta_voting_member.go

@@ -0,0 +1,19 @@
+package model
+
+import (
+	"time"
+)
+
+type TaVotingMember struct {
+	MemberId   int       `xorm:"not null pk INT(11)"`
+	ActivityId int       `xorm:"not null pk SMALLINT(6)"`
+	Name       string    `xorm:"VARCHAR(200)"`
+	Phone      string    `xorm:"VARCHAR(20)"`
+	Addr       string    `xorm:"TEXT"`
+	Message    string    `xorm:"TEXT"`
+	Remark     string    `xorm:"VARCHAR(500)"`
+	CreateDate time.Time `xorm:"DATETIME"`
+	Status     int       `xorm:"SMALLINT(6)"`
+	Vote       int       `xorm:"INT(11)"`
+	Openid     string    `xorm:"VARCHAR(64)"`
+}

models/model/ta_voting_member_photos.go

@@ -0,0 +1,16 @@
+package model
+
+import (
+	"time"
+)
+
+type TaVotingMemberPhotos struct {
+	PhotoId    int       `xorm:"not null pk autoincr INT(11)"`
+	MemberId   int       `xorm:"INT(11)"`
+	ActivityId int       `xorm:"SMALLINT(6)"`
+	Photo      string    `xorm:"TEXT"`
+	Type       string    `xorm:"VARCHAR(20)"`
+	Desc       string    `xorm:"VARCHAR(300)"`
+	CreateDate time.Time `xorm:"DATETIME"`
+	Status     int       `xorm:"SMALLINT(6)"`
+}

models/model/ta_voting_records.go

@@ -0,0 +1,15 @@
+package model
+
+import (
+	"time"
+)
+
+type TaVotingRecords struct {
+	VoteId     int       `xorm:"not null pk autoincr INT(11)"`
+	MemberId   int       `xorm:"INT(11)"`
+	ActivityId int       `xorm:"SMALLINT(6)"`
+	VoteDate   time.Time `xorm:"DATETIME"`
+	NickName   string    `xorm:"VARCHAR(200)"`
+	Avatar     string    `xorm:"TEXT"`
+	Openid     string    `xorm:"VARCHAR(64)"`
+}

models/model/ta_voting_rules.go

@@ -0,0 +1,8 @@
+package model
+
+type TaVotingRules struct {
+	RuleId    int    `xorm:"not null pk autoincr SMALLINT(6)"`
+	RuleValue string `xorm:"TEXT"`
+	Desc      string `xorm:"TEXT"`
+	Status    int    `xorm:"SMALLINT(6)"`
+}

models/models.go

@@ -0,0 +1,54 @@
+package models
+
+import (
+	"voting/utils"
+
+	"github.com/astaxie/beego"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/go-xorm/xorm"
+)
+
+var (
+	DBEngine *xorm.Engine
+	logger   *utils.Logger
+)
+
+func InitDB() {
+	DBEngine = NewDBEngine()
+	logger = utils.GetLogger()
+}
+
+// NewDBEngine 初始化数据库连接
+func NewDBEngine() *xorm.Engine {
+	dbType := "mysql"
+	dns := getMySQLDNS()
+
+	engine, err := xorm.NewEngine(dbType, dns)
+	// engine.ShowSQL()
+
+	if err != nil {
+		panic(err)
+	}
+
+	return engine
+}
+
+func getMySQLDNS() (dns string) {
+	conProt := beego.AppConfig.DefaultString("db::con_protocol", "tcp")
+	dbAddr := beego.AppConfig.DefaultString("db::db_addr", "localhost")
+	dbPort := beego.AppConfig.DefaultString("db::db_port", "3306")
+	userName := beego.AppConfig.DefaultString("db::username", "root")
+	password := beego.AppConfig.String("db::password")
+	database := beego.AppConfig.String("db::database")
+	charSet := beego.AppConfig.DefaultString("db::char_set", "utf8")
+
+	dns = userName
+
+	if len(password) > 0 {
+		dns += ":" + password
+	}
+
+	dns += "@" + conProt + "(" + dbAddr + ":" + dbPort + ")" + "/" + database + "?charset=" + charSet
+
+	return
+}

models/voting.go

@@ -0,0 +1,330 @@
+package models
+
+import (
+	"encoding/json"
+	"errors"
+	"strconv"
+	"time"
+	"voting/models/model"
+)
+
+const (
+	// 默认图片类型
+	DefaultPhotoType = "main"
+
+	// 列表排序类型
+	OrderByDate = 1
+	OrderByVote = 2
+
+	// 活动限制规则
+	RuleOfLimitEveryday = "limiteveryday"
+)
+
+var (
+	ErrVoteLimit        = errors.New("votelimit")
+	ErrActivityNotStart = errors.New("Activity not start")
+	ErrActivityIsOver   = errors.New("Activity is over")
+)
+
+type VotingModel struct{}
+
+type MemInfo struct {
+	model.TaVotingMember `xorm:"extends"`
+	Photo                string `xorm:"TEXT"`
+	Rank                 int
+}
+
+// List 报名列表
+func (m *VotingModel) List(actID int, name string, id, orderby, page int) (list []MemInfo, err error) {
+	query := `
+		SELECT t.*, s.photo
+		FROM
+			ta_voting_member t
+		JOIN ta_voting_member_photos s ON t.member_id = s.member_id
+			AND t.activity_id = s.activity_id
+			AND s.type = '` + DefaultPhotoType + `'
+			AND s.status = ` + strconv.Itoa(StatusNormal) + `
+		Where t.activity_id = '` + strconv.Itoa(actID) + `' AND 
+	`
+
+	// 依据 ID 或者 名称查询
+	// 全部列表的时候, 名称为空即可
+	if id != 0 {
+		query += `t.member_id = ` + strconv.Itoa(id)
+	} else {
+		query += `t.name like "%` + name + `%"`
+	}
+	query += " and t.status = " + strconv.Itoa(StatusNormal)
+
+	// 排序
+	if orderby == OrderByVote {
+		query += " order by t.vote desc"
+	} else {
+		query += " order by t.create_date desc"
+	}
+
+	// 分页
+	offset := (page - 1) * PageNum
+	if offset < 0 {
+		offset = 0
+	}
+
+	query += " limit " + strconv.Itoa(PageNum) + " offset " + strconv.Itoa(offset)
+
+	if err = DBEngine.SQL(query).Find(&list); err != nil {
+		logger.Error("查询活动列表失败: ", err)
+		return
+	}
+
+	return
+}
+
+// GetActivityInfo 查询活动
+func (m *VotingModel) GetActivityInfo(actID int) (act *model.TaVotingActivities, err error) {
+	_, err = DBEngine.Where("t.activity_id=?", actID).Get(act)
+	if err != nil {
+		logger.Error("查询活动失败: ", err)
+		return
+	}
+
+	return
+}
+
+// GetMemInfo 获取报名人员信息
+func (m *VotingModel) GetMemInfo(actID, memID int) (mem *MemInfo, err error) {
+	query := `
+		SELECT t.*, s.photo
+		FROM
+			ta_voting_member t
+		JOIN ta_voting_member_photos s ON t.member_id = s.member_id
+			AND t.activity_id = s.activity_id
+			AND s.type = '` + DefaultPhotoType + `'
+			AND s.status = ` + strconv.Itoa(StatusNormal) + `
+		Where t.activity_id = ? AND t.member_id = ?
+	`
+
+	if _, err = DBEngine.SQL(query, actID, memID).Get(mem); err != nil {
+		logger.Error("获取报名人员失败: ", err)
+		return
+	}
+
+	// 获取排名
+	cnt, e := DBEngine.Where("activity_id=?", actID).
+		And("status=?", StatusNormal).
+		And("vote>?", mem.Vote).
+		Count()
+	if e != nil {
+		logger.Error("获取报名人员排名失败:", err)
+
+		// -1 代表发生错误
+		mem.Rank = -1
+		return
+	}
+
+	mem.Rank = int(cnt)
+	return
+}
+
+// Registe 报名保存
+func (m *VotingModel) Registe(memInfo *MemInfo) error {
+	if memInfo.ActivityId == 0 {
+		return errors.New("活动报名失败, 没有指定活动内容")
+	}
+
+	now := time.Now().Local()
+
+	mem := memInfo.TaVotingMember
+	photo := model.TaVotingMemberPhotos{
+		ActivityId: mem.ActivityId,
+		Photo:      memInfo.Photo,
+		Type:       DefaultPhotoType,
+		CreateDate: now,
+		Status:     StatusNormal,
+	}
+
+	// 检查已有最大值
+	maxID, err := m.getMaxMemID(mem.ActivityId)
+	if err != nil {
+		return err
+	}
+
+	mem.MemberId = int(maxID + 1)
+	photo.MemberId = mem.MemberId
+	memInfo.MemberId = mem.MemberId
+
+	if _, err := Insert(&photo); err != nil {
+		logger.Error("保存活动报名图片失败: ", err)
+		return err
+	}
+
+	if _, err := Insert(&mem); err != nil {
+		logger.Error("保存活动报名失败: ", err)
+		return err
+	}
+
+	// 反更新报名人数
+	// TODO
+
+	return nil
+}
+
+func (m *VotingModel) getMaxMemID(actID int) (maxID int64, err error) {
+	query := `
+		SELECT
+			max(start_id) as max_id
+		FROM
+			(
+				SELECT
+					max(t.member_id) AS start_id
+				FROM
+					ta_voting_member t
+				UNION
+					SELECT
+						s.start_id
+					FROM
+						ta_voting_activities s
+					WHERE
+						s.activity_id = ?
+			) a
+	`
+
+	if _, err = DBEngine.SQL(query, actID).Get(&maxID); err != nil {
+		logger.Error("查询活动报名最大ID失败: ", err)
+		return
+	}
+
+	return
+}
+
+// Vote 投票
+func (m *VotingModel) Vote(rec *model.TaVotingRecords) error {
+	// 校验活动
+	if err := m.checkActivity(rec.ActivityId); err != nil {
+		return err
+	}
+
+	rule, err := m.getActivityRule(rec.ActivityId)
+	if err != nil {
+		return err
+	}
+
+	var ruleParam map[string]interface{}
+	if err := json.Unmarshal([]byte(rule.RuleValue), &ruleParam); err != nil {
+		logger.Error("转换活动规则类型失败: ", err)
+		return err
+	}
+
+	// 如果有每天最大投票限制
+	if v, ok := ruleParam[RuleOfLimitEveryday]; ok {
+		limitEveryday := int64(v.(float64))
+
+		voteNum, err := m.getVoteNumOfMember(rec)
+		if err != nil {
+			return err
+		}
+
+		if voteNum >= limitEveryday {
+			return ErrVoteLimit
+		}
+	}
+
+	// 每天投票
+	rec.VoteDate = time.Now().Local()
+	if _, err := Insert(rec); err != nil {
+		logger.Error("活动投票失败: ", err)
+		return err
+	}
+
+	// 反更新次数
+	if err := m.updateVoteOfMember(rec.ActivityId, rec.MemberId); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// 获取投票规则
+func (m *VotingModel) getActivityRule(actID int) (rule *model.TaVotingRules, err error) {
+	query := `
+		SELECT
+			t.*
+		FROM
+			ta_voting_rules t
+		JOIN ta_voting_activities s ON t.rule_id = s.rule_id
+		AND s.activity_id = ?
+	`
+
+	if _, err = DBEngine.SQL(query, actID).Get(rule); err != nil {
+		logger.Error("查询活动规则失败: ", err)
+		return
+	}
+
+	if rule.RuleValue == "" {
+		rule.RuleValue = "{}"
+	}
+
+	return
+}
+
+// 获取当前人员投票数
+func (m *VotingModel) getVoteNumOfMember(rec *model.TaVotingRecords) (num int64, err error) {
+	query := `
+		SELECT
+			*
+		FROM
+			ta_voting_records t
+		WHERE
+			t.activity_id = ?
+		AND t.member_id = ?
+		AND t.openid = ?
+		AND date_format(t.vote_date, "%Y-%m-%d") = date_format(now(), "%Y-%m-%d")
+	`
+
+	num, err = DBEngine.SQL(query, rec.ActivityId, rec.MemberId, rec.Openid).Count()
+	if err != nil {
+		logger.Error("查询人员投票次数失败: ", err)
+		return
+	}
+
+	return
+}
+
+// 反更新投票次数
+func (m *VotingModel) updateVoteOfMember(actID, memID int) error {
+	query := `
+	UPDATE ta_voting_member t
+		SET t.vote = t.vote + 1
+		WHERE
+			t.activity_id = ?
+		AND t.member_id = ?
+	`
+
+	if _, err := DBEngine.Exec(query, actID, memID); err != nil {
+		logger.Error("更新投票次数失败: ", err)
+		return err
+	}
+
+	return nil
+}
+
+// 校验活动
+// 只校验活动时间
+func (m *VotingModel) checkActivity(actID int) error {
+	act := model.TaVotingActivities{}
+
+	if _, err := DBEngine.Where("activity_id=?", actID).Get(&act); err != nil {
+		logger.Error("校验活动失败: ", err)
+		return err
+	}
+
+	now := time.Now().Local()
+	if act.StartTime.After(now) {
+		return ErrActivityNotStart
+	}
+
+	if act.EndTime.Before(now) {
+		return ErrActivityIsOver
+	}
+
+	return nil
+}

routers/router.go

@@ -0,0 +1,25 @@
+// @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 (
+	"voting/controllers"
+
+	"github.com/astaxie/beego"
+)
+
+func init() {
+	ns := beego.NewNamespace("/api",
+		beego.NSNamespace("/file",
+			beego.NSInclude(
+				&controllers.ObjectController{},
+			),
+		),
+	)
+	beego.AddNamespace(ns)
+}

tests/default_test.go

@@ -0,0 +1,38 @@
+package test
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"runtime"
+	"path/filepath"
+	_ "voting/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/aliyun.go

@@ -0,0 +1,85 @@
+package utils
+
+import (
+	"errors"
+	"net/http"
+	"strings"
+
+	"github.com/aliyun/aliyun-oss-go-sdk/oss"
+	"github.com/astaxie/beego"
+)
+
+var ossCli *oss.Client
+
+func OssInit() {
+	logger := GetLogger()
+
+	cert, err := beego.AppConfig.GetSection("alioss")
+	if err != nil {
+		logger.Error("读取阿里 OSS 配置失败: ", err)
+		return
+	}
+
+	OssClientSingleton(cert["Endpoint"], cert["AccessKeyId"], cert["AccessKeySecret"])
+}
+
+func OssClientSingleton(endpoint, accessKeyID, accessKeySecret string) {
+	if ossCli == nil {
+		var err error
+		logger := GetLogger()
+		ossCli, err = oss.New(endpoint, accessKeyID, accessKeySecret)
+		if err != nil {
+			logger.Error("初始化阿里 OSS 失败: ", err)
+		}
+	}
+}
+
+// GetOssClient 获取 OSS 客户端
+func GetOssClient() *oss.Client {
+	return ossCli
+}
+
+// UploadFileToBucket 简单上传文件
+func UploadFileToBucket(req *http.Request, fromFront string, nameSuffix ...string) (string, error) {
+	logger := GetLogger()
+	bucket := beego.AppConfig.String("alioss::Bucket")
+	endpoint := beego.AppConfig.String("alioss::Endpoint")
+
+	if ossCli == nil {
+		logger.Error("未成功初始化OSS 客户端")
+		return "", errors.New("未成功初始化OSS 客户端")
+	}
+
+	bkt, err := ossCli.Bucket(bucket)
+	if err != nil {
+		logger.Error("初始化OSS Bucket 失败: ", err)
+		return "", errors.New("初始化OSS Bucket 失败")
+	}
+
+	file, fh, err := req.FormFile(fromFront)
+	if err != nil {
+		logger.Error("读取文件失败: ", err)
+		return "", errors.New("未读取到文件或读取失败")
+	}
+	defer file.Close()
+
+	fname := fh.Filename
+	if len(nameSuffix) > 0 && nameSuffix[0] != "" {
+		fArr := strings.Split(fname, ".")
+		fArr[0] = fArr[0] + nameSuffix[0]
+		fname = strings.Join(fArr, ".")
+	}
+
+	err = bkt.PutObject(fname, file)
+	if err != nil {
+		logger.Error("上传文件到 OSS 失败: ", err)
+		return "", errors.New("上传文件到 OSS 失败")
+	}
+
+	return strings.Join([]string{
+		"http://",
+		bucket + ".",
+		endpoint + "/",
+		fname,
+	}, ""), nil
+}

utils/log.go

@@ -0,0 +1,67 @@
+package utils
+
+import (
+	"github.com/astaxie/beego"
+)
+
+type Logger struct{}
+
+var logger *Logger
+
+func GetLogger() *Logger {
+	if logger == nil {
+		logger = new(Logger)
+	}
+
+	return logger
+}
+
+func (l *Logger) Critical(v string, o ...interface{}) {
+	f := []interface{}{v}
+
+	if o != nil && len(o) > 0 {
+		f = append(f, o...)
+	}
+
+	beego.Critical(f)
+}
+
+func (l *Logger) Error(v string, o ...interface{}) {
+	f := []interface{}{v}
+
+	if o != nil && len(o) > 0 {
+		f = append(f, o...)
+	}
+
+	beego.Error(f)
+}
+
+func (l *Logger) Warning(v string, o ...interface{}) {
+	f := []interface{}{v}
+
+	if o != nil && len(o) > 0 {
+		f = append(f, o...)
+	}
+
+	beego.Warning(f)
+}
+
+func (l *Logger) Info(v string, o ...interface{}) {
+	f := []interface{}{v}
+
+	if o != nil && len(o) > 0 {
+		f = append(f, o...)
+	}
+
+	beego.Info(f)
+}
+
+func (l *Logger) Debug(v string, o ...interface{}) {
+	f := []interface{}{v}
+
+	if o != nil && len(o) > 0 {
+		f = append(f, o...)
+	}
+
+	beego.Debug(f)
+}

utils/wechat.go

@@ -0,0 +1,100 @@
+package utils
+
+import (
+	"github.com/astaxie/beego"
+	"github.com/zjxpcyc/wechat/wx"
+)
+
+var wxClients *wx.Client
+
+type WechatUser struct {
+	OpenID     string  `json:"openid"`
+	NickName   string  `json:"nickname"`
+	Sex        float64 `json:"sex"`
+	Province   string  `json:"province"`
+	City       string  `json:"city"`
+	Country    string  `json:"country"`
+	HeadImgURL string  `json:"headimgurl"`
+	UnionID    string  `json:"unionid"`
+}
+
+// WxClientSingleton 初始化
+func WxClientSingleton(cert map[string]string) {
+	if wxClients == nil {
+		wxClients = wx.NewClient(cert)
+	}
+}
+
+func GetWxClient() *wx.Client {
+	return wxClients
+}
+
+// GetWxAppID 获取微信ID
+func GetWxAppID() string {
+	if wxClients != nil {
+		return wxClients.GetAppID()
+	} else {
+		return ""
+	}
+}
+
+func WechatInit() {
+	logger := GetLogger()
+	wx.SetLogInst(logger)
+
+	cert, err := beego.AppConfig.GetSection("wechat")
+	if err != nil {
+		logger.Error("初始化微信失败: ", err)
+	}
+
+	WxClientSingleton(cert)
+}
+
+// MapToWechatUser 映射微信人员
+func MapToWechatUser(data map[string]interface{}) *WechatUser {
+	subscribe, has := data["subscribe"]
+	if has {
+		if subscribe == nil {
+			return nil
+		}
+
+		sub := subscribe.(float64)
+		if sub == 0 {
+			return nil
+		}
+	}
+
+	user := WechatUser{
+		OpenID: data["openid"].(string),
+	}
+
+	if data["sex"] != nil {
+		user.Sex = data["sex"].(float64)
+	}
+
+	if data["nickname"] != nil {
+		user.NickName = data["nickname"].(string)
+	}
+
+	if data["province"] != nil {
+		user.Province = data["province"].(string)
+	}
+
+	if data["city"] != nil {
+		user.City = data["city"].(string)
+	}
+
+	if data["country"] != nil {
+		user.Country = data["country"].(string)
+	}
+
+	if data["headimgurl"] != nil {
+		user.HeadImgURL = data["headimgurl"].(string)
+	}
+
+	if data["unionid"] != nil {
+		user.UnionID = data["unionid"].(string)
+	}
+
+	return &user
+}