forked from volatiletech/sqlboiler
/
table.go
66 lines (53 loc) · 1.64 KB
/
table.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package drivers
import "fmt"
// Table metadata from the database schema.
type Table struct {
Name string `json:"name"`
// For dbs with real schemas, like Postgres.
// Example value: "schema_name"."table_name"
SchemaName string `json:"schema_name"`
Columns []Column `json:"columns"`
PKey *PrimaryKey `json:"p_key"`
FKeys []ForeignKey `json:"f_keys"`
IsJoinTable bool `json:"is_join_table"`
ToOneRelationships []ToOneRelationship `json:"to_one_relationships"`
ToManyRelationships []ToManyRelationship `json:"to_many_relationships"`
}
// GetTable by name. Panics if not found (for use in templates mostly).
func GetTable(tables []Table, name string) (tbl Table) {
for _, t := range tables {
if t.Name == name {
return t
}
}
panic(fmt.Sprintf("could not find table name: %s", name))
}
// GetColumn by name. Panics if not found (for use in templates mostly).
func (t Table) GetColumn(name string) (col Column) {
for _, c := range t.Columns {
if c.Name == name {
return c
}
}
panic(fmt.Sprintf("could not find column name: %s", name))
}
// CanLastInsertID checks the following:
// 1. Is there only one primary key?
// 2. Does the primary key column have a default value?
// 3. Is the primary key column type one of uintX/intX?
// If the above is all true, this table can use LastInsertId
func (t Table) CanLastInsertID() bool {
if t.PKey == nil || len(t.PKey.Columns) != 1 {
return false
}
col := t.GetColumn(t.PKey.Columns[0])
if len(col.Default) == 0 {
return false
}
switch col.Type {
case "int", "int8", "int16", "int32", "int64", "uint", "uint8", "uint16", "uint32", "uint64":
default:
return false
}
return true
}