diff --git a/app/app.go b/app/app.go index 0323f9c..01d84f6 100644 --- a/app/app.go +++ b/app/app.go @@ -19,12 +19,14 @@ type App struct { func (a *App) ApplyRule(c config.SchemaConfigData, globalColumns map[string]string, generators map[string][]string) error { var query string + if c.Table == "" { + return errors.New("Table must be defined") + } + if c.Query != "" { query = c.Query - } else if c.Table != "" { - query = fmt.Sprintf("SELECT * FROM %s", c.Table) } else { - return errors.New("No query and no table defined") + query = fmt.Sprintf("SELECT * FROM %s", c.Table) } if len(c.PrimaryKey) == 0 {