forked from hrucinska/cwiczenia
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from pyspark.sql import SparkSession
|
|
import logging
|
|
import datetime
|
|
import time
|
|
import pyspark.sql.functions as f
|
|
|
|
spark = SparkSession \
|
|
.builder \
|
|
.appName("too_few_partitions") \
|
|
.enableHiveSupport() \
|
|
.getOrCreate()
|
|
|
|
log4jLogger = spark._jvm.org.apache.log4j
|
|
logger = log4jLogger.LogManager.getLogger(__name__)
|
|
|
|
try:
|
|
logger.info("SPARKAPP START")
|
|
start = datetime.datetime.fromtimestamp(time.time()).strftime('%Y-%m-%d %H:%M:%S')
|
|
|
|
orders = spark.sql("select * from hrucinska.uam_orders").withColumnRenamed("seller_id", "seller_id_order")
|
|
offers = spark.sql("select * from hrucinska.uam_offers").withColumnRenamed("offer_id", "offerId")
|
|
user_offers_orders = offers.join(orders, offers.seller_id == orders.seller_id_order, "left")
|
|
res = user_offers_orders.groupBy(user_offers_orders.seller_id).agg(f.count(user_offers_orders.offerId).alias("user_offers"), f.count(user_offers_orders.order_offer_id).alias("user_orders"))
|
|
res.write.mode("overwrite").saveAsTable("hrucinska.data_skew")
|
|
|
|
stop = datetime.datetime.fromtimestamp(time.time()).strftime('%Y-%m-%d %H:%M:%S')
|
|
|
|
print "czas rozpoczecia obliczen 1: ", start
|
|
print "czas zakonczenia obliczen1: ", stop
|
|
|
|
time.sleep(180)
|
|
|
|
except Exception as inst:
|
|
logger.info("SPARKAPP ERROR {0}".format(inst))
|
|
raise
|
|
finally:
|
|
logger.info("SPARKAPP STOP")
|