EMsnap commented on code in PR #6203: URL: https://github.com/apache/inlong/pull/6203#discussion_r1005412527
########## inlong-agent/agent-plugins/src/main/java/org/apache/inlong/agent/plugin/sources/reader/OracleReader.java: ########## @@ -0,0 +1,230 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.inlong.agent.plugin.sources.reader; + +import org.apache.commons.codec.binary.Base64; +import org.apache.commons.lang3.CharUtils; +import org.apache.commons.lang3.StringUtils; +import org.apache.inlong.agent.conf.JobProfile; +import org.apache.inlong.agent.message.DefaultMessage; +import org.apache.inlong.agent.metrics.audit.AuditUtils; +import org.apache.inlong.agent.plugin.Message; +import org.apache.inlong.agent.utils.AgentDbUtils; +import org.apache.inlong.agent.utils.AgentUtils; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import java.nio.charset.StandardCharsets; +import java.sql.Connection; +import java.sql.PreparedStatement; +import java.sql.ResultSet; +import java.util.ArrayList; +import java.util.List; + +import static java.sql.Types.BINARY; +import static java.sql.Types.BLOB; +import static java.sql.Types.LONGVARBINARY; +import static java.sql.Types.VARBINARY; + +/** + * Read data from Oracle database by SQL + */ +public class OracleReader extends AbstractReader { + + public static final String ORACLE_READER_TAG_NAME = "AgentOracleMetric"; + public static final String JOB_DATABASE_USER = "job.oracleJob.user"; + public static final String JOB_DATABASE_PASSWORD = "job.oracleJob.password"; + public static final String JOB_DATABASE_HOSTNAME = "job.oracleJob.hostname"; + public static final String JOB_DATABASE_PORT = "job.oracleJob.port"; + public static final String JOB_DATABASE_SID = "job.oracleJob.sid"; + public static final String DEFAULT_JOB_DATABASE_SID = ""; + public static final String JOB_DATABASE_SERVICE_NAME = "job.oracleJob.serviceName"; + public static final String JOB_DATABASE_BATCH_SIZE = "job.oracleJob.batchSize"; + public static final int DEFAULT_JOB_DATABASE_BATCH_SIZE = 1000; + public static final String JOB_DATABASE_DRIVER_CLASS = "job.database.driverClass"; + public static final String DEFAULT_JOB_DATABASE_DRIVER_CLASS = "oracle.jdbc.driver.OracleDriver"; + public static final String STD_FIELD_SEPARATOR_SHORT = "\001"; + public static final String JOB_DATABASE_SEPARATOR = "job.sql.separator"; + // pre-set sql lines, commands like "set xxx=xx;" + public static final String JOB_DATABASE_TYPE = "job.database.type"; + public static final String ORACLE = "oracle"; + private static final Logger LOGGER = LoggerFactory.getLogger(SqlReader.class); + private static final String[] NEW_LINE_CHARS = new String[]{String.valueOf(CharUtils.CR), + String.valueOf(CharUtils.LF)}; + private static final String[] EMPTY_CHARS = new String[]{StringUtils.EMPTY, StringUtils.EMPTY}; + + private final String sql; + + private PreparedStatement preparedStatement; + private Connection conn; + private ResultSet resultSet; + private int columnCount; + + // column types + private String[] columnTypeNames; + private int[] columnTypeCodes; + private boolean finished = false; + private String separator; + + public OracleReader(String sql) { + this.sql = sql; + } + + @Override + public Message read() { + try { + if (!resultSet.next()) { + finished = true; + return null; + } + final List<String> lineColumns = new ArrayList<>(); + for (int i = 1; i <= columnCount; i++) { + final String dataValue; + /* handle special blob value, encode with base64, BLOB=2004 */ + final int typeCode = columnTypeCodes[i - 1]; + final String typeName = columnTypeNames[i - 1]; + + // binary type + if (typeCode == BLOB || typeCode == BINARY || typeCode == VARBINARY + || typeCode == LONGVARBINARY || typeName.contains("BLOB")) { + final byte[] data = resultSet.getBytes(i); + dataValue = new String(Base64.encodeBase64(data, false), StandardCharsets.UTF_8); + } else { + // non-binary type + dataValue = StringUtils.replaceEachRepeatedly(resultSet.getString(i), + NEW_LINE_CHARS, EMPTY_CHARS); + } + lineColumns.add(dataValue); + } + long dataSize = lineColumns.stream().mapToLong(column -> column.length()).sum(); + AuditUtils.add(AuditUtils.AUDIT_ID_AGENT_READ_SUCCESS, inlongGroupId, inlongStreamId, + System.currentTimeMillis(), 1, dataSize); + readerMetric.pluginReadSuccessCount.incrementAndGet(); + readerMetric.pluginReadCount.incrementAndGet(); + return generateMessage(lineColumns); + } catch (Exception ex) { + LOGGER.error("error while reading data", ex); + readerMetric.pluginReadFailCount.incrementAndGet(); + readerMetric.pluginReadCount.incrementAndGet(); + throw new RuntimeException(ex); + } + } + + private Message generateMessage(List<String> lineColumns) { + return new DefaultMessage(StringUtils.join(lineColumns, separator).getBytes(StandardCharsets.UTF_8)); + } + + @Override + public boolean isFinished() { + return finished; + } + + @Override + public String getReadSource() { + return sql; + } + + @Override + public void setReadTimeout(long mill) { + + } + + @Override + public void setWaitMillisecond(long millis) { + + } + + @Override + public String getSnapshot() { + return StringUtils.EMPTY; + } + + @Override + public void finishRead() { + destroy(); + } + + @Override + public boolean isSourceExist() { + return true; + } + + @Override + public void init(JobProfile jobConf) { + super.init(jobConf); + int batchSize = jobConf.getInt(JOB_DATABASE_BATCH_SIZE, DEFAULT_JOB_DATABASE_BATCH_SIZE); + String userName = jobConf.get(JOB_DATABASE_USER); Review Comment: finalize string -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: commits-unsubscr...@inlong.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
