Function newKeyValueFromJSONConfig

pkg/sorted/postgres/postgreskv.go:39–116

(cfg jsonconfig.Obj)

}

func newKeyValueFromJSONConfig(cfg jsonconfig.Obj) (sorted.KeyValue, error) {
	var (
		user     = cfg.RequiredString("user")
		database = cfg.RequiredString("database")
		host     = cfg.OptionalString("host", "localhost")
		password = cfg.OptionalString("password", "")
		sslmode  = cfg.OptionalString("sslmode", "require")
	)
	if err := cfg.Validate(); err != nil {
		return nil, err
	}

	// connect without a database, it may not exist yet
	conninfo := fmt.Sprintf("user=%s host=%s sslmode=%s", user, host, sslmode)
	if password != "" {
		conninfo += fmt.Sprintf(" password=%s", password)
	}
	db, err := sql.Open("postgres", conninfo)
	if err != nil {
		return nil, err
	}
	err = createDB(db, database)
	db.Close() // ignoring error, if createDB failed db.Close() will likely also fail
	if err != nil {
		return nil, err
	}

	// reconnect after database is created
	conninfo += fmt.Sprintf(" dbname=%s", database)
	db, err = sql.Open("postgres", conninfo)
	if err != nil {
		return nil, err
	}

	for _, tableSQL := range SQLCreateTables() {
		if _, err := db.Exec(tableSQL); err != nil {
			return nil, fmt.Errorf("error creating table with %q: %v", tableSQL, err)
		}
	}
	for _, statement := range SQLDefineReplace() {
		if _, err := db.Exec(statement); err != nil {
			return nil, fmt.Errorf("error setting up replace statement with %q: %v", statement, err)
		}
	}
	r, err := db.Query(fmt.Sprintf(`SELECT replaceintometa('version', '%d')`, SchemaVersion()))
	if err != nil {
		return nil, fmt.Errorf("error setting schema version: %v", err)
	}
	r.Close()

	kv := &keyValue{
		db: db,
		KeyValue: &sqlkv.KeyValue{
			DB:              db,
			SetFunc:         altSet,
			BatchSetFunc:    altBatchSet,
			PlaceHolderFunc: replacePlaceHolders,
		},