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

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

Comments

Loading...