579597e3 |
1 | (in-package :meta-model) |
2 | |
3 | (eval-when (:compile-toplevel :load-toplevel :execute) |
4 | (use-package :clsql)) |
5 | |
6 | (export 'def-meta-model-from-table) |
7 | (export 'def-meta-models) |
8 | (export 'def-view-class/meta) |
9 | (export 'list-base-classes) |
10 | |
11 | (defparameter *clsql-base-classes* (list) ) |
12 | |
13 | (defmethod list-base-classes ((type (eql :clsql))) |
14 | *clsql-base-classes*) |
15 | |
16 | (defmethod def-base-class-expander ((model meta-model-class) (base-type (eql :clsql)) (name t) (args t)) |
17 | `(def-view-class ,name () |
18 | ,(meta-model.metadata model))) |
19 | |
20 | (defmethod def-base-class-expander :after ((model meta-model-class) (base-type (eql :clsql)) (name t) (args t)) |
21 | (unless (member name *clsql-base-classes*) |
22 | (setf *clsql-base-classes* (cons name *clsql-base-classes*)))) |
23 | |
24 | (defparameter *sql-type-map* '((:INT4 integer) (:TEXT string) (:VARCHAR string) (:TIMESTAMP clsql-sys::wall-time) (:NUMERIC number)(:BYTEA string))) |
25 | |
26 | (defun gen-type (table column) |
27 | (cadr (assoc |
28 | (cadr (assoc |
29 | column |
30 | (list-attribute-types table) |
31 | :test #'equalp )) |
32 | *sql-type-map*))) |
33 | |
34 | (defun sql->sym (name &optional (package nil)) |
35 | (flet ((xform (x) |
36 | (string-upcase (substitute #\- #\_ x)))) |
37 | (if package |
38 | (intern (xform (string name)) package) |
39 | (intern (xform (string name)))))) |
40 | |
41 | (defun table->slots (table pkey) |
42 | (mapcar |
43 | #'(lambda (col) |
44 | `(,(sql->sym col) |
45 | :accessor ,(sql->sym col) |
46 | :initarg ,(sql->sym col "KEYWORD") |
47 | :type ,(gen-type table col) |
48 | :db-kind |
49 | ,(if (equalp col pkey) |
50 | `:key |
51 | `:base))) |
52 | (list-attributes table))) |
53 | |
54 | (defun view-class-definition-list () |
55 | (mapcar #'(lambda (x) `(def-meta-model-from-table ,x)) |
56 | (list-tables))) |
57 | |
58 | (defmacro def-meta-models () |
59 | (let ((defs (view-class-definition-list))) |
60 | `(progn ,@defs))) |
61 | |
62 | |
63 | |
64 | |
65 | |
66 | (defun get-pkeys () |
67 | (let ((keys '())) |
68 | (dolist (row (get-pkeys-query)) |
69 | (setf keys (acons (car row) (list (cadr row)) keys))) |
70 | keys)) |
71 | |
72 | (defun get-pkeys-query() |
73 | (query |
74 | "SELECT pg_class.relname, pg_attribute.attname, pg_catalog.quote_ident(conname) AS constraint_n |
75 | , pg_catalog.pg_get_indexdef(d.objid) AS constraint_definition |
76 | , CASE |
77 | WHEN contype = 'p' THEN |
78 | 'PRIMARY KEY' |
79 | ELSE |
80 | 'UNIQUE' |
81 | END as constraint_type |
82 | FROM |
83 | pg_class, pg_attribute, |
84 | pg_catalog.pg_constraint AS c |
85 | JOIN pg_catalog.pg_depend AS d ON (d.refobjid = c.oid) |
86 | WHERE contype IN ('p', 'u') |
87 | AND deptype = 'i' |
88 | and conrelid = pg_class.oid |
89 | and pg_attribute.attnum = ANY (c.conkey) |
90 | and pg_attribute.attrelid = pg_class.oid")) |
91 | |
92 | ;;here is how this works |
93 | ;;from the postgres system tables we get |
94 | ;;list of all the has-a relationships. |
95 | ;;the inverse of a has-a is an implicit has-many |
96 | ;;and any relation having more than one foreign key |
97 | ;;is a join table hosting a many-to-many relationship |
98 | |
99 | (defun get-fkey-explosions () |
100 | (let ((key-table (get-fkey-explosions-query)) |
101 | (keys '())) |
102 | (dolist (row key-table) |
103 | (setf row (mapcar #'(lambda (x) |
104 | (sql->sym x)) |
105 | row)) |
106 | ;;this one does the has-a |
107 | (setf keys (acons (car row) (gen-has-a row) |
108 | keys)) |
109 | ;;the inverse of the previous represents a has-many. |
110 | (setf keys |
111 | (acons (fourth row) (gen-has-many row) |
112 | keys)) |
113 | |
114 | ;;many-to-many |
115 | (dolist (mrow |
116 | (remove-if #'(lambda (r) (or (not (equal (car row) (car r))) |
117 | (equal (last row) (last r)))) |
118 | (mapcar #'(lambda (x) |
119 | (mapcar #'sql->sym x)) |
120 | key-table))) |
121 | (setf keys (acons (fourth row) |
122 | (gen-many-to-many mrow (third row) (second row)) |
123 | keys)))) |
124 | keys )) |
125 | |
126 | |
127 | (defun get-fkey-explosions-query () |
128 | ;;these query's are a mess, i don't even know how they work :) |
129 | (query " |
130 | SELECT pg_class.relname, |
131 | pg_attribute.attname, |
132 | fa.attname , |
133 | f.relname |
134 | FROM pg_class, |
135 | pg_constraint, |
136 | pg_attribute, |
137 | pg_class as f , |
138 | pg_attribute as fa |
139 | WHERE pg_class.relname in (select tablename from pg_tables where schemaname = 'public') |
140 | AND pg_class.oid = pg_constraint.conrelid |
141 | AND pg_attribute.attnum = ANY (pg_constraint.conkey) |
142 | AND pg_attribute.attrelid = pg_class.oid |
143 | AND f.oid = confrelid |
144 | AND fa.attrelid = f.oid |
145 | AND fa.attnum = ANY (pg_constraint.confkey)")) |
146 | |
147 | |
148 | ;; i chose keyword args here so as to make the code more understandable. |
149 | ;; it didn't really work. |
150 | (defun gen-join-slot (&key name home-key foreign-key join-class (set nil)) |
151 | `(,(intern name) |
152 | :accessor ,(intern name) |
153 | :db-kind :join |
154 | :db-info (:join-class ,join-class |
155 | :home-key ,home-key |
156 | :foreign-key ,foreign-key |
157 | :set ,set))) |
158 | |
159 | (defun gen-has-a (row) |
160 | (gen-join-slot |
161 | :name |
162 | (format nil "~A->~A" (string (car row))(string (second row))) |
163 | :home-key (second row) |
164 | :foreign-key (third row) |
165 | :join-class (fourth row))) |
166 | |
167 | (defun gen-has-many (row) |
168 | (gen-join-slot |
169 | :name |
170 | (format nil "~A->~A" (string (car row))(string (second row))) |
171 | :home-key (third row) |
172 | :foreign-key (second row) |
173 | :join-class (car row) |
174 | :set t)) |
175 | |
176 | (defun gen-many-to-many (row home-key foreign-key) |
177 | (let ((name (sql->sym (string-upcase (format nil "~A->~A" (string (car row)) (string (second row))))))) |
178 | (setf row (mapcar #'sql->sym row)) |
179 | `(,name |
180 | :accessor ,name |
181 | :db-kind :join |
182 | :db-info (:join-class ,(car row) |
183 | :home-key ,home-key |
184 | :foreign-key ,foreign-key |
185 | :target-slot ,name |
186 | :set t)))) |
187 | |
188 | (defmethod update-records-from-instance :before ((view clsql::standard-db-object) &key database) |
189 | (declare (ignorable database)) |
190 | (labels ((sym->sql (sym) (string-downcase (substitute #\_ #\- (string sym)))) |
191 | (get-def (slot) (caar (query |
192 | (format nil "SELECT DISTINCT adsrc from pg_attrdef join pg_attribute on attnum = adnum where adrelid = (select oid from pg_class where relname = '~A') and attname = '~A'" (sym->sql (class-name (class-of view))) (sym->sql slot))))) |
193 | (get-default-value (slot) (caar (query (format nil "SELECT ~A" (get-def slot)))))) |
194 | |
195 | (dolist (slot (list-slots view)) |
196 | (when (and (primary-key-p view slot) |
197 | (or (not (slot-boundp view slot)) |
198 | (equal (slot-value view slot) nil))) |
199 | (setf (slot-value view slot) (get-default-value slot)))))) |
200 | |
201 | ;;;; |
202 | |
203 | (defmacro def-view-class/meta (name supers slots &rest args) |
204 | `(progn |
205 | (let* ((m (def-meta-model model-name ,supers ,slots ,args)) |
206 | (i (make-instance m))) |
207 | (prog1 (eval (def-base-class-expander i :clsql ',name ',args)) |
208 | (defmethod meta-model.metadata ((self ,name)) |
209 | (meta-model.metadata i)))))) |
210 | |
211 | |
212 | (defmacro def-view-class/table (table &optional name) |
213 | "takes the name of a table as a string and |
214 | creates a clsql view-class" |
215 | (let* ((pkey (cadr (assoc table (get-pkeys) :test #'equalp))) |
216 | (table-slots (table->slots table pkey)) |
217 | (join-slots |
218 | (let ((slots nil)) |
219 | (dolist (exp (get-fkey-explosions)) |
220 | (when (equalp (car exp) (sql->sym table)) |
221 | (setf slots (cons (cdr exp) slots)))) |
222 | slots))) |
223 | `(def-view-class/meta ,(if name name (sql->sym table)) |
224 | () |
225 | ,(append table-slots join-slots)))) |
226 | |
227 | |
228 | |