Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

gr-index-ratings.sql 3.7 KB

You have to be logged in to leave a comment. Sign In
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
  1. --- #dep gr-interactions
  2. --- #dep cluster
  3. --- #step Add interaction PK
  4. --- #allow invalid_table_definition
  5. ALTER TABLE gr.raw_interaction ADD CONSTRAINT gr_raw_interaction_pk PRIMARY KEY (gr_interaction_rid);
  6. --- #step Index book clusters
  7. CREATE MATERIALIZED VIEW IF NOT EXISTS gr.book_cluster
  8. AS SELECT DISTINCT gr_book_id, cluster
  9. FROM gr.book_isbn JOIN isbn_cluster USING (isbn_id);
  10. CREATE UNIQUE INDEX IF NOT EXISTS book_cluster_book_idx ON gr.book_cluster (gr_book_id);
  11. CREATE INDEX IF NOT EXISTS book_cluster_cluster_idx ON gr.book_cluster (cluster);
  12. ANALYZE gr.book_cluster;
  13. --- #step Extract user info
  14. CREATE TABLE IF NOT EXISTS gr.user_info (
  15. gr_user_rid SERIAL PRIMARY KEY,
  16. gr_user_id VARCHAR NOT NULL
  17. );
  18. CREATE TEMPORARY TABLE gr_new_users
  19. AS SELECT gr_interaction_data->>'user_id' AS gr_user_id
  20. FROM gr.raw_interaction LEFT JOIN gr.user_info ON (gr_user_id = gr_interaction_data->>'user_id')
  21. WHERE gr_user_rid IS NULL;
  22. INSERT INTO gr.user_info (gr_user_id)
  23. SELECT DISTINCT gr_user_id FROM gr_new_users;
  24. CREATE UNIQUE INDEX IF NOT EXISTS user_id_idx ON gr.user_info (gr_user_id);
  25. ANALYZE gr.user_info;
  26. --- #step Extract interaction data
  27. CREATE TABLE IF NOT EXISTS gr.interaction
  28. AS SELECT gr_interaction_rid, book_id AS gr_book_id, gr_user_rid, rating, (gr_interaction_data->'isRead')::boolean AS is_read, date_added, date_updated
  29. FROM gr.raw_interaction,
  30. jsonb_to_record(gr_interaction_data) AS
  31. x(book_id INTEGER, user_id VARCHAR, rating INTEGER,
  32. date_added TIMESTAMP WITH TIME ZONE, date_updated TIMESTAMP WITH TIME ZONE),
  33. gr.user_info
  34. WHERE user_id = gr_user_id;
  35. DROP VIEW IF EXISTS gr.work_interaction;
  36. CREATE VIEW gr.work_interaction
  37. AS SELECT ix.*, bid.gr_work_id
  38. FROM gr.interaction ix
  39. LEFT JOIN gr.book_ids bid USING (gr_book_id);
  40. --- #step Index interaction data
  41. --- #allow invalid_table_definition
  42. -- if this step is already done, first ALTER TABLE will fail
  43. ALTER TABLE gr.interaction ADD CONSTRAINT gr_interaction_pk PRIMARY KEY (gr_interaction_rid);
  44. CREATE INDEX IF NOT EXISTS interaction_book_idx ON gr.interaction (gr_book_id);
  45. CREATE INDEX IF NOT EXISTS interaction_user_idx ON gr.interaction (gr_user_rid);
  46. ALTER TABLE gr.interaction ADD CONSTRAINT gr_interaction_book_fk FOREIGN KEY (gr_book_id) REFERENCES gr.book_ids (gr_book_id);
  47. ALTER TABLE gr.interaction ADD CONSTRAINT gr_interaction_user_fk FOREIGN KEY (gr_user_rid) REFERENCES gr.user_info (gr_user_rid);
  48. ANALYZE gr.interaction;
  49. --- #step Extract ratings
  50. CREATE MATERIALIZED VIEW IF NOT EXISTS gr.rating
  51. AS SELECT gr_user_rid AS user_id, cluster AS book_id,
  52. MEDIAN(rating) AS rating,
  53. (array_agg(rating ORDER BY date_updated DESC))[1] AS last_rating,
  54. MEDIAN(EXTRACT(EPOCH FROM date_updated)) AS timestamp,
  55. COUNT(rating) AS nratings
  56. FROM gr.interaction
  57. JOIN gr.book_cluster USING (gr_book_id)
  58. WHERE rating > 0
  59. GROUP BY gr_user_rid, cluster;
  60. CREATE INDEX IF NOT EXISTS rating_user_idx ON gr.rating (user_id);
  61. CREATE INDEX IF NOT EXISTS rating_item_idx ON gr.rating (book_id);
  62. ANALYZE gr.rating;
  63. --- #step Extract add actions
  64. CREATE MATERIALIZED VIEW IF NOT EXISTS gr.add_action
  65. AS SELECT gr_user_rid AS user_id, cluster AS book_id,
  66. COUNT(rating) AS nactions,
  67. MIN(EXTRACT(EPOCH FROM date_updated)) AS first_time,
  68. MAX(EXTRACT(EPOCH FROM date_updated)) AS last_time
  69. FROM gr.interaction
  70. JOIN gr.book_cluster USING (gr_book_id)
  71. GROUP BY gr_user_rid, cluster;
  72. CREATE INDEX IF NOT EXISTS add_action_user_idx ON gr.add_action (user_id);
  73. CREATE INDEX IF NOT EXISTS add_action_item_idx ON gr.add_action (book_id);
  74. ANALYZE gr.add_action;
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...