xueyiming 3 днів тому
батько
коміт
1d7a0fc744

+ 4 - 3
src/main/scala/com/aliyun/odps/spark/examples/makedata_ad/v20240718/makedata_ad_33_bucketDataFromOriginToHive_20250228.scala

@@ -139,7 +139,8 @@ object makedata_ad_33_bucketDataFromOriginToHive_20250228 {
               // 当前小时-刻钟(15分钟一个间隔,0~95)
               val tsHourQuarter = zonedDateTime.getMinute() / 15 + zonedDateTime.getHour() * 4
               // 获取星期几(1=周一,7=周日)
-              val dayOfWeek = zonedDateTime.getDayOfWeek.getValue()
+              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")
@@ -151,8 +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", dayOfWeek)
+              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"))