🔨 perf(core): 时间格式化工具支持自定义设置单位

This commit is contained in:
bwcx_jzy 2024-06-14 15:54:04 +08:00
parent c8a1d67726
commit 24a4c479c4
2 changed files with 86 additions and 11 deletions

View File

@ -3,6 +3,7 @@ package cn.hutool.core.date;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import java.io.Serializable; import java.io.Serializable;
import java.util.function.Function;
/** /**
* 时长格式化器用于格式化输出两个日期相差的时长<br> * 时长格式化器用于格式化输出两个日期相差的时长<br>
@ -18,6 +19,10 @@ import java.io.Serializable;
public class BetweenFormatter implements Serializable { public class BetweenFormatter implements Serializable {
private static final long serialVersionUID = 1L; private static final long serialVersionUID = 1L;
/**
* 单位格式化器
*/
public static Function<Level, String> DEFAULT_LEVEL_FORMATTER = (level) -> level.name;
/** /**
* 时长毫秒数 * 时长毫秒数
*/ */
@ -30,6 +35,14 @@ public class BetweenFormatter implements Serializable {
* 格式化级别的最大个数 * 格式化级别的最大个数
*/ */
private final int levelMaxCount; private final int levelMaxCount;
/**
* 格式化器
*/
private Function<Level, String> levelFormatter = DEFAULT_LEVEL_FORMATTER;
/**
* 分隔符
*/
private String separator = StrUtil.EMPTY;
/** /**
* 构造 * 构造
@ -74,31 +87,34 @@ public class BetweenFormatter implements Serializable {
int levelCount = 0; int levelCount = 0;
if (isLevelCountValid(levelCount) && 0 != day && level >= Level.DAY.ordinal()) { if (isLevelCountValid(levelCount) && 0 != day && level >= Level.DAY.ordinal()) {
sb.append(day).append(Level.DAY.name); sb.append(day).append(levelFormatter.apply(Level.DAY)).append(separator);
levelCount++; levelCount++;
} }
if (isLevelCountValid(levelCount) && 0 != hour && level >= Level.HOUR.ordinal()) { if (isLevelCountValid(levelCount) && 0 != hour && level >= Level.HOUR.ordinal()) {
sb.append(hour).append(Level.HOUR.name); sb.append(hour).append(levelFormatter.apply(Level.HOUR)).append(separator);
levelCount++; levelCount++;
} }
if (isLevelCountValid(levelCount) && 0 != minute && level >= Level.MINUTE.ordinal()) { if (isLevelCountValid(levelCount) && 0 != minute && level >= Level.MINUTE.ordinal()) {
sb.append(minute).append(Level.MINUTE.name); sb.append(minute).append(levelFormatter.apply(Level.MINUTE)).append(separator);
levelCount++; levelCount++;
} }
if (isLevelCountValid(levelCount) && 0 != second && level >= Level.SECOND.ordinal()) { if (isLevelCountValid(levelCount) && 0 != second && level >= Level.SECOND.ordinal()) {
sb.append(second).append(Level.SECOND.name); sb.append(second).append(levelFormatter.apply(Level.SECOND)).append(separator);
levelCount++; levelCount++;
} }
if (isLevelCountValid(levelCount) && 0 != millisecond && level >= Level.MILLISECOND.ordinal()) { if (isLevelCountValid(levelCount) && 0 != millisecond && level >= Level.MILLISECOND.ordinal()) {
sb.append(millisecond).append(Level.MILLISECOND.name); sb.append(millisecond).append(levelFormatter.apply(Level.MILLISECOND)).append(separator);
// levelCount++; // levelCount++;
} }
} }
if (StrUtil.isEmpty(sb)) { if (StrUtil.isEmpty(sb)) {
sb.append(0).append(this.level.name); sb.append(0).append(levelFormatter.apply(this.level));
} else {
if (StrUtil.isNotEmpty(separator)) {
sb.delete(sb.length() - separator.length(), sb.length());
}
} }
return sb.toString(); return sb.toString();
} }
@ -138,6 +154,29 @@ public class BetweenFormatter implements Serializable {
this.level = level; this.level = level;
} }
/**
* 设置级别格式化器
*
* @param levelFormatter 级别格式化器
* @return this
*/
public BetweenFormatter setLevelFormatter(Function<Level, String> levelFormatter) {
this.levelFormatter = levelFormatter;
return this;
}
/**
* 设置分隔符
*
* @param separator 分割符
* @return this
*/
public BetweenFormatter setSeparator(String separator) {
this.separator = separator == null ? StrUtil.EMPTY : separator;
return this;
}
/** /**
* 格式化等级枚举 * 格式化等级枚举
* *

View File

@ -4,8 +4,27 @@ import cn.hutool.core.date.BetweenFormatter.Level;
import org.junit.Assert; import org.junit.Assert;
import org.junit.Test; import org.junit.Test;
import java.util.function.Function;
public class BetweenFormatterTest { public class BetweenFormatterTest {
Function<Level, String> levelFormatterEn = level -> {
switch (level) {
case DAY:
return " day";
case HOUR:
return " hour";
case MINUTE:
return " minute";
case SECOND:
return " second";
case MILLISECOND:
return " millisecond";
default:
return " " + level.name();
}
};
@Test @Test
public void formatTest() { public void formatTest() {
long betweenMs = DateUtil.betweenMs(DateUtil.parse("2017-01-01 22:59:59"), DateUtil.parse("2017-01-02 23:59:58")); long betweenMs = DateUtil.betweenMs(DateUtil.parse("2017-01-01 22:59:59"), DateUtil.parse("2017-01-02 23:59:58"));
@ -13,6 +32,23 @@ public class BetweenFormatterTest {
Assert.assertEquals(formater.toString(), "1天"); Assert.assertEquals(formater.toString(), "1天");
} }
@Test
public void formatTestEn() {
final long betweenMs = DateUtil.betweenMs(DateUtil.parse("2017-01-01 22:59:59"), DateUtil.parse("2017-01-02 23:59:58"));
final BetweenFormatter formatter = new BetweenFormatter(betweenMs, BetweenFormatter.Level.MILLISECOND, 1);
formatter.setLevelFormatter(levelFormatterEn);
Assert.assertEquals(formatter.toString(), "1 day");
}
@Test
public void formatTestEn2() {
final long betweenMs = 3610001;
final BetweenFormatter formatter = new BetweenFormatter(betweenMs, BetweenFormatter.Level.MILLISECOND, 5);
formatter.setSeparator(",");
formatter.setLevelFormatter(levelFormatterEn);
Assert.assertEquals(formatter.toString(), "1 hour,10 second,1 millisecond");
}
@Test @Test
public void formatBetweenTest() { public void formatBetweenTest() {
long betweenMs = DateUtil.betweenMs(DateUtil.parse("2018-07-16 11:23:19"), DateUtil.parse("2018-07-16 11:23:20")); long betweenMs = DateUtil.betweenMs(DateUtil.parse("2018-07-16 11:23:19"), DateUtil.parse("2018-07-16 11:23:20"));