Add comments so lint will shut up

This commit is contained in:
Roland Shoemaker 2015-08-06 16:23:11 -07:00
parent ab2a7fce28
commit 1bd02fbeb3
1 changed files with 8 additions and 14 deletions

View File

@ -24,11 +24,15 @@ type domainRule struct {
Type string `db:"type"`
}
// PolicyAuthorityDatabaseImpl enforces policy decisions based on various rule
// lists
type PolicyAuthorityDatabaseImpl struct {
log *blog.AuditLogger
dbMap *gorp.DbMap
}
// NewPolicyAuthorityDatabaseImpl constructs a Policy Authority Database (and
// creates tables if they are non-existent)
func NewPolicyAuthorityDatabaseImpl(driver, name string) (padb core.PolicyAuthorityDatabase, err error) {
logger := blog.GetAuditLogger()
dbMap, err := sa.NewDbMap(driver, name)
@ -51,6 +55,7 @@ func NewPolicyAuthorityDatabaseImpl(driver, name string) (padb core.PolicyAuthor
return padb, nil
}
// AddRule will add a whitelist or blacklist rule to the database
func (padb *PolicyAuthorityDatabaseImpl) AddRule(rule string, string string) error {
tx, err := padb.dbMap.Begin()
if err != nil {
@ -78,16 +83,11 @@ func (padb *PolicyAuthorityDatabaseImpl) AddRule(rule string, string string) err
return err
}
// CheckRules will query the database for white/blacklist rules that match host,
// if both whitelist and blacklist rules are found the whitelist will always win
func (padb *PolicyAuthorityDatabaseImpl) CheckRules(host string) error {
// Wrap in transaction so the whitelist doesn't change under us
tx, err := padb.dbMap.Begin()
if err != nil {
tx.Rollback()
return err
}
var rules []domainRule
_, err = tx.Select(
_, err := padb.dbMap.Select(
&rules,
`SELECT type,rule FROM ruleList WHERE :host LIKE rule`,
map[string]interface{}{"host": host},
@ -96,12 +96,6 @@ func (padb *PolicyAuthorityDatabaseImpl) CheckRules(host string) error {
return err
}
err = tx.Commit()
if err != nil {
tx.Rollback()
return err
}
var wRules []string
var bRules []string
for _, rule := range rules {