e4832fdbcf
* redoing connection handling * a little more cleanup * empty implementation of rotation * updating rotate signature * signature update * updating interfaces again :( * changing back to interface * adding templated url support and rotation for postgres * adding correct username * return updates * updating statements to be a list * adding error sanitizing middleware * fixing log sanitizier * adding postgres rotate test * removing conf from rotate * adding rotate command * adding mysql rotate * finishing up the endpoint in the db backend for rotate * no more structs, just store raw config * fixing tests * adding db instance lock * adding support for statement list in cassandra * wip redoing interface to support BC * adding falllback for Initialize implementation * adding backwards compat for statements * fix tests * fix more tests * fixing up tests, switching to new fields in statements * fixing more tests * adding mssql and mysql * wrapping all the things in middleware, implementing templating for mongodb * wrapping all db servers with error santizer * fixing test * store the name with the db instance * adding rotate to cassandra * adding compatibility translation to both server and plugin * reordering a few things * store the name with the db instance * reordering * adding a few more tests * switch secret values from slice to map * addressing some feedback * reinstate execute plugin after resetting connection * set database connection to closed * switching secret values func to map[string]interface for potential future uses * addressing feedback
53 lines
1.7 KiB
Go
53 lines
1.7 KiB
Go
package dbutil
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/hashicorp/vault/builtin/logical/database/dbplugin"
|
|
)
|
|
|
|
var (
|
|
ErrEmptyCreationStatement = errors.New("empty creation statements")
|
|
)
|
|
|
|
// Query templates a query for us.
|
|
func QueryHelper(tpl string, data map[string]string) string {
|
|
for k, v := range data {
|
|
tpl = strings.Replace(tpl, fmt.Sprintf("{{%s}}", k), v, -1)
|
|
}
|
|
|
|
return tpl
|
|
}
|
|
|
|
// StatementCompatibilityHelper will populate the statements fields to support
|
|
// compatibility
|
|
func StatementCompatibilityHelper(statements dbplugin.Statements) dbplugin.Statements {
|
|
switch {
|
|
case len(statements.Creation) > 0 && len(statements.CreationStatements) == 0:
|
|
statements.CreationStatements = strings.Join(statements.Creation, ";")
|
|
case len(statements.CreationStatements) > 0:
|
|
statements.Creation = []string{statements.CreationStatements}
|
|
}
|
|
switch {
|
|
case len(statements.Revocation) > 0 && len(statements.RevocationStatements) == 0:
|
|
statements.RevocationStatements = strings.Join(statements.Revocation, ";")
|
|
case len(statements.RevocationStatements) > 0:
|
|
statements.Revocation = []string{statements.RevocationStatements}
|
|
}
|
|
switch {
|
|
case len(statements.Renewal) > 0 && len(statements.RenewStatements) == 0:
|
|
statements.RenewStatements = strings.Join(statements.Renewal, ";")
|
|
case len(statements.RenewStatements) > 0:
|
|
statements.Renewal = []string{statements.RenewStatements}
|
|
}
|
|
switch {
|
|
case len(statements.Rollback) > 0 && len(statements.RollbackStatements) == 0:
|
|
statements.RollbackStatements = strings.Join(statements.Rollback, ";")
|
|
case len(statements.RollbackStatements) > 0:
|
|
statements.Rollback = []string{statements.RollbackStatements}
|
|
}
|
|
return statements
|
|
}
|