Pull Request #5103
Add referral codes to user signup
Generates a unique 8-character referralCode for every new user at signup time. Referral credits are applied when a new user signs up using an existing code.
Ready to merge
service/ReferralService.java+34 additions
Viewed1
+ @Service
2
+ public class ReferralService {
3
+ @Autowired private UserRepository userRepo;
4
+
5
+ public String generateUniqueCode() {
6
+ String code;
7
+ do {
8
+ code = RandomStringUtils.randomAlphanumeric(8).toUpperCase();
9
+ } while (userRepo.existsByReferralCode(code));
10
+ return code;
11
+ }
12
+
13
+ @Transactional
14
+ public void assignReferralCode(User user) {
15
+ String code = generateUniqueCode();
16
+ user.setReferralCode(code);
17
+ userRepo.save(user);
18
+ }
19
+ }
ML
meera_lead
Approved review
The uniqueness check before insert is a nice touch. LGTM.