Vendor packages update

vendor/github.com/uptrace/bun/schema/tables.go

@@ -0,0 +1,114 @@
+package schema
+
+import (
+	"fmt"
+	"reflect"
+	"sync"
+
+	"github.com/puzpuzpuz/xsync/v3"
+)
+
+type Tables struct {
+	dialect Dialect
+
+	mu     sync.Mutex
+	tables *xsync.MapOf[reflect.Type, *Table]
+
+	inProgress map[reflect.Type]*Table
+}
+
+func NewTables(dialect Dialect) *Tables {
+	return &Tables{
+		dialect:    dialect,
+		tables:     xsync.NewMapOf[reflect.Type, *Table](),
+		inProgress: make(map[reflect.Type]*Table),
+	}
+}
+
+func (t *Tables) Register(models ...any) {
+	for _, model := range models {
+		_ = t.Get(reflect.TypeOf(model).Elem())
+	}
+}
+
+func (t *Tables) Get(typ reflect.Type) *Table {
+	typ = indirectType(typ)
+	if typ.Kind() != reflect.Struct {
+		panic(fmt.Errorf("got %s, wanted %s", typ.Kind(), reflect.Struct))
+	}
+
+	if v, ok := t.tables.Load(typ); ok {
+		return v
+	}
+
+	t.mu.Lock()
+	defer t.mu.Unlock()
+
+	if v, ok := t.tables.Load(typ); ok {
+		return v
+	}
+
+	table := t.InProgress(typ)
+	table.initRelations()
+
+	t.dialect.OnTable(table)
+	for _, field := range table.FieldMap {
+		if field.UserSQLType == "" {
+			field.UserSQLType = field.DiscoveredSQLType
+		}
+		if field.CreateTableSQLType == "" {
+			field.CreateTableSQLType = field.UserSQLType
+		}
+	}
+
+	t.tables.Store(typ, table)
+	return table
+}
+
+func (t *Tables) InProgress(typ reflect.Type) *Table {
+	if table, ok := t.inProgress[typ]; ok {
+		return table
+	}
+
+	table := new(Table)
+	t.inProgress[typ] = table
+	table.init(t.dialect, typ)
+
+	return table
+}
+
+// ByModel gets the table by its Go name.
+func (t *Tables) ByModel(name string) *Table {
+	var found *Table
+	t.tables.Range(func(typ reflect.Type, table *Table) bool {
+		if table.TypeName == name {
+			found = table
+			return false
+		}
+		return true
+	})
+	return found
+}
+
+// ByName gets the table by its SQL name.
+func (t *Tables) ByName(name string) *Table {
+	var found *Table
+	t.tables.Range(func(typ reflect.Type, table *Table) bool {
+		if table.Name == name {
+			found = table
+			return false
+		}
+		return true
+	})
+	return found
+}
+
+// All returns all registered tables.
+func (t *Tables) All() []*Table {
+	var found []*Table
+	t.tables.Range(func(typ reflect.Type, table *Table) bool {
+		found = append(found, table)
+		return true
+	})
+	return found
+}