feats

controllers/customer/customer.go

@@ -0,0 +1,82 @@
+package customer
+
+import (
+	"errors"
+	"net/http"
+	"spaceofcheng/services/controllers"
+	"spaceofcheng/services/models"
+	"spaceofcheng/services/models/model"
+	"spaceofcheng/services/service/customer"
+	"spaceofcheng/services/utils"
+)
+
+// CustomerController 用户
+type CustomerController struct {
+	serv *customer.CustomerServ
+	controllers.BaseController
+}
+
+// Constructor 初始化 Controller
+// @Title Constructor
+// @Description 初始化 Controller, 系统自动调用
+func (c *CustomerController) Constructor() {
+	c.serv = customer.NewCustomerServ(c.Context)
+}
+
+// CustWXList 获取客户列表
+func (c *CustomerController) CustWXList() {
+	phone := c.GetString("phone")
+	page, _ := c.GetInt("page")
+
+	custList, total, err := c.serv.GetCustWithWXList(phone, page)
+	if err != nil {
+		utils.LogError("获取客户列表失败: " + err.Error())
+		c.ResponseError(
+			errors.New("获取客户列表失败"),
+			http.StatusInternalServerError,
+		)
+	}
+
+	c.ResponseJSON(map[string]interface{}{
+		"list":  custList,
+		"total": total,
+	})
+}
+
+// GetCustOrUserByID 获取用户信息
+func (c *CustomerController) GetCustOrUserByID() {
+	userMapRaw := c.Context.Get("userMap")
+	if userMapRaw == nil {
+		c.ResponseError(errors.New("获取用户信息失败, 请确认使用微信浏览"))
+	}
+
+	userMap := userMapRaw.(model.TaUserMapping)
+
+	if userMap.UserType == models.USERMAP_USER {
+		// 如果是用户
+		userID := userMap.UserId
+		user, err := c.serv.GetCustUserByID(userID)
+		if err != nil {
+			utils.LogError(err)
+			c.ResponseError(errors.New("获取用户信息失败, 请重试"))
+		}
+
+		c.ResponseJSON(map[string]interface{}{
+			"user":     user,
+			"userType": userMap.UserType,
+		})
+	} else {
+		// 如果是客户
+		custID := userMap.UserId
+		cust, err := c.serv.GetCustWithWXByID(custID)
+		if err != nil {
+			utils.LogError(err)
+			c.ResponseError(errors.New("获取用户信息失败, 请重试"))
+		}
+
+		c.ResponseJSON(map[string]interface{}{
+			"customer": cust,
+			"userType": userMap.UserType,
+		})
+	}
+}

controllers/message.go

@@ -0,0 +1,42 @@
+package controllers
+
+import (
+	"errors"
+	"spaceofcheng/services/utils"
+)
+
+// SendCaptcha 发送验证码
+func (c *BaseController) SendCaptcha() {
+	phone := c.GetString("phone")
+
+	if phone == "" {
+		c.ResponseError(errors.New("请确定手机号码"))
+	}
+
+	if err := utils.SendCaptcha(phone); err != nil {
+		utils.LogError(err)
+		c.ResponseError(errors.New("验证码发送失败, 请重试"))
+	}
+
+	c.ResponseJSON("ok")
+}
+
+// ValidCaptcha 发送验证码
+func (c *BaseController) ValidCaptcha() {
+	phone := c.GetString("phone")
+
+	if phone == "" {
+		c.ResponseError(errors.New("请确定手机号码"))
+	}
+
+	captcha := c.GetString("captcha")
+	if captcha == "" {
+		c.ResponseError(errors.New("验证码不能为空"))
+	}
+
+	if err := utils.ValidCaptcha(phone, captcha); err != nil {
+		c.ResponseError(err)
+	}
+
+	c.ResponseJSON("ok")
+}

controllers/sms.go

@@ -1,15 +0,0 @@
-package controllers
-
-import (
-	"errors"
-)
-
-// SendSMS 发送短信
-func (c *BaseController) SendSMS() {
-	phone := c.GetString("phone")
-
-	if phone == "" {
-		c.ResponseError(errors.New("请确定手机号码"))
-	}
-
-}

models/customer/customer.go

@@ -5,6 +5,9 @@ import (
 	"spaceofcheng/services/models"
 	"spaceofcheng/services/models/model"
 	"spaceofcheng/services/utils"
+	"time"
+
+	"github.com/yl10/kit/guid"
 
 	"github.com/go-xorm/xorm"
 )
@@ -24,7 +27,7 @@ func NewCustomerDAO(ctx *utils.Context) *CustomerDAO {
 }
 
 // CustWithWXList 客户列表
-func (m *CustomerDAO) CustWithWXList(phone string, page ...int) ([]CustWithWX, error) {
+func (m *CustomerDAO) CustWithWXList(phone string, page ...int) ([]CustWithWX, int64, error) {
 	if phone == "" {
 		phone = "%"
 	} else {
@@ -57,21 +60,16 @@ func (m *CustomerDAO) CustWithWXList(phone string, page ...int) ([]CustWithWX, e
 		AND t.status > ?
 		ORDER BY
 			t.create_date DESC
-		LIMIT ? OFFSET ?	
 	`
 
 	var cust []CustWithWX
-	if err := m.db.SQL(
-		query,
-		phone,
-		models.STATUS_DEL,
-		pageNum,
-		offset,
-	).Find(&cust); err != nil {
-		return nil, err
+
+	cnt, err := utils.GetPageList(m.db, &cust, []int{pageNum, offset}, query, phone, models.STATUS_DEL)
+	if err != nil {
+		return nil, 0, err
 	}
 
-	return cust, nil
+	return cust, cnt, nil
 }
 
 // GetCustWithWXByID 获取客户(微信)信息
@@ -101,17 +99,22 @@ func (m *CustomerDAO) GetCustWithWXByID(custID string) (*CustWithWX, error) {
 
 // GetCustomerByPhone 依据手机号获取客户信息
 func (m *CustomerDAO) GetCustomerByPhone(phone string) (*model.TaCustomer, error) {
-	cust := model.TaCustomer{}
+	custs := []model.TaCustomer{}
 
-	has, err := m.db.Where("phone=?", phone).Get(&cust)
-	if err != nil {
+	if err := m.db.Where("phone=?", phone).Find(&custs); err != nil {
 		return nil, err
 	}
 
-	if !has {
+	if len(custs) == 0 {
 		return nil, nil
 	}
 
+	if len(custs) > 0 {
+		return nil, errors.New("数据异常, 一个手机绑定了多个用户")
+	}
+
+	cust := custs[0]
+
 	return &cust, nil
 }
 
@@ -124,3 +127,51 @@ func (m *CustomerDAO) UpdateUserMapping(userMap *model.TaUserMapping, cols []str
 	_, err := m.db.Where("mapping_id=?", userMap.MappingId).Cols(cols...).Update(&userMap)
 	return err
 }
+
+// GetCaseByID 获取案场
+func (m *CustomerDAO) GetCaseByID(caseID string) (*model.SysCase, error) {
+	if caseID == "" {
+		return nil, errors.New("获取案场信息失败")
+	}
+
+	cs := model.SysCase{}
+	has, err := m.db.Where("case_id=?", caseID).Get(&cs)
+	if err != nil {
+		return nil, err
+	}
+
+	if !has {
+		return nil, nil
+	}
+
+	return &cs, nil
+}
+
+// GetUserByID 获取系统用户
+func (m *CustomerDAO) GetUserByID(userID string) (*model.SysUser, error) {
+	if userID == "" {
+		return nil, errors.New("获取相关人员信息失败")
+	}
+
+	user := model.SysUser{}
+	has, err := m.db.Where("user_id=?", userID).Get(&user)
+	if err != nil {
+		return nil, err
+	}
+
+	if !has {
+		return nil, nil
+	}
+
+	return &user, nil
+}
+
+// SaveCustomer 新增客户
+func (m *CustomerDAO) SaveCustomer(cust *model.TaCustomer) error {
+	cust.CustomerId = guid.NewGUIDString()
+	cust.CreateDate = time.Now().Local()
+	cust.Status = models.STATUS_NORMAL
+
+	_, err := m.db.Insert(&cust)
+	return err
+}

models/customer/types.go

@@ -8,4 +8,5 @@ import (
 type CustWithWX struct {
 	model.TaCustomer    `xorm:"extends"`
 	model.TaUserMapping `xorm:"extends"`
+	Points              int
 }

service/customer/customer.go

@@ -1,12 +1,14 @@
 package customer
 
 import (
+	"encoding/json"
 	"errors"
 	"spaceofcheng/services/models"
 	"spaceofcheng/services/models/customer"
 	"spaceofcheng/services/models/model"
 	"spaceofcheng/services/service/events"
 	"spaceofcheng/services/utils"
+	"strconv"
 )
 
 // CustomerServ 系统处理
@@ -24,7 +26,7 @@ func NewCustomerServ(ctx *utils.Context) *CustomerServ {
 }
 
 // GetCustWithWXList 获取客户列表
-func (s *CustomerServ) GetCustWithWXList(phone string, page ...int) ([]customer.CustWithWX, error) {
+func (s *CustomerServ) GetCustWithWXList(phone string, page ...int) ([]customer.CustWithWX, int64, error) {
 	return s.dao.CustWithWXList(phone, page...)
 }
 
@@ -42,15 +44,46 @@ func (s *CustomerServ) GetCustWithWXByID(custID string) (*customer.CustWithWX, e
 	return cust, nil
 }
 
+// GetCustUserByID 获取用户信息(自定义内容)
+func (s *CustomerServ) GetCustUserByID(userID string) (*UserProfile, error) {
+	user, err := s.dao.GetUserByID(userID)
+	if err != nil {
+		return nil, err
+	}
+
+	if user.Status != models.STATUS_NORMAL {
+		return nil, errors.New("用户状态异常")
+	}
+
+	userProfile := UserProfile{
+		UserId:     user.UserId,
+		UserName:   user.UserName,
+		RealName:   user.RealName,
+		Sex:        user.Sex,
+		WorkNo:     user.WorkNo,
+		Email:      user.Email,
+		Phone:      user.Phone,
+		Headimgurl: user.Headimgurl,
+	}
+
+	return &userProfile, nil
+}
+
 // BindWechat 绑定微信用户
-func (s *CustomerServ) BindWechat(userMap *model.TaUserMapping, phone string) (*models.TaCustomer, error) {
+func (s *CustomerServ) BindWechat(userMap *model.TaUserMapping, phone, caseID, userID string) (*model.TaCustomer, error) {
 	cust, err := s.dao.GetCustomerByPhone(phone)
 	if err != nil {
 		return nil, err
 	}
 
+	// 用户不存在, 则新增
 	if cust == nil {
-		return nil, errors.New("手机号对应用户不存在")
+		newCust, err := s.SaveNewCustomer(phone, caseID, userID)
+		if err != nil {
+			return nil, err
+		}
+
+		cust = newCust
 	}
 
 	if cust.Status != models.STATUS_NORMAL {
@@ -76,3 +109,61 @@ func (s *CustomerServ) BindWechat(userMap *model.TaUserMapping, phone string) (*
 
 	return cust, nil
 }
+
+// SaveNewCustomer 新增客户
+func (s *CustomerServ) SaveNewCustomer(phone, caseID, userID string) (*model.TaCustomer, error) {
+	// 校验案场信息
+	cs, err := s.dao.GetCaseByID(caseID)
+	if err != nil {
+		return nil, err
+	}
+
+	if cs == nil || cs.Status == models.STATUS_DEL {
+		return nil, errors.New("系统无该推荐案场")
+	}
+
+	// 校验推荐人
+	user, err := s.dao.GetUserByID(userID)
+	if err != nil {
+		return nil, err
+	}
+
+	if user == nil || user.Status == models.STATUS_DEL {
+		return nil, errors.New("系统无该推荐人")
+	}
+
+	// 微信信息
+	userMap := s.ctx.Get("userMap").(model.TaUserMapping)
+	wxInfo := make(map[string]interface{})
+	if err := json.Unmarshal([]byte(userMap.AccountInfo), &wxInfo); err != nil {
+		utils.LogError(err.Error())
+		return nil, errors.New("校验用户微信信息失败")
+	}
+
+	nickyName := wxInfo["nickname"].(string)
+	sex := wxInfo["sex"].(string)
+	headimgurl := wxInfo["headimgurl"].(string)
+
+	sexInt, _ := strconv.Atoi(sex)
+	if sexInt == 0 {
+		sexInt = 1 // 默认男性
+	}
+
+	cust := model.TaCustomer{
+		OrgId:         cs.OrgId,
+		CustomerName:  nickyName,
+		Name:          nickyName,
+		Sex:           sexInt,
+		Headimgurl:    headimgurl,
+		RecommendId:   user.UserId,
+		RecommendName: user.RealName,
+		RecommendCase: cs.CaseId,
+	}
+
+	if err := s.dao.SaveCustomer(&cust); err != nil {
+		utils.LogError("更新客户信息失败: " + err.Error())
+		return nil, errors.New("更新客户信息失败")
+	}
+
+	return &cust, nil
+}

service/customer/types.go

@@ -0,0 +1,16 @@
+package customer
+
+type UserProfile struct {
+	UserId     string
+	UserName   string
+	RealName   string
+	Sex        int
+	WorkNo     string
+	Email      string
+	Phone      string
+	Headimgurl string
+}
+
+func (t *UserProfile) TableName() string {
+	return "sys_user"
+}

utils/pagenavi.go

@@ -0,0 +1,61 @@
+package utils
+
+import (
+	"errors"
+	"regexp"
+	"strconv"
+
+	"github.com/go-xorm/xorm"
+)
+
+// GetPageList 获取列表
+func GetPageList(db *xorm.Session, target interface{}, limit []int, sqlorArgs ...interface{}) (int64, error) {
+	if sqlorArgs == nil || len(sqlorArgs) == 0 {
+		return 0, errors.New("SQL 语句不存在")
+	}
+
+	countSQLorArgs := GetCountSQL(sqlorArgs...)
+	cntRes, err := db.Query(countSQLorArgs...)
+	if err != nil {
+		return 0, err
+	}
+
+	cntStr := string(cntRes[0]["cnt"])
+	cnt, _ := strconv.ParseInt(cntStr, 10, 64)
+
+	sql, args := PackLimitToSQL(limit, sqlorArgs...)
+
+	err = db.SQL(sql, args...).Find(target)
+	if err != nil {
+		return 0, err
+	}
+
+	return cnt, nil
+}
+
+// GetCountSQL 获取计算 Count 的相关 SQL 脚本
+func GetCountSQL(sqlorArgs ...interface{}) []interface{} {
+	sql := sqlorArgs[0].(string)
+
+	re := regexp.MustCompile("(?ims)(select .* from)")
+	cntSQL := re.ReplaceAllString(sql, "select count(*) as cnt from")
+
+	newSQLorArgs := []interface{}{cntSQL}
+	newSQLorArgs = append(newSQLorArgs, sqlorArgs[1:])
+	return newSQLorArgs
+}
+
+// PackLimitToSQL 添加 limit 语句到 sql 最后
+func PackLimitToSQL(limit []int, sqlorArgs ...interface{}) (string, []interface{}) {
+	sql := sqlorArgs[0].(string)
+	args := sqlorArgs[1:]
+
+	switch len(limit) {
+	case 1:
+		return sql + " LIMIT " + strconv.Itoa(limit[0]), args
+	case 2:
+		return sql + " LIMIT " + strconv.Itoa(limit[0]) + " OFFSET " + strconv.Itoa(limit[1]), args
+	default:
+		return sql, args
+	}
+}

utils/pagenavi_test.go

@@ -0,0 +1,71 @@
+package utils_test
+
+import (
+	"spaceofcheng/services/utils"
+	"testing"
+)
+
+func TestGetCountSQL(t *testing.T) {
+	testCases := []map[string]string{
+		map[string]string{
+			"sql":      "Select field1, field2 From table",
+			"expected": "select count(*) as cnt from table",
+		},
+		map[string]string{
+			"sql": `Select 
+								field1, field2 From table where status = ?`,
+			"expected": "select count(*) as cnt from table where status = ?",
+		},
+	}
+
+	for inx, cs := range testCases {
+		sql := cs["sql"]
+		expected := cs["expected"]
+
+		var res []interface{}
+		if inx < 1 {
+			res = utils.GetCountSQL(sql)
+		} else {
+			res = utils.GetCountSQL(sql, inx)
+		}
+
+		if res == nil || len(res) == 0 {
+			t.Fatalf("Test GetCountSQL fail --1")
+		}
+
+		newSQL := res[0].(string)
+		if newSQL != expected {
+			t.Fatalf("Test GetCountSQL fail: %s", newSQL)
+		}
+	}
+}
+
+func TestPackLimitToSQL(t *testing.T) {
+	testCases := []map[string]string{
+		map[string]string{
+			"sql":      "Select * From table",
+			"expected": "Select * From table LIMIT 1 OFFSET 0",
+		},
+		map[string]string{
+			"sql":      "Select * From table",
+			"expected": "Select * From table LIMIT 1",
+		},
+	}
+
+	for inx, cs := range testCases {
+		sql := cs["sql"]
+		expected := cs["expected"]
+
+		var sqlRes string
+		if inx < 1 {
+			sqlRes, _ = utils.PackLimitToSQL([]int{inx + 1, inx}, sql)
+		} else {
+			sqlRes, _ = utils.PackLimitToSQL([]int{inx}, sql, inx)
+		}
+
+		if sqlRes != expected {
+			t.Fatalf("\n%v: Test PackLimitToSQL fail: %s", inx+1, sqlRes)
+		}
+	}
+
+}