deemoliu commented on code in PR #10927:
URL: https://github.com/apache/pinot/pull/10927#discussion_r1292535583


##########
pinot-segment-local/src/main/java/org/apache/pinot/segment/local/upsert/PartialUpsertHandler.java:
##########
@@ -60,29 +63,59 @@ public PartialUpsertHandler(Schema schema, Map<String, 
UpsertConfig.Strategy> pa
    * For example, overwrite merger will only override the prev value if the 
new value is not null.
    * Null values will override existing values if not configured. They can be 
ignored by using ignoreMerger.
    *
-   * @param previousRecord the last derived full record during ingestion.
+   * @param indexSegment the segment of the last derived full record during 
ingestion.
+   * @param docId the docId of the last derived full record during ingestion 
in the segment.
    * @param newRecord the new consumed record.
-   * @return a new row after merge
    */
-  public GenericRow merge(GenericRow previousRecord, GenericRow newRecord) {
-    for (String column : previousRecord.getFieldToValueMap().keySet()) {
+  public void merge(IndexSegment indexSegment, int docId, GenericRow 
newRecord) {
+    for (String column: indexSegment.getColumnNames()) {
       if (!_primaryKeyColumns.contains(column)) {
-        if (!previousRecord.isNullValue(column)) {
+        PartialUpsertMerger merger = _column2Mergers.getOrDefault(column, 
_defaultPartialUpsertMerger);
+        // Non-overwrite mergers
+        // (1) If the value of the previous is null value, skip merging and 
use the new value
+        // (2) Else If the value of new value is null, use the previous value 
(even for comparison columns).
+        // (3) Else If the column is not a comparison column, we applied the 
merged value to it.
+        if (!(merger instanceof OverwriteMerger)) {
+          PinotSegmentColumnReader pinotSegmentColumnReader = new 
PinotSegmentColumnReader(indexSegment, column);

Review Comment:
   thanks for suggestion, i updated the PR with try-with-resource



-- 
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...@pinot.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org
For additional commands, e-mail: commits-h...@pinot.apache.org

Reply via email to