This repository has been archived on 2022-08-17. You can view files and clone it, but cannot push or open issues or pull requests.
dex/db/connector_config.go

122 lines
2.4 KiB
Go
Raw Normal View History

2015-08-18 05:57:27 +05:30
package db
import (
"encoding/json"
"errors"
"fmt"
"github.com/coopernurse/gorp"
"github.com/lib/pq"
"github.com/coreos/dex/connector"
)
const (
connectorConfigTableName = "connector_config"
)
func init() {
register(table{
name: connectorConfigTableName,
model: connectorConfigModel{},
autoinc: false,
pkey: []string{"id"},
})
}
func newConnectorConfigModel(cfg connector.ConnectorConfig) (*connectorConfigModel, error) {
b, err := json.Marshal(cfg)
if err != nil {
return nil, err
}
m := &connectorConfigModel{
ID: cfg.ConnectorID(),
Type: cfg.ConnectorType(),
Config: string(b),
}
return m, nil
}
type connectorConfigModel struct {
ID string `db:"id"`
Type string `db:"type"`
Config string `db:"config"`
}
func (m *connectorConfigModel) ConnectorConfig() (connector.ConnectorConfig, error) {
cfg, err := connector.NewConnectorConfigFromType(m.Type)
if err != nil {
return nil, err
}
if err = json.Unmarshal([]byte(m.Config), cfg); err != nil {
return nil, err
}
return cfg, nil
}
func NewConnectorConfigRepo(dbm *gorp.DbMap) *ConnectorConfigRepo {
return &ConnectorConfigRepo{dbMap: dbm}
}
type ConnectorConfigRepo struct {
dbMap *gorp.DbMap
}
func (r *ConnectorConfigRepo) All() ([]connector.ConnectorConfig, error) {
qt := pq.QuoteIdentifier(connectorConfigTableName)
q := fmt.Sprintf("SELECT * FROM %s", qt)
objs, err := r.dbMap.Select(&connectorConfigModel{}, q)
if err != nil {
return nil, err
}
cfgs := make([]connector.ConnectorConfig, len(objs))
for i, obj := range objs {
m, ok := obj.(*connectorConfigModel)
if !ok {
return nil, errors.New("unable to cast connector to connectorConfigModel")
}
cfg, err := m.ConnectorConfig()
if err != nil {
return nil, err
}
cfgs[i] = cfg
}
return cfgs, nil
}
func (r *ConnectorConfigRepo) Set(cfgs []connector.ConnectorConfig) error {
insert := make([]interface{}, len(cfgs))
for i, cfg := range cfgs {
m, err := newConnectorConfigModel(cfg)
if err != nil {
return err
}
insert[i] = m
}
tx, err := r.dbMap.Begin()
if err != nil {
return err
}
qt := pq.QuoteIdentifier(connectorConfigTableName)
q := fmt.Sprintf("DELETE FROM %s", qt)
if _, err = r.dbMap.Exec(q); err != nil {
return err
}
if err = r.dbMap.Insert(insert...); err != nil {
return fmt.Errorf("DB insert failed %#v: %v", insert, err)
}
return tx.Commit()
}