42b30802cd6fc8e9aa535a9b70edf3d472c6ad0e
[umlisp.git] / sql.lisp
1 ;;;; -*- Mode: Lisp; Syntax: ANSI-Common-Lisp; Base: 10; Package: umlisp -*-
2 ;;;; *************************************************************************
3 ;;;; FILE IDENTIFICATION
4 ;;;;
5 ;;;; Name:     sql.lisp
6 ;;;; Purpose:  Low-level SQL routines data for UMLisp
7 ;;;; Author:   Kevin M. Rosenberg
8 ;;;; Created:  Apr 2000
9 ;;;;
10 ;;;; This file, part of UMLisp, is
11 ;;;;    Copyright (c) 2000-2010 by Kevin M. Rosenberg, M.D.
12 ;;;;
13 ;;;; UMLisp users are granted the rights to distribute and use this software
14 ;;;; as governed by the terms of the GNU General Public License.
15 ;;;; *************************************************************************
16
17 (in-package #:umlisp)
18
19 (defvar +umls-sql-map+
20     '((:2004aa . "KUMLS2004AA")
21       (:2006ac . "KUMLS2006AC")
22       (:2006ad . "MTS2006AD")
23       (:2009ab . "MTS2009AB")))
24 (defvar +default-umls-db+ "MTS2009AB")
25
26 (defun lookup-db-name (db)
27   (cdr (assoc (ensure-keyword db) +umls-sql-map+)))
28
29 (defvar *umls-sql-db* +default-umls-db+)
30 (defun umls-sql-db ()
31   *umls-sql-db*)
32
33 (defun set-umls-sql-db (db)
34   (etypecase db
35     (string
36      (setq *umls-sql-db* db))
37     (keyword
38      (setq *umls-sql-db* (lookup-db-name db)))))
39
40 (defvar *umls-sql-user* "secret")
41 (defun umls-sql-user ()
42   *umls-sql-user*)
43 (defun set-umls-sql-user (u)
44   (sql-disconnect-pooled)
45   (setq *umls-sql-user* u))
46
47 (defvar *umls-sql-passwd* "secret")
48 (defun umls-sql-passwd ()
49   *umls-sql-passwd*)
50 (defun set-umls-sql-passwd (p)
51   (sql-disconnect-pooled)
52   (setq *umls-sql-passwd* p))
53
54 (defvar *umls-sql-host* "localhost")
55 (defun umls-sql-host ()
56   *umls-sql-host*)
57 (defun set-umls-sql-host (h)
58   (sql-disconnect-pooled)
59   (setq *umls-sql-host* h))
60
61 (defvar *umls-sql-type* :mysql)
62 (defun umls-sql-type ()
63   *umls-sql-type*)
64 (defun set-umls-sql-type (h)
65   (sql-disconnect-pooled)
66   (setq *umls-sql-type* h))
67
68 (defun umls-connection-spec ()
69   (if (eql *umls-sql-type* :mysql)
70       (list *umls-sql-host* *umls-sql-db*
71             *umls-sql-user* *umls-sql-passwd*
72             nil '((:local-infile . 1)))
73     (list *umls-sql-host* *umls-sql-db*
74           *umls-sql-user* *umls-sql-passwd*)))
75
76 (defun sql-connect ()
77   "Connect to UMLS database, automatically used pooled connections"
78   (clsql:connect (umls-connection-spec)
79                  :database-type *umls-sql-type*
80                  :encoding :utf-8
81                  :pool t))
82
83 (defun sql-disconnect (conn)
84   "Disconnect from UMLS database, but put connection back into pool"
85   (clsql:disconnect :database conn))
86
87 (defun sql-disconnect-pooled ()
88   (clsql:disconnect-pooled))
89
90 (defmacro with-sql-connection ((conn) &body body)
91   `(let ((,conn (sql-connect)))
92      (unwind-protect
93          (progn ,@body)
94        (when ,conn (clsql:disconnect :database ,conn)))))
95
96 (defun sql-query (cmd conn &key (result-types :auto))
97   (clsql:query cmd :database conn :result-types result-types :field-names nil))
98
99 (defun sql-execute (cmd conn)
100   (clsql:execute-command cmd :database conn))
101
102 (defun sql (stmt conn)
103   (if (string-equal "SELECT" (subseq stmt 0 6))
104       (sql-query stmt conn)
105     (sql-execute stmt conn)))
106
107 (defun umls-sql (stmt)
108   (check-type stmt string)
109   (with-sql-connection (conn)
110     (sql stmt conn)))
111
112 ;;; Pool of open connections
113
114 (defmacro with-mutex-sql ((conn) &body body)
115   `(let ((,conn (sql-connect)))
116      (unwind-protect
117          (progn ,@body)
118        (when ,conn (sql-disconnect ,conn)))))
119
120 (defun mutex-sql-execute (cmd)
121   (with-mutex-sql (conn)
122     (sql-execute cmd conn)))
123
124 (defun mutex-sql-query (cmd &key (result-types :auto))
125   (with-mutex-sql (conn)
126     (sql-query cmd conn :result-types result-types)))