Ejemplo n.º 1
0
// GetColumns returns all columns from a table. It discards the column entity_type_id from some
// entity tables.
func GetColumns(dbrSess dbr.SessionRunner, table string) (Columns, error) {
	var cols = make(Columns, 0, 100)

	sel := dbrSess.SelectBySql("SHOW COLUMNS FROM " + dbr.Quoter.Table(table))
	selSql, selArg := sel.ToSql()
	rows, err := sel.Query(selSql, selArg...)

	if err != nil {
		return nil, log.Error("csdb.GetColumns.Query", "err", err, "query", selSql, "args", selArg)
	}
	defer rows.Close()

	col := Column{}
	for rows.Next() {
		err := rows.Scan(&col.Field, &col.Type, &col.Null, &col.Key, &col.Default, &col.Extra)
		if err != nil {
			return nil, log.Error("csdb.GetColumns.Rows.Scan", "err", err, "query", selSql, "args", selArg)
		}
		cols = append(cols, col)
	}
	err = rows.Err()
	if err != nil {
		return nil, log.Error("csdb.GetColumns.Rows.Err", "err", err, "query", selSql, "args", selArg)
	}
	return cols, nil
}
Ejemplo n.º 2
0
// Select generates a SELECT * FROM tableName statement
func (ts *Table) Select(dbrSess dbr.SessionRunner) (*dbr.SelectBuilder, error) {
	if ts == nil {
		return nil, ErrTableNotFound
	}
	return dbrSess.
		Select(ts.AllColumnAliasQuote("main_table")...).
		From(ts.Name, "main_table"), nil
}
Ejemplo n.º 3
0
// GetTables returns all tables from a database. AndWhere can be optionally applied.
// Only first index (0) will be added.
func GetTables(dbrSess dbr.SessionRunner, sql ...string) ([]string, error) {

	qry := "SHOW TABLES"
	if len(sql) > 0 && sql[0] != "" {
		if false == dbr.Stmt.IsSelect(sql[0]) {
			qry = qry + " LIKE '" + sql[0] + "'"
		} else {
			qry = sql[0]
		}
	}
	if PkgLog.IsDebug() { // this if reduces 9 allocs ...
		defer log.WhenDone(PkgLog).Debug("Stats", "Package", "codegen", "Step", "GetTables", "query", qry)
	}

	sb := dbrSess.SelectBySql(qry)
	query, args, err := sb.ToSql()
	if err != nil {
		return nil, errgo.Mask(err)
	}
	rows, err := sb.Query(query, args...)
	if err != nil {
		return nil, errgo.Mask(err)
	}
	defer rows.Close()

	var tableName string
	var tableNames = make([]string, 0, 200)
	for rows.Next() {
		if err := rows.Scan(&tableName); err != nil {
			return nil, errgo.Mask(err)
		}
		tableNames = append(tableNames, tableName)
	}

	if err = rows.Err(); err != nil {
		return nil, errgo.Mask(err)
	}
	return tableNames, nil
}
Ejemplo n.º 4
0
// GetAttributeSelectSql generates the select query to retrieve full attribute configuration
// Implements the scope on a SQL query basis so that attribute functions does not need to deal with it.
// Tests see the tools package
// @see magento2/app/code/Magento/Eav/Model/Resource/Attribute/Collection.php::_initSelect()
func GetAttributeSelectSql(dbrSess dbr.SessionRunner, aat EntityTypeAdditionalAttributeTabler, entityTypeID, websiteID int64) (*dbr.SelectBuilder, error) {

	ta, err := TableCollection.Structure(TableIndexAttribute)
	if err != nil {
		return nil, log.Error("eav.GetAttributeSelectSql.TableCollection.Structure", "err", err, "entityTypeID", entityTypeID, "websiteID", websiteID)
	}
	taa, err := aat.TableAdditionalAttribute()
	if err != nil {
		return nil, log.Error("eav.GetAttributeSelectSql.TableAdditionalAttribute", "err", err, "ta", ta, "entityTypeID", entityTypeID, "websiteID", websiteID)
	}
	tew, err := aat.TableEavWebsite()
	if err != nil {
		return nil, log.Error("eav.GetAttributeSelectSql.TableEavWebsite", "err", err, "ta", ta, "taa", taa, "entityTypeID", entityTypeID, "websiteID", websiteID)
	}
	// tew table can now contains columns names which can occur in table eav_attribute and
	// or [catalog|customer|entity]_eav_attribute
	var (
		ifnull           []string
		tewAddedCols     []string
		taColumnsQuoted  = utils.StringSlice(ta.AllColumnAliasQuote(csdb.MainTable))
		taaColumnsQuoted = utils.StringSlice(taa.ColumnAliasQuote(csdb.AdditionalTable))
	)

	if tew != nil {
		ifnull = make([]string, len(tew.Columns))
		for i, tewC := range tew.Columns.ColumnsNoPK().FieldNames() {
			t := ""
			switch {
			case ta.In(tewC):
				t = csdb.MainTable
				break
			case taa.In(tewC):
				t = csdb.AdditionalTable
				break
			default:
				return nil, log.Error("eav.GetAttributeSelectSql.Columns.FieldNames.default", "err",
					fmt.Errorf("Cannot find column name %s.%s neither in table %s nor in %s.", tew.Name, tewC, ta.Name, taa.Name),
					"ta", ta, "taa", taa,
					"entityTypeID", entityTypeID, "websiteID", websiteID)
			}
			ifnull[i] = dbr.IfNullAs(csdb.ScopeTable, tewC, t, tewC, tewC)
			tewAddedCols = append(tewAddedCols, tewC)
		}
		taColumnsQuoted.ReduceContains(tewAddedCols...)
		taaColumnsQuoted.ReduceContains(tewAddedCols...)
	}

	selectSql := dbrSess.
		Select(taColumnsQuoted...).
		From(ta.Name, csdb.MainTable).
		Join(
			dbr.JoinTable(taa.Name, csdb.AdditionalTable),
			taaColumnsQuoted,
			dbr.JoinOn(dbr.Quote+csdb.AdditionalTable+"`.`attribute_id` = `"+csdb.MainTable+"`.`attribute_id`"),
			dbr.JoinOn(dbr.Quote+csdb.MainTable+"`.`entity_type_id` = ?", entityTypeID),
		)

	if len(tewAddedCols) > 0 {
		selectSql.
			LeftJoin(
				dbr.JoinTable(tew.Name, csdb.ScopeTable),
				append(ifnull),
				dbr.JoinOn(dbr.Quote+csdb.ScopeTable+dbr.Quote+"."+dbr.Quote+"attribute_id"+dbr.Quote+" = "+dbr.Quote+csdb.MainTable+dbr.Quote+"."+dbr.Quote+"attribute_id"+dbr.Quote),
				dbr.JoinOn(dbr.Quote+csdb.ScopeTable+dbr.Quote+"."+dbr.Quote+"website_id"+dbr.Quote+" = ?", websiteID),
			)
	}
	return selectSql, nil
}