一个深漂近10年的程序员立志用未来10年时间去维护一个项目，这个项目的功能未定，用到什么就做什么，遇到什么就写什么。其主要目的有二，一为加深自己的技术深度，二为其他学习者提供参考。
感兴趣的可以跟着我一起做这个项目，不收费、不套路、人间自有真情在。
备注：前面项目中用到的代码已经分享到GitHub中去了，并且以后所有项目中会出现的代码都会提交上去，欢迎查阅。
地址：https://github.com/kaiyuan10nian/kaiyuan10nian 感兴趣的可以点个star哦~开源十年项目的更新首发于公众号：计算机自学平台，有兴趣的小伙伴可以持续关注，并欢迎各位加我的微信(kaiyuan10nian)跟我一起完成并推动项目的发展。
Go实战项目-给项目添加一个邀请码的功能
前面总结里面说过，我要给我们的blog项目添加一个邀请码的功能，因为这个项目做完之后我要部署出去的，虽说不靠这个东西去盈利，但是实战要有实战的样子至少把一个项目从0做到1的过程还是要有的。这里就涉及到一个很严重的问题：怎么控制内容？（在国内不可能不控制内容的）
- 人工去审核（我没那么多时间）
- 接入第三方内容审核（要花钱的）
- 智能审核（没这个技术~后面也许可以考虑在开源十年中去实现它）
所以没有办法了，既然我控制不了内容，那我就从源头控制：直接控制输出内容的人。
所以我想到的办法就是在注册的时候做一个限制，不能让每个人都可以注册成功，只能让那些得到我们这个圈子成员信任的人才可以注册体验。那么实现的逻辑要提前去思考了：
- 邀请码怎么来的？
邀请码是系统生成的，由已注册会员手动生成，没给会员允许生成5个邀请码，每个邀请码只能使用1次。
- 邀请码在哪使用？
邀请码在新会员注册的时候使用（为必填项）
- 邀请码怎么追溯？
因为邀请码的生成是已注册会员手动生成的，所以我会在库里面去拉一张表单独去记录，如果有新会员使用了邀请码那么我也会记录下来，后面若想去追溯就很简单了
根据上面的思路，我们一步一步来实现其功能：
1、先写接口路由：
r.GET("/v1/account/invite", common.AuthMiddleware(),controller.Invite)
因为邀请码只能是已注册会员去生成，所以必须加上common.AuthMiddleware()权限限制。
2、实现邀请码的生成：
func RandomString(n int) string {
var letters = []byte("23456789qwertyupkjhgfdsazxcvbnmMNBVCXZASDFGHJKPOUYTREWQ")
result := make([]byte,n)
rand.Seed(time.Now().Unix())
for i := range result {
result[i] = letters[rand.Intn(len(letters))]
}
return string(result)
}
这个其实就是一个生成随机值的函数，其中我把'0'、'1'、'i'、'l'、'I'、'L'等不容易辨认的字符都去掉了。
3、实现邀请码接口功能：
//创建用户邀请码
func Invite(ctx *gin.Context){
udata ,_:= ctx.Get("user")//首先通过token去获取用户信息
dto := dto.ToUserDto(udata.(model.User))
DB := common.GetDB()
var inviteCodes []model.InviteCode
DB.Where("userid = ?",dto.ID).Find(&inviteCodes)//根据用户信息去查询这个用户有多少个邀请码
fmt.Println(len(inviteCodes))
if len(inviteCodes) < 5{//如果不够5个则允许他继续生成邀请码
code := RandomString(5)
newCode := model.InviteCode{
Userid: dto.ID,
Code: code,
Status: 0,
}
//生成成功后把它记录进库表里面
if err := DB.Create(&newCode).Error;err != nil{
response.Response(ctx,http.StatusInternalServerError,500,nil,err.Error())
return
}
response.Response(ctx,http.StatusOK,6000,gin.H{"inviteCode":code},"邀请码生成成功")
}else{
response.Response(ctx,http.StatusOK,60001,nil,"每个人只可以拥有5个邀请码")
return
}
}
上述提到了邀请码的创建，我们创建了一个表invitecode专门用来存储（下文有model代码）
4、去修改注册功能
原本我们的注册只需要手机号并设置密码即可，但是我们要控制注册了，那么要加几个验证条件：
1）推荐人是否存在？
2）邀请码是否存在？
3）邀请码是否和推荐人匹配？
4）邀请码是否被人使用过了？
这几个条件判断之后，如果都通过那么给予放行，放行后还要做一个操作：
1）标记该邀请码已被使用
2）把该邀请码跟注册账号进行关联
针对上述逻辑，首先针对库表进行设计和改造
直接在用户表新增了一个字段invitecode，目的就是把邀请码和注册者进行关联。
type User struct {
gorm.Model
Name string `gorm:"type:varchar(20);not null"`
Mobile string `gorm:"varchar(11);not null;unique"`
Password string `gorm:"size:255;not null"`
InviteCode string `json:"invitecode" gorm:"not null"`
}
然后，新建了一个表，专门用来存我们生成的邀请码，以及记录邀请码和生成邀请码的用户进行关联
type InviteCode struct {
gorm.Model
Userid uint `json:"user_id" gorm:"not null"`
Code string `json:"code" gorm:"not null"`
Status uint `json:"status" gorm:"not null"`
}
库表建好了，下面就是在注册时，我们针对以上逻辑的代码实现：（全部代码请查看github）
if isTelephoneExist(DB,recommender){//查询这个推荐者存不存在
id := getRecommenderId(DB,recommender)//先获取推荐者的ID
//判断其邀请码的准确性
var inviteCodes []model.InviteCode
DB.Where("userid = ?",id).Find(&inviteCodes)//根据id去查这个推荐者有多少邀请码
isValid := false
for _,inviteCode := range inviteCodes{//轮寻他所有邀请码
isValid = strings.EqualFold(code,inviteCode.Code)//一一对比，查看注册者输入的是否和库里的一致
if isValid {
if inviteCode.Status == 0 {//如果一致，则再验证这个是否已经被用了
DB.Model(&inviteCode).Update("status",1)
break
}else{
response.Response(ctx,http.StatusOK,60002,nil,"邀请码已使用")
return
}
}
}
if !isValid {
response.Response(ctx,http.StatusOK,60002,nil,"邀请码不存在")
return
}
}else {
//最后这个提示，我是想避免被人拿这个接口测试用户存在与否
response.Response(ctx,http.StatusOK,60002,nil,"邀请码不存在")
return
}
好了，到这里我们基本就完成这个简单的小功能了，直接go run main.go 去亲自体验一下吧。
ok just this.see you next...
有疑问加站长微信联系（非本文作者））
