Merge pull request #23 from aliyun/feature/fix_ots_seq

fix: handle empty playtime field in getting sequence features
aliyun · Jul 11, 2024 · 60a437e · 60a437e
2 parents 5068426 + f14441e
commit 60a437e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dao/feature_view_tablestore_dao.go b/dao/feature_view_tablestore_dao.go
@@ -161,7 +161,11 @@ func (d *FeatureViewTableStoreDao) GetUserSequenceFeature(keys []interface{}, us
 				rangeRowQueryCriteria.StartPrimaryKey = startPK
 				rangeRowQueryCriteria.EndPrimaryKey = endPK
 				rangeRowQueryCriteria.Direction = tablestore.FORWARD
-				rangeRowQueryCriteria.ColumnsToGet = []string{sequenceConfig.ItemIdField, sequenceConfig.EventField, sequenceConfig.PlayTimeField, sequenceConfig.TimestampField}
+				if sequenceConfig.PlayTimeField == "" {
+					rangeRowQueryCriteria.ColumnsToGet = []string{sequenceConfig.ItemIdField, sequenceConfig.EventField, sequenceConfig.TimestampField}
+				} else {
+					rangeRowQueryCriteria.ColumnsToGet = []string{sequenceConfig.ItemIdField, sequenceConfig.EventField, sequenceConfig.PlayTimeField, sequenceConfig.TimestampField}
+				}
 				timeRange := new(tablestore.TimeRange)
 				timeRange.End = currTime * 1000
 				timeRange.Start = (currTime - 86400*5) * 1000