From 740caf8fd9f605802150b6f403138868157bad12 Mon Sep 17 00:00:00 2001 From: Omar Roth Date: Thu, 30 Nov 2017 18:23:16 -0600 Subject: [PATCH] Update shard.yml and fix postgres queries --- shard.yml | 2 +- src/invidious.cr | 23 +++++++++++++---------- 2 files changed, 14 insertions(+), 11 deletions(-) diff --git a/shard.yml b/shard.yml index 9831ff5e..b2f4d3a9 100644 --- a/shard.yml +++ b/shard.yml @@ -16,6 +16,6 @@ dependencies: github: will/crystal-pg branch: crystalv024 -crystal: 0.24.0 +crystal: 0.23.1 license: MIT diff --git a/src/invidious.cr b/src/invidious.cr index e8df4e62..2cb70915 100644 --- a/src/invidious.cr +++ b/src/invidious.cr @@ -136,7 +136,7 @@ get "/" do |env| templated "index" end -def update_record(context, pg, video_id) +def get_record(context, video_id) client = HTTP::Client.new("www.youtube.com", 443, context) video_info = client.get("/get_video_info?video_id=#{video_id}&el=info&ps=default&eurl=&gl=US&hl=en").body info = HTTP::Params.parse(video_info) @@ -161,11 +161,8 @@ def update_record(context, pg, video_id) dislikes = 1 end - pg.exec("update invidious set last_updated = $1, video_info = $2, video_html = $3, views = $4, likes = $5,\ - dislikes = $6, rating = $7 where video_id = $8", - Time.now, video_info, video_html, views, likes, dislikes, rating, video_id) - - return {Time.now, video_id, video_info, video_html, views, likes, dislikes, rating} + args = {Time.now, video_id, video_info, video_html, views, likes, dislikes, rating} + return args end get "/watch/:video_id" do |env| @@ -176,11 +173,16 @@ get "/watch/:video_id" do |env| video_id, as: {Time, String, String, String, Int64, Int32, Int32, Float64}) - # If record was last updated less than 5 minutes ago, use data, otherwise refresh + # If record was last updated more than 5 minutes ago or doesn't exist, refresh if video_record.nil? - video_record = update_record(context, pg, video_id) + video_record = get_record(context, video_id) + pg.exec("insert into invidious values ($1, $2, $3, $4, $5, $6, $7, $8)", + get_record(context, video_id).to_a) elsif Time.now.epoch - video_record[0].epoch > 300 - video_record = update_record(context, pg, video_id) + video_record = get_record(context, video_id) + pg.exec("update invidious set last_updated = $1, video_info = $2, video_html = $3,\ + views = $4, likes = $5, dislikes = $6, rating = $7 where video_id = $8", + video_record.to_a) end video_info = HTTP::Params.parse(video_record[2]) @@ -200,7 +202,8 @@ get "/watch/:video_id" do |env| likes = likes.to_f dislikes = dislikes.to_f views = views.to_f - engagement = (((dislikes + likes)*100)/views).significant(2) + + engagement = ((dislikes + likes)/views * 100).significant(2) calculated_rating = likes/(likes + dislikes) * 4 + 1 likes = likes.to_s