sqlboiler/bdb/relationships.go

100 lines
2.8 KiB
Go
Raw Normal View History

package bdb
2016-06-23 07:03:05 +02:00
// ToManyRelationship describes a relationship between two tables where the
// local table has no id, and the foreign table has an id that matches a column
// in the local table.
type ToManyRelationship struct {
Table string
Column string
Nullable bool
2016-07-15 21:09:32 +02:00
Unique bool
ForeignTable string
ForeignColumn string
ForeignColumnNullable bool
2016-07-15 21:09:32 +02:00
ForeignColumnUnique bool
ToJoinTable bool
JoinTable string
JoinLocalColumn string
JoinLocalColumnNullable bool
JoinLocalColumnUnique bool
JoinForeignColumn string
JoinForeignColumnNullable bool
2016-07-15 21:09:32 +02:00
JoinForeignColumnUnique bool
2016-06-23 07:03:05 +02:00
}
// ToManyRelationships relationship lookups
// Input should be the sql name of a table like: videos
func ToManyRelationships(table string, tables []Table) []ToManyRelationship {
localTable := GetTable(tables, table)
return toManyRelationships(localTable, tables)
}
func toManyRelationships(table Table, tables []Table) []ToManyRelationship {
2016-06-23 07:03:05 +02:00
var relationships []ToManyRelationship
for _, t := range tables {
for _, f := range t.FKeys {
if f.ForeignTable != table.Name {
2016-06-23 07:03:05 +02:00
continue
}
relationships = append(relationships, buildRelationship(table, f, t, tables))
2016-06-23 07:03:05 +02:00
}
}
return relationships
}
func buildRelationship(localTable Table, foreignKey ForeignKey, foreignTable Table, tables []Table) ToManyRelationship {
if !foreignTable.IsJoinTable {
col := localTable.GetColumn(foreignKey.ForeignColumn)
return ToManyRelationship{
2016-08-28 02:04:51 +02:00
Table: localTable.Name,
Column: foreignKey.ForeignColumn,
Nullable: col.Nullable,
2016-07-15 21:09:32 +02:00
Unique: col.Unique,
ForeignTable: foreignTable.Name,
ForeignColumn: foreignKey.Column,
ForeignColumnNullable: foreignKey.Nullable,
2016-07-15 21:09:32 +02:00
ForeignColumnUnique: foreignKey.Unique,
ToJoinTable: false,
}
}
col := foreignTable.GetColumn(foreignKey.Column)
relationship := ToManyRelationship{
2016-08-28 02:04:51 +02:00
Table: localTable.Name,
Column: foreignKey.ForeignColumn,
Nullable: col.Nullable,
2016-07-15 21:09:32 +02:00
Unique: col.Unique,
ToJoinTable: true,
JoinTable: foreignTable.Name,
}
for _, fk := range foreignTable.FKeys {
if fk.ForeignTable != localTable.Name {
relationship.JoinForeignColumn = fk.Column
relationship.JoinForeignColumnNullable = fk.Nullable
2016-07-15 21:09:32 +02:00
relationship.JoinForeignColumnUnique = fk.Unique
foreignTable := GetTable(tables, fk.ForeignTable)
foreignCol := foreignTable.GetColumn(fk.ForeignColumn)
relationship.ForeignTable = fk.ForeignTable
relationship.ForeignColumn = fk.ForeignColumn
relationship.ForeignColumnNullable = foreignCol.Nullable
2016-07-15 21:09:32 +02:00
relationship.ForeignColumnUnique = foreignCol.Unique
} else {
relationship.JoinLocalColumn = fk.Column
relationship.JoinLocalColumnNullable = fk.Nullable
2016-07-15 21:09:32 +02:00
relationship.JoinLocalColumnUnique = fk.Unique
}
}
return relationship
}