飞天班第11节：SpringBoot开发单体应用（中）

@Bean
public MessageSource messageSource(MessageSourceProperties properties) {
  ResourceBundleMessageSource messageSource = new ResourceBundleMessageSource();
  if (StringUtils.hasText(properties.getBasename())) {
    messageSource.setBasenames(StringUtils
                               .commaDelimitedListToStringArray(StringUtils.trimAllWhitespace(properties.getBasename())));
  }
  // 获取properties的编码信息
  if (properties.getEncoding() != null) {
    messageSource.setDefaultEncoding(properties.getEncoding().name());
  }
  messageSource.setFallbackToSystemLocale(properties.isFallbackToSystemLocale());
  Duration cacheDuration = properties.getCacheDuration();
  if (cacheDuration != null) {
    messageSource.setCacheMillis(cacheDuration.toMillis());
  }
  messageSource.setAlwaysUseMessageFormat(properties.isAlwaysUseMessageFormat());
  messageSource.setUseCodeAsDefaultMessage(properties.isUseCodeAsDefaultMessage());
  return messageSource;
}

@Bean
@ConditionalOnMissingBean // 不存在这个bean才生效
@ConditionalOnProperty(prefix = "spring.mvc", name = "locale")
public LocaleResolver localeResolver() {
  // 如果用户没配置国际化解析对象，就使用默认的
  if (this.mvcProperties.getLocaleResolver() == WebMvcProperties.LocaleResolver.FIXED) {
    return new FixedLocaleResolver(this.mvcProperties.getLocale());
  }
  // 否则，使用接受请求头关于国际化的对象
  AcceptHeaderLocaleResolver localeResolver = new AcceptHeaderLocaleResolver();
  localeResolver.setDefaultLocale(this.mvcProperties.getLocale());
  return localeResolver;
}

@Override
public Locale resolveLocale(HttpServletRequest request) {
  Locale defaultLocale = getDefaultLocale();
  // 从request请求中获取Accept-Language的语言信息
  if (defaultLocale != null && request.getHeader("Accept-Language") == null) {
    return defaultLocale; // 返回默认的国际化对象
  }
  Locale requestLocale = request.getLocale(); // 得到请求中的国际化信息
  List<Locale> supportedLocales = getSupportedLocales();
  if (supportedLocales.isEmpty() || supportedLocales.contains(requestLocale)) {
    return requestLocale; // 返回请求的国际化对象
  }
  Locale supportedLocale = findSupportedLocale(request, supportedLocales);
  if (supportedLocale != null) {
    return supportedLocale; // 返回支持的国际化对象，具体看源码中的findSupportedLocale方法
  }
  return (defaultLocale != null ? defaultLocale : requestLocale);
}

public Locale(String language) {
  this(language, "", "");
}
public Locale(String language, String country) {
  this(language, country, "");
}
/**
     * This method must be called only for creating the Locale.*
     * constants due to making shortcuts.
     */
private static Locale createConstant(String lang, String country) {
  BaseLocale base = BaseLocale.createInstance(lang, country);
  return getInstance(base, null);
}

public class MyLocaleResolver implements LocaleResolver {
	@Override
	public Locale resolveLocale(HttpServletRequest request) {
		String language = request.getParameter("l");
		Locale locale = Locale.getDefault(); // 如果我们没有配置，就使用默认的
		if(!StringUtils.isEmpty(language)){
			String[] split = language.split("_");
			locale = new Locale(split[0],split[1]); // 创建国际化对象
		}

		return locale;
	}

	@Override
	public void setLocale(HttpServletRequest request, HttpServletResponse response, Locale locale) {

	}
}

@Configuration
public class MyMvcConfig implements WebMvcConfigurer {

	@Override
	public void addViewControllers(ViewControllerRegistry registry) {
		// 视图跳转的控制！
		registry.addViewController("/index").setViewName("login2");
		registry.addViewController("/").setViewName("login2");
		registry.addViewController("/index.html").setViewName("login2");
	}

  // 注意bean的id 只能为localeResolver，因为Springboot会去扫描并识别
	@Bean
	public LocaleResolver localeResolver(){
		return new MyLocaleResolver();
	}
}

public enum RccErrorEnum implements ErrorInterface, EnumInterface<String>, BaseEnum {
    OK("000000", "操作成功"),
    FAIL("999999", "操作失败"),
    ID_NOT_NULL("PRT002", "主键id不能为空"),
    FOREIGN_KEY_NOT_NULL("PRT003", "外键不能为空"),
    HEAD_TABLE_ID_NOT_NULL("PRT004", "头表id不能为空"),
    ERROR_MESSAGE("errorMessage", "errorMessage"),
    ERROR_PERIOD_FORMAT("ERROR_PERIOD_FORMAT","账期格式错误"),
    ERROR_PERIOD_PARAM("ERROR_PERIOD_PARAM","账期不能为空"),
    ERROR_PAGE_PARAM("ERROR_PAGE_PARAM","分页参数不正确"),
    ERROR_NO_DETAILS("ERROR_NO_DETAILS","没有未汇总的明细"),
    ERROR_ON_SUM("ERROR_ON_SUM","汇总处理失败,批次号{0}")
    ;
    private String code;
    private String message;

    RccErrorEnum(String code, String message) {
        this.code = code;
        this.message = message;
    }
  
  public String getMessage() {
    return MessageUtils.getMessage(LanguageConstants.LANGUAGE_ENUM_MAP.get(InitConfig.MODULE_LANGUAGE).getLocale(),
                                   "i18n/fsc_message", this.code);
  }
  
      @Override
    public String getMessage(Object... msgParams) {
        String message = MessageUtils.getMessage(LanguageConstants.LANGUAGE_ENUM_MAP.get(InitConfig.MODULE_LANGUAGE).getLocale(),
                "i18n/fsc_message", this.code, msgParams);
        String substring = message.substring(0, 4);
        if("系统错误".equals(substring)){
            message = message.substring(4);
        }
        return message;
    }
} 

public interface SaleReceiptRecFacade {
      static void assertParam(CommonRequest<CreateSrDetailReqDTO> request){
        Assert.isTrue(StringUtils.isNotBlank(request.getHeadParams().getTenantCode()), RccErrorEnum.ERROR_TENANT_PARAM.getMessage());
        Assert.isTrue(StringUtils.isNotBlank(request.getRestParams().getMerchantCode()), RccErrorEnum.ERROR_MERCHANT_CODE_PARAM.getMessage());
        Assert.isTrue(StringUtils.isNotBlank(request.getRestParams().getMerchantName()), RccErrorEnum.ERROR_MERCHANT_NAME_PARAM.getMessage());
        Assert.isTrue(StringUtils.isNotBlank(request.getRestParams().getPeriod()), RccErrorEnum.ERROR_PERIOD_PARAM.getMessage());
    }
}

static String getPeriodEnd(String period){
  try {
    Date dateStart = DateUtil.parse(period + "-01", "yyyy-MM-dd").toJdkDate();
    DateTime dateTime = DateUtil.endOfMonth(dateStart);
    return dateTime.toDateStr();
  } catch (Exception e) {
    LogFactory.get().error("账期格式错误",e);
    throw new BussinessException(RccErrorEnum.ERROR_PERIOD_FORMAT);
  }
}

public BussinessException(ErrorInterface errorEnum, Object... msgParam) {
  this(errorEnum, errorEnum.getMessage(msgParam));
}

private BussinessException(ErrorInterface errorEnum, String errorMsg) {
  super(errorMsg);
  this.printLog = 0;
  this.rootCauseMsgList = new ArrayList();
  this.errorCode = errorEnum.getMsgKey();
  this.errorMsg = errorMsg;
  this.appendRootCauseMsg(errorEnum, errorMsg);
}

@Configuration
@EnableWebSecurity
public class SecurityConfig extends WebSecurityConfigurerAdapter {
	// 授权规则
	@Override
	protected void configure(HttpSecurity http) throws Exception {
		// htpp的方法基本分为4个类别：过滤器，登录注销规则，安全配置，OAuth2配置
		// 一般我们只配置规则即可
		// 首页允许所有人访问
		http.authorizeRequests().antMatchers("/","/toLogin").permitAll()
				.antMatchers("/level1/**").hasRole("vip1")
				.antMatchers("/level2/**").hasRole("vip2")
				.antMatchers("/level3/**").hasRole("vip3")
				.antMatchers("/*").authenticated();

//		表单登录用户和密码的默认参数名是username和password
		http.formLogin()
				//.defaultSuccessUrl("/dashboard") // 设置默认登录成功后跳转的页面
				.loginProcessingUrl("/login") // 登陆表单提交请求,默认的提交登录请求是/login(Post请求),认证失败默认会跳转到/login?error(Get请求)
				.loginPage("/toLogin"); // 默认的登录页请求是/login(Get请求)

		// 注销,默认开启的注销功能
		http.logout().logoutSuccessUrl("/");    // 注销成功跳转到首页

		// 自定义的登录页需要配置 rememberMe 的参数名,就可以绑定到我们前端的！
		// 记住我功能
		http.rememberMe().rememberMeParameter("remember");
	}

@PostMapping("/user/login")
public String login(@RequestParam("username") String username,
                    @RequestParam("password") String password,
                    Model model, HttpSession session){

  // 真实的操作应该去数据库中获取数据
  if (!StringUtils.isEmpty(username) && "123456".equals(password)){
    // 登录成功，放入Session
    session.setAttribute("loginUser",username);
    System.out.println(session.getAttribute("LoginUser"));
    return "redirect:/main.html"; // 转发和重定向！
  }else {
    model.addAttribute("msg", "用户名和密码输入错误");
    return "login";
  }
}

public class LoginInterceptor implements HandlerInterceptor {
  // false  拦截
  // true 通过放行
  @Override
  public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

    String user = (String) request.getSession().getAttribute("loginUser");
    System.out.println(user);
    System.out.println("=====LoginInterceptor=====");
    if (user==null){ // 未登录，不放行，并且跳转到登录页！
      request.setAttribute("msg","未登录，请先登录！");
      // response.sendRedirect("/") ; 重定向
      request.getRequestDispatcher("/index.html").forward(request,response); // 转发
      return false;
    }else {
      // 放行
      return true;
    }
  }
}

@Configuration
//@EnableWebMvc
public class MyMvcConfig implements WebMvcConfigurer {

  @Override
  public void addViewControllers(ViewControllerRegistry registry) {
    // 视图跳转的控制！
    registry.addViewController("/index").setViewName("index");
    registry.addViewController("/").setViewName("index");
    registry.addViewController("/index.html").setViewName("index");
    registry.addViewController("/toLogin").setViewName("login2");
  }

  // 注册拦截器
  @Override
  public void addInterceptors(InterceptorRegistry registry) {
    // 拦截器需要放行第一次登录的请求，否则就用户为null 了
    registry.addInterceptor(new LoginInterceptor())
      .addPathPatterns("/**")
      .excludePathPatterns("/index.html","/","index","/asserts/**","/toLogin","/user/login");
    // 放开首页、登录页、登录请求的拦截
  }

  // 注册我们的国际化组件
  // id localeResolver只能为这个，因为SpringBoot会去扫描并识别
  @Bean
  public LocaleResolver localeResolver(){
    return new MyLocaleResolver();
  }
}

// 跳转到员工新增页面
@GetMapping("/add")
public String addEmployee(Model model){
  // 应该要携带部门的数据信息
  List<Department> departmentList = departmentMapper.listDepartment();
  model.addAttribute("departmentList",departmentList);
  return "emp/add";
}

@PostMapping("/save")
public String save(Employee employee){
  employeeMapper.save(employee);
  // 添加完毕后应该返回首页，重定向url会变化，请求转发url不会变化
  return "redirect:/employee/list";
}

// 跳转到员工修改页面
@GetMapping("/{id}")
public String toUpdateEmp(@PathVariable("id") Integer id,Model model){
  // 回显出用户的信息
  Employee employee = employeeMapper.get(id);
  model.addAttribute("employee",employee);
  // 应该要携带部门的数据信息
  List<Department> departments = departmentMapper.listDepartment();
  model.addAttribute("departments", departments);
  return "emp/update";
}

// 提交修改员工
@PostMapping("update")
public String update(Employee employee){
  employeeMapper.update(employee);
  return "redirect:/employee/list";
}

// 提交删除
@GetMapping("/delete/{id}")
public String delete(@PathVariable("id") Integer id){
  employeeMapper.delete(id);
  return "redirect:/employee/list";
}