aboutsummaryrefslogtreecommitdiffstats
path: root/store.go
blob: b744d887b8174e173e9e5aae787ffb3e7a3830bc (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
package main

import (
	"database/sql"
	"log"

	_ "github.com/lib/pq"
)

type Session struct {
	Id     string
	UserId string
}

type SessionStore interface {
	Get(id string) (*Session, bool)
}

type PgSessionStore struct {
	*sql.DB
	cache map[string]*Session
}

func NewPgStore() *PgSessionStore {
	db, err := sql.Open("postgres", "postgres://auth_master:pass@localhost/authdb")
	if err != nil {
		log.Panic(err)
	}
	return &PgSessionStore{db, make(map[string]*Session)}
}

func (store *PgSessionStore) Get(id string) (*Session, bool) {
	s, ok := store.cache[id]
	return s, ok
}