aboutsummaryrefslogtreecommitdiff
path: root/gn/db/mysql.scm
blob: f6513afc6a8ea95a089ada1d3d33c42dbe27a82a (plain)
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
#!
  Module for handling SQL DB primitives. Note that GN queries should go into gn/data
!#

(define-module (gn db mysql)
  #:use-module (json)
  #:use-module (ice-9 match)
  #:use-module (ice-9 format)
  #:use-module (ice-9 iconv)
  #:use-module (ice-9 receive)
  #:use-module (ice-9 string-fun)
  #:use-module (rnrs base)
  #:use-module (dbi dbi)

  #:export (
            db-check
            db-display-rows
            db-get-rows
            db-get-rows-apply
            ))

(define (db-check db)
  "Use DBI-style handle to report an error. On error the program will stop."
  (match (dbi-get_status db)
    ((stat . msg) (if (= stat 0)
                     #t
                     (begin
                       (display msg)
                       (newline)
                       (assert stat))))))

(define (db-display-rows db)
  (let [(row  (dbi-get_row db))]
    (if row
        (begin
          (display row)
          (get-rows db)
          )
        #t)))

(define (db-get-rows db list)
  "After running dbi-query we can fetch all rows and return them as a list of records, which are assoc list:

    (dbi-query db \"SELECT StrainId,Strain.Name FROM Strain, StrainXRef WHERE StrainXRef.StrainId = Strain.Id AND StrainXRef.InbredSetId = 1 ORDER BY StrainId;\")
    (db-check db)
    (display (db-get-rows db '()))

    (((StrainId . 4) (Name . BXD1)) ((StrainId . 5) (Name . BXD2))...
"
  (let [(row (dbi-get_row db))]
    (if row
        (db-get-rows db (append list `(,row)))
        list)))

(define (db-get-rows-apply db func list)
  "Similar to db-get-rows with a function that gets applied on every record, e.g.

    (define ids (db-get-rows-apply db (lambda (r) `(,(assoc-ref r \"StrainId\") . ,(assoc-ref r \"Name\"))) '()))

"
  (let [(row (dbi-get_row db))]
    (if row
        (db-get-rows-apply db func (append list `(,(func row))))
        list)))