package com.yuanchu.mom.utils;

import com.yuanchu.mom.mapper.SystemLogMapper;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;

@Component
public class GiveCode {

    @Resource
    private SystemLogMapper systemLogMapper;

    public String giveCode(String code, String tableName, String symbol, String patten){
        String date = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd"));
        String date2 = LocalDate.now().format(DateTimeFormatter.ofPattern(patten));
        int num = systemLogMapper.countRowsByNow(tableName, LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd"))) + 1;
        String nums = num + "";
        if(nums.length() == 1) nums = "00" + num;
        else if(nums.length() == 2) nums = "0" + num;
        return code + (date2==null?date:date2) + symbol + nums;
    }

}