用户类 邀请码 商城
Change-Id: If3a9ef5c464386647ae3b876104cbf4c71265c4b
diff --git a/src/main/java/com/pt5/pthouduan/service/InviteService.java b/src/main/java/com/pt5/pthouduan/service/InviteService.java
index f6ca8a6..a97929c 100644
--- a/src/main/java/com/pt5/pthouduan/service/InviteService.java
+++ b/src/main/java/com/pt5/pthouduan/service/InviteService.java
@@ -2,27 +2,28 @@
import com.pt5.pthouduan.entity.Invites;
import com.pt5.pthouduan.mapper.InvitesMapper;
+import com.pt5.pthouduan.mapper.UserMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.HashMap;
+import java.util.List;
import java.util.Map;
@Service
public class InviteService {
@Autowired
private InvitesMapper invitesMapper;
- //批量生成邀请码并存入数据库（6位）(这个直接在数据库更新)
- //邀请码卖出（set buyername）
- public Map<String, Object> setbuyername(String buyername){
+ //获取“我的”邀请码
+ public Map<String, Object> getInvitesByUsername(String username){
Map<String, Object> result = new HashMap<>();
- Invites invite=invitesMapper.selectFirstEmptyBuyername();
- invitesMapper.updatebuyer(invite.getCode(),buyername);
+ List<Map<String, Object>> invites = invitesMapper.selectInvitesByUsername(username);
result.put("success", true);
- result.put("invitecode", invite.getCode());
- result.put("message", "邀请码购买成功");
+ result.put("invites", invites);
+ result.put("message", "获取成功");
return result;
}
+
}