StringFormatter.java 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. package org.leumasjaffe.format;
  2. import java.util.HashMap;
  3. import java.util.Objects;
  4. import lombok.AccessLevel;
  5. import lombok.experimental.FieldDefaults;
  6. @FieldDefaults(level=AccessLevel.PRIVATE, makeFinal=true)
  7. public class StringFormatter {
  8. private static final class NameMap extends HashMap<String, Object> {
  9. /**
  10. *
  11. */
  12. private static final long serialVersionUID = -2243574068000774442L;
  13. NameMap(Object... args) {
  14. for (Object arg : args) {
  15. Objects.requireNonNull(arg);
  16. if (arg instanceof Named) {
  17. put(((Named) arg).getName(), arg);
  18. }
  19. }
  20. }
  21. }
  22. String fmt;
  23. public StringFormatter(String logFmtString) {
  24. this.fmt = logFmtString;
  25. }
  26. public String format(Object... args) {
  27. final StringBuilder str = new StringBuilder(fmt.length());
  28. int currentIdx = 0;
  29. final NameMap named = new NameMap(args);
  30. int lpos = 0;
  31. for (int pos = fmt.indexOf('{'); pos != -1; lpos = pos+1, pos = fmt.indexOf('{', lpos)) {
  32. str.append(fmt.substring(lpos, pos));
  33. int epos = fmt.indexOf('}', pos);
  34. if (fmt.charAt(pos+1) == '{') {
  35. str.append(fmt.substring(++pos, ++epos));
  36. } else if (epos == pos+1) {
  37. str.append(args[currentIdx++]);
  38. } else {
  39. final String token = fmt.substring(pos+1, epos);
  40. if (Character.isDigit(token.charAt(0))) {
  41. str.append(args[Integer.parseInt(token)]);
  42. } else { // In theory, I could add a format-mode here too
  43. str.append(named.get(token));
  44. }
  45. }
  46. pos = epos;
  47. }
  48. str.append(fmt.substring(lpos));
  49. return str.toString();
  50. }
  51. }