浏览代码

Merge branch 'dev-xym-add-weekday' of algorithm/recommend-emr-dataprocess into feature/20250104-zt-update

xueyiming 2 天之前
父节点
当前提交
b8905eef76

+ 7 - 1
src/main/scala/com/aliyun/odps/spark/examples/makedata_ad/v20240718/makedata_ad_33_bucketDataFromOriginToHive_20250228.scala

@@ -84,7 +84,8 @@ object makedata_ad_33_bucketDataFromOriginToHive_20250228 {
       "user_adverid_conver_3d", "user_adverid_conver_7d", "user_adverid_conver_30d",
       "user_skuid_view_3d", "user_skuid_view_7d", "user_skuid_view_30d",
       "user_skuid_click_3d", "user_skuid_click_7d", "user_skuid_click_30d",
-      "user_skuid_conver_3d", "user_skuid_conver_7d", "user_skuid_conver_30d")
+      "user_skuid_conver_3d", "user_skuid_conver_7d", "user_skuid_conver_30d",
+      "is_weekday", "day_of_the_week")
 
 
     // 2 读取odps+表信息
@@ -137,6 +138,9 @@ object makedata_ad_33_bucketDataFromOriginToHive_20250228 {
               val tsHour = zonedDateTime.getHour()
               // 当前小时-刻钟(15分钟一个间隔,0~95)
               val tsHourQuarter = zonedDateTime.getMinute() / 15 + zonedDateTime.getHour() * 4
+              // 获取星期几(1=周一,7=周日)
+              val dayOfTheWeek = zonedDateTime.getDayOfWeek.getValue()
+              val isWeekday = if (dayOfTheWeek <= 5) 1 else 2
               val cid = record.getString("cid")
               val mid = record.getString("mid")
               val pqtid = record.getString("pqtid")
@@ -148,6 +152,8 @@ object makedata_ad_33_bucketDataFromOriginToHive_20250228 {
               featureMap.put("pqtid", pqtid)
               featureMap.put("hour", tsHour)
               featureMap.put("hour_quarter", tsHourQuarter)
+              featureMap.put("is_weekday", isWeekday)
+              featureMap.put("day_of_the_week", dayOfTheWeek)
 
               val extend: JSONObject = if (record.isNull("extend")) new JSONObject() else
                 JSON.parseObject(record.getString("extend"))