mod/database.cxx


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

// file      : mod/database.cxx -*- C++ -*-
// copyright : Copyright (c) 2014-2016 Code Synthesis Ltd
// license   : MIT; see accompanying LICENSE file

#include <mod/database>

#include <map>

#include <odb/pgsql/database.hxx>
#include <odb/pgsql/connection-factory.hxx>

namespace brep
{
  namespace options
  {
    bool
    operator< (const db& x, const db& y)
    {
      int r;
      if ((r = x.db_user ().compare (y.db_user ())) != 0 ||
          (r = x.db_password ().compare (y.db_password ())) != 0 ||
          (r = x.db_name ().compare (y.db_name ())) != 0 ||
          (r = x.db_host ().compare (y.db_host ())))
        return r < 0;

      return x.db_port () < y.db_port ();
    }
  }

  using namespace odb;

  shared_ptr<database>
  shared_database (const options::db& o)
  {
    static std::map<options::db, weak_ptr<database>> databases;

    auto i (databases.find (o));
    if (i != databases.end ())
    {
      if (shared_ptr<database> d = i->second.lock ())
        return d;
    }

    unique_ptr<pgsql::connection_factory>
      f (new pgsql::connection_pool_factory (o.db_max_connections ()));

    shared_ptr<database> d (
      make_shared<pgsql::database> (
        o.db_user (),
        o.db_password (),
        o.db_name (),
        o.db_host (),
        o.db_port (),
        "options='-c default_transaction_isolation=serializable'",
        move (f)));

    databases[o] = d;
    return d;
  }
}