JAVA实现生成网站邀请码

这篇文章将为大家详细讲解有关JAVA实现生成网站邀请码，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

1. 数据库设计

创建名为 InvitationCode 的数据库表，其中包含以下列：

| 列名 | 数据类型 | 主键 | 非空 |
|---|---|---|---|
| id | int | 是 | 是 |
| code | varchar(255) | 是 | 是 |
| created_at | timestamp | 是 | 是 |
| expired_at | timestamp | 是 | 是 |

2. Java 实现

2.1. 实体类

创建 InvitationCode 实体类来表示数据库表：

@Entity
@Table(name = "invitation_code")
public class InvitationCode {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @Column(name = "code", unique = true)
    private String code;

    @Column(name = "created_at")
    private LocalDateTime createdAt;

    @Column(name = "expired_at")
    private LocalDateTime expiredAt;

    // 省略 getter 和 setter 方法
}

2.2. 仓库接口

定义 InvitationCodeRepository 仓库接口：

public interface InvitationCodeRepository extends JpaRepository<InvitationCode, Long> {

    InvitationCode findByCode(String code);

    boolean existsByCode(String code);
}

2.3. 服务类

创建 InvitationCodeService 服务类：

@Service
public class InvitationCodeService {

    @Autowired
    private InvitationCodeRepository invitationCodeRepository;

    public InvitationCode generateInvitationCode() {
        String code = generateRandomCode();
        InvitationCode invitationCode = new InvitationCode();
        invitationCode.setCode(code);
        invitationCode.setCreatedAt(LocalDateTime.now());
        invitationCode.setExpiredAt(LocalDateTime.now().plusDays(1));
        return invitationCodeRepository.save(invitationCode);
    }

    private String generateRandomCode() {
        // 这里可以根据需要定制随机码生成逻辑
        return UUID.randomUUID().toString();
    }

    public InvitationCode getInvitationCodeByCode(String code) {
        return invitationCodeRepository.findByCode(code);
    }

    public boolean isInvitationCodeValid(String code) {
        return invitationCodeRepository.existsByCode(code) && !hasExpired(code);
    }

    private boolean hasExpired(String code) {
        InvitationCode invitationCode = invitationCodeRepository.findByCode(code);
        return invitationCode != null && invitationCode.getExpiredAt().isBefore(LocalDateTime.now());
    }

    // 省略其他方法
}

2.4. 控制器

创建 InvitationCodeController 控制器：

@RestController
@RequestMapping("/api/invitation-codes")
public class InvitationCodeController {

    @Autowired
    private InvitationCodeService invitationCodeService;

    @PostMapping
    public InvitationCode generateInvitationCode() {
        return invitationCodeService.generateInvitationCode();
    }

    @GetMapping("/{code}")
    public InvitationCode getInvitationCodeByCode(@PathVariable String code) {
        return invitationCodeService.getInvitationCodeByCode(code);
    }

    // 省略其他方法
}

3. 使用示例

生成邀请码：

InvitationCode invitationCode = invitationCodeService.generateInvitationCode();

检查邀请码是否有效：

boolean valid = invitationCodeService.isInvitationCodeValid(code);

获取邀请码信息：

InvitationCode invitationCode = invitationCodeService.getInvitationCodeByCode(code);

以上就是JAVA实现生成网站邀请码的详细内容，更多请关注编程学习网其它相关文章！