|
37 | 37 | @Data
|
38 | 38 | public class BaseFileSinkConfig implements DelimiterConfig, Serializable {
|
39 | 39 | private static final long serialVersionUID = 1L;
|
40 |
| - protected CompressFormat compressFormat = FileBaseSinkOptions.COMPRESS_CODEC.defaultValue(); |
41 |
| - protected String fieldDelimiter = FileBaseSinkOptions.FIELD_DELIMITER.defaultValue(); |
42 |
| - protected String rowDelimiter = FileBaseSinkOptions.ROW_DELIMITER.defaultValue(); |
43 |
| - protected int batchSize = FileBaseSinkOptions.BATCH_SIZE.defaultValue(); |
| 40 | + protected CompressFormat compressFormat = BaseSinkConfig.COMPRESS_CODEC.defaultValue(); |
| 41 | + protected String fieldDelimiter = BaseSinkConfig.FIELD_DELIMITER.defaultValue(); |
| 42 | + protected String rowDelimiter = BaseSinkConfig.ROW_DELIMITER.defaultValue(); |
| 43 | + protected int batchSize = BaseSinkConfig.BATCH_SIZE.defaultValue(); |
44 | 44 | protected String path;
|
45 |
| - protected String fileNameExpression = FileBaseSinkOptions.FILE_NAME_EXPRESSION.defaultValue(); |
46 |
| - protected boolean singleFileMode = FileBaseSinkOptions.SINGLE_FILE_MODE.defaultValue(); |
| 45 | + protected String fileNameExpression = BaseSinkConfig.FILE_NAME_EXPRESSION.defaultValue(); |
| 46 | + protected boolean singleFileMode = BaseSinkConfig.SINGLE_FILE_MODE.defaultValue(); |
47 | 47 | protected boolean createEmptyFileWhenNoData =
|
48 |
| - FileBaseSinkOptions.CREATE_EMPTY_FILE_WHEN_NO_DATA.defaultValue(); |
| 48 | + BaseSinkConfig.CREATE_EMPTY_FILE_WHEN_NO_DATA.defaultValue(); |
49 | 49 | protected FileFormat fileFormat;
|
50 |
| - protected String filenameExtension = FileBaseSinkOptions.FILENAME_EXTENSION.defaultValue(); |
| 50 | + protected String filenameExtension = BaseSinkConfig.FILENAME_EXTENSION.defaultValue(); |
51 | 51 | protected DateUtils.Formatter dateFormat = DateUtils.Formatter.YYYY_MM_DD;
|
52 | 52 | protected DateTimeUtils.Formatter datetimeFormat = DateTimeUtils.Formatter.YYYY_MM_DD_HH_MM_SS;
|
53 | 53 | protected TimeUtils.Formatter timeFormat = TimeUtils.Formatter.HH_MM_SS;
|
54 | 54 | protected Boolean enableHeaderWriter = false;
|
55 | 55 |
|
56 | 56 | public BaseFileSinkConfig(@NonNull Config config) {
|
57 |
| - if (config.hasPath(FileBaseSinkOptions.COMPRESS_CODEC.key())) { |
58 |
| - String compressCodec = config.getString(FileBaseSinkOptions.COMPRESS_CODEC.key()); |
| 57 | + if (config.hasPath(BaseSinkConfig.COMPRESS_CODEC.key())) { |
| 58 | + String compressCodec = config.getString(BaseSinkConfig.COMPRESS_CODEC.key()); |
59 | 59 | this.compressFormat = CompressFormat.valueOf(compressCodec.toUpperCase());
|
60 | 60 | }
|
61 |
| - if (config.hasPath(FileBaseSinkOptions.BATCH_SIZE.key())) { |
62 |
| - this.batchSize = config.getInt(FileBaseSinkOptions.BATCH_SIZE.key()); |
| 61 | + if (config.hasPath(BaseSinkConfig.BATCH_SIZE.key())) { |
| 62 | + this.batchSize = config.getInt(BaseSinkConfig.BATCH_SIZE.key()); |
63 | 63 | }
|
64 |
| - if (config.hasPath(FileBaseSinkOptions.FIELD_DELIMITER.key()) |
65 |
| - && StringUtils.isNotEmpty( |
66 |
| - config.getString(FileBaseSinkOptions.FIELD_DELIMITER.key()))) { |
67 |
| - this.fieldDelimiter = config.getString(FileBaseSinkOptions.FIELD_DELIMITER.key()); |
| 64 | + if (config.hasPath(BaseSinkConfig.FIELD_DELIMITER.key()) |
| 65 | + && StringUtils.isNotEmpty(config.getString(BaseSinkConfig.FIELD_DELIMITER.key()))) { |
| 66 | + this.fieldDelimiter = config.getString(BaseSinkConfig.FIELD_DELIMITER.key()); |
68 | 67 | }
|
69 | 68 |
|
70 |
| - if (config.hasPath(FileBaseSinkOptions.ROW_DELIMITER.key())) { |
71 |
| - this.rowDelimiter = config.getString(FileBaseSinkOptions.ROW_DELIMITER.key()); |
| 69 | + if (config.hasPath(BaseSinkConfig.ROW_DELIMITER.key())) { |
| 70 | + this.rowDelimiter = config.getString(BaseSinkConfig.ROW_DELIMITER.key()); |
72 | 71 | }
|
73 | 72 |
|
74 |
| - if (config.hasPath(FileBaseSinkOptions.FILE_PATH.key()) |
75 |
| - && !StringUtils.isBlank(config.getString(FileBaseSinkOptions.FILE_PATH.key()))) { |
76 |
| - this.path = config.getString(FileBaseSinkOptions.FILE_PATH.key()); |
| 73 | + if (config.hasPath(BaseSinkConfig.FILE_PATH.key()) |
| 74 | + && !StringUtils.isBlank(config.getString(BaseSinkConfig.FILE_PATH.key()))) { |
| 75 | + this.path = config.getString(BaseSinkConfig.FILE_PATH.key()); |
77 | 76 | }
|
78 | 77 | checkNotNull(path);
|
79 | 78 |
|
80 | 79 | if (path.equals(File.separator)) {
|
81 | 80 | this.path = "";
|
82 | 81 | }
|
83 | 82 |
|
84 |
| - if (config.hasPath(FileBaseSinkOptions.FILE_NAME_EXPRESSION.key()) |
| 83 | + if (config.hasPath(BaseSinkConfig.FILE_NAME_EXPRESSION.key()) |
85 | 84 | && !StringUtils.isBlank(
|
86 |
| - config.getString(FileBaseSinkOptions.FILE_NAME_EXPRESSION.key()))) { |
87 |
| - this.fileNameExpression = |
88 |
| - config.getString(FileBaseSinkOptions.FILE_NAME_EXPRESSION.key()); |
| 85 | + config.getString(BaseSinkConfig.FILE_NAME_EXPRESSION.key()))) { |
| 86 | + this.fileNameExpression = config.getString(BaseSinkConfig.FILE_NAME_EXPRESSION.key()); |
89 | 87 | }
|
90 | 88 |
|
91 |
| - if (config.hasPath(FileBaseSinkOptions.SINGLE_FILE_MODE.key())) { |
92 |
| - this.singleFileMode = config.getBoolean(FileBaseSinkOptions.SINGLE_FILE_MODE.key()); |
| 89 | + if (config.hasPath(BaseSinkConfig.SINGLE_FILE_MODE.key())) { |
| 90 | + this.singleFileMode = config.getBoolean(BaseSinkConfig.SINGLE_FILE_MODE.key()); |
93 | 91 | }
|
94 | 92 |
|
95 |
| - if (config.hasPath(FileBaseSinkOptions.CREATE_EMPTY_FILE_WHEN_NO_DATA.key())) { |
| 93 | + if (config.hasPath(BaseSinkConfig.CREATE_EMPTY_FILE_WHEN_NO_DATA.key())) { |
96 | 94 | this.createEmptyFileWhenNoData =
|
97 |
| - config.getBoolean(FileBaseSinkOptions.CREATE_EMPTY_FILE_WHEN_NO_DATA.key()); |
| 95 | + config.getBoolean(BaseSinkConfig.CREATE_EMPTY_FILE_WHEN_NO_DATA.key()); |
98 | 96 | }
|
99 | 97 |
|
100 |
| - if (config.hasPath(FileBaseSinkOptions.FILE_FORMAT_TYPE.key()) |
101 |
| - && !StringUtils.isBlank( |
102 |
| - config.getString(FileBaseSinkOptions.FILE_FORMAT_TYPE.key()))) { |
| 98 | + if (config.hasPath(BaseSinkConfig.FILE_FORMAT_TYPE.key()) |
| 99 | + && !StringUtils.isBlank(config.getString(BaseSinkConfig.FILE_FORMAT_TYPE.key()))) { |
103 | 100 | this.fileFormat =
|
104 | 101 | FileFormat.valueOf(
|
105 |
| - config.getString(FileBaseSinkOptions.FILE_FORMAT_TYPE.key()) |
| 102 | + config.getString(BaseSinkConfig.FILE_FORMAT_TYPE.key()) |
106 | 103 | .toUpperCase(Locale.ROOT));
|
107 | 104 | } else {
|
108 | 105 | // fall back to the default
|
109 |
| - this.fileFormat = FileBaseSinkOptions.FILE_FORMAT_TYPE.defaultValue(); |
| 106 | + this.fileFormat = BaseSinkConfig.FILE_FORMAT_TYPE.defaultValue(); |
110 | 107 | }
|
111 | 108 |
|
112 |
| - if (config.hasPath(FileBaseSinkOptions.FILENAME_EXTENSION.key()) |
| 109 | + if (config.hasPath(BaseSinkConfig.FILENAME_EXTENSION.key()) |
113 | 110 | && !StringUtils.isBlank(
|
114 |
| - config.getString(FileBaseSinkOptions.FILENAME_EXTENSION.key()))) { |
115 |
| - this.filenameExtension = config.getString(FileBaseSinkOptions.FILENAME_EXTENSION.key()); |
| 111 | + config.getString(BaseSinkConfig.FILENAME_EXTENSION.key()))) { |
| 112 | + this.filenameExtension = config.getString(BaseSinkConfig.FILENAME_EXTENSION.key()); |
116 | 113 | }
|
117 | 114 |
|
118 |
| - if (config.hasPath(FileBaseSinkOptions.DATE_FORMAT.key())) { |
| 115 | + if (config.hasPath(BaseSinkConfig.DATE_FORMAT.key())) { |
119 | 116 | dateFormat =
|
120 |
| - DateUtils.Formatter.parse( |
121 |
| - config.getString(FileBaseSinkOptions.DATE_FORMAT.key())); |
| 117 | + DateUtils.Formatter.parse(config.getString(BaseSinkConfig.DATE_FORMAT.key())); |
122 | 118 | }
|
123 | 119 |
|
124 |
| - if (config.hasPath(FileBaseSinkOptions.DATETIME_FORMAT.key())) { |
| 120 | + if (config.hasPath(BaseSinkConfig.DATETIME_FORMAT.key())) { |
125 | 121 | datetimeFormat =
|
126 | 122 | DateTimeUtils.Formatter.parse(
|
127 |
| - config.getString(FileBaseSinkOptions.DATETIME_FORMAT.key())); |
| 123 | + config.getString(BaseSinkConfig.DATETIME_FORMAT.key())); |
128 | 124 | }
|
129 | 125 |
|
130 |
| - if (config.hasPath(FileBaseSinkOptions.TIME_FORMAT.key())) { |
| 126 | + if (config.hasPath(BaseSinkConfig.TIME_FORMAT.key())) { |
131 | 127 | timeFormat =
|
132 |
| - TimeUtils.Formatter.parse( |
133 |
| - config.getString(FileBaseSinkOptions.TIME_FORMAT.key())); |
| 128 | + TimeUtils.Formatter.parse(config.getString(BaseSinkConfig.TIME_FORMAT.key())); |
134 | 129 | }
|
135 | 130 |
|
136 |
| - if (config.hasPath(FileBaseSinkOptions.ENABLE_HEADER_WRITE.key())) { |
137 |
| - enableHeaderWriter = config.getBoolean(FileBaseSinkOptions.ENABLE_HEADER_WRITE.key()); |
| 131 | + if (config.hasPath(BaseSinkConfig.ENABLE_HEADER_WRITE.key())) { |
| 132 | + enableHeaderWriter = config.getBoolean(BaseSinkConfig.ENABLE_HEADER_WRITE.key()); |
138 | 133 | }
|
139 | 134 | }
|
140 | 135 |
|
|
0 commit comments