io.github.taills.common.security.controller.AuthenticationController Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-boot-common Show documentation

Taills's Spring Boot Project Common

There is a newer version: 1.3.0

package io.github.taills.common.security.controller;

import io.github.taills.common.annotation.ApiResponseBody;
import io.github.taills.common.exception.ExceptionManager;
import io.github.taills.common.jpa.entity.SecurityUser;
import io.github.taills.common.jpa.service.SecurityUserService;
import io.github.taills.common.response.ApiResult;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;

import java.util.Optional;

/**
 * @ClassName AuthenticationController
 * @Description
 * @Author nil
 * @Date 2021/10/22 10:47 下午
 **/
@RestController
@ApiResponseBody
@RequestMapping("/security")
@Slf4j
@Api(tags = {"用户登录"})
public class AuthenticationController {

    @Autowired
    private SecurityUserService userService;


    @PostMapping("/login")
    @ApiOperation("登录")
    public ApiResult login(@RequestParam("username") String username, @RequestParam("password") String password) {
        Optional optional;
        optional = this.userService.verifyUsernameAndPassword(username, password);
        if (optional.isPresent()) {
            return ApiResult.success(this.userService.issueToken(optional.get()));
        } else {
            throw ExceptionManager.create(1001);
        }
    }

    @PostMapping("/admin/register")
    @ApiOperation(value = "注册管理员账号",notes = "仅在数据库为空时可以使用")
    @Transactional
    public ApiResult registerAdmin(@RequestParam("username") String username, @RequestParam("password") String password) {
        if (userService.count() > 0) {
            throw ExceptionManager.create(1002);
        }
        SecurityUser securityUser = new SecurityUser();
        securityUser.setUsername(username);
        securityUser.setPassword(password);
        securityUser.setIsEnabled(true);
        securityUser.setNickname(username);
        return ApiResult.success(userService.registerAdmin(securityUser));
    }
}