dosco
diff --git a/‎.gitattributes‎
Lines changed: 1 addition & 0 deletions b/‎.gitattributes‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎Makefile‎
Lines changed: 6 additions & 0 deletions b/‎Makefile‎
Lines changed: 6 additions & 0 deletions
diff --git a/‎core/api.go‎
Lines changed: 105 additions & 1 deletion b/‎core/api.go‎
Lines changed: 105 additions & 1 deletion
@@ -0,0 +1 @@
+tests-large/02_adventureworks_data.sql filter=lfs diff=lfs merge=lfs -text
@@ -57,6 +57,12 @@ test-mongodb:
 	@echo "Running MongoDB tests..."
 	@cd tests; go test -v -timeout 30m -race -db=mongodb .
 
+test-adventureworks:
+	@echo "Running AdventureWorks tests..."
+	@cd tests; go test -v -timeout 60m -race -db=adventureworks -run TestAdventureWorks .
+
+test-large: test-adventureworks
+
 BIN_DIR := $(GOPATH)/bin
 WEB_BUILD_DIR := ./serv/web/build/manifest.json
 
 
@@ -120,10 +120,99 @@ type GraphJin struct {
 	done     chan bool
 	stopOnce sync.Once
 	reloadMu sync.Mutex // serializes reload operations
+
+	// Schema change callbacks
+	schemaCallbacks []func(dbName string, hash string)
+	callbackMu      sync.RWMutex
+
+	// Discovery document cache
+	discovery sync.Map // map[string]*DiscoveryDocument
 }
 
 type Option func(*graphjinEngine) error
 
+// OnSchemaChange registers a callback that fires when the database schema changes.
+// The callback receives the database name and a hex-encoded hash of the schema.
+// Callbacks also fire once at startup after initial schema discovery.
+func (g *GraphJin) OnSchemaChange(fn func(dbName string, hash string)) {
+	g.callbackMu.Lock()
+	defer g.callbackMu.Unlock()
+	g.schemaCallbacks = append(g.schemaCallbacks, fn)
+}
+
+// fireSchemaCallbacks invokes all registered schema change callbacks.
+// Runs each callback in a goroutine to avoid blocking the caller (which may hold reloadMu).
+func (g *GraphJin) fireSchemaCallbacks(dbName string, hash string) {
+	g.callbackMu.RLock()
+	callbacks := make([]func(string, string), len(g.schemaCallbacks))
+	copy(callbacks, g.schemaCallbacks)
+	g.callbackMu.RUnlock()
+
+	for _, fn := range callbacks {
+		fn := fn
+		go fn(dbName, hash)
+	}
+}
+
+// DefaultDatabase returns the name of the default (primary) database.
+func (g *GraphJin) DefaultDatabase() string {
+	gj, err := g.getEngine()
+	if err != nil {
+		return ""
+	}
+	return gj.defaultDB
+}
+
+// DatabaseNames returns the names of all configured databases.
+func (g *GraphJin) DatabaseNames() []string {
+	gj, err := g.getEngine()
+	if err != nil {
+		return nil
+	}
+	return gj.sortedDatabaseNames()
+}
+
+// fireAllSchemaCallbacks fires schema change callbacks for all databases with initialized schemas.
+func (g *GraphJin) fireAllSchemaCallbacks() {
+	gj, err := g.getEngine()
+	if err != nil {
+		return
+	}
+	for name, ctx := range gj.databases {
+		if ctx.dbinfo != nil {
+			g.fireSchemaCallbacks(name, fmt.Sprintf("%x", ctx.dbinfo.Hash()))
+		}
+	}
+}
+
+// generateAllDiscovery generates discovery documents for all databases with initialized schemas.
+// Called at startup and after schema changes.
+func (g *GraphJin) generateAllDiscovery() {
+	gj, err := g.getEngine()
+	if err != nil {
+		return
+	}
+	ctx := context.Background()
+	for name, dbCtx := range gj.databases {
+		if dbCtx.schema == nil {
+			continue
+		}
+		start := time.Now()
+		doc, err := g.GenerateDiscovery(ctx, name)
+		if err != nil {
+			gj.log.Printf("ERR discovery: %s: %v", name, err)
+			continue
+		}
+		dbLabel := name
+		if dbLabel == "" {
+			dbLabel = "(default)"
+		}
+		tables := len(dbCtx.schema.GetTables())
+		gj.log.Printf("INF discovery: %s — %d tables, %d bytes, hash %s (%s)",
+			dbLabel, tables, len(doc.Markdown), doc.Hash, time.Since(start).Round(time.Millisecond))
+	}
+}
+
 // NewGraphJin creates the GraphJin struct, this involves querying the database to learn its
 // schemas and relationships
 func NewGraphJin(conf *Config, db *sql.DB, options ...Option) (g *GraphJin, err error) {
@@ -142,6 +231,9 @@ func NewGraphJin(conf *Config, db *sql.DB, options ...Option) (g *GraphJin, err
 		g = nil
 		return
 	}
+
+	g.generateAllDiscovery()
+	g.fireAllSchemaCallbacks()
 	return
 }
 
@@ -157,6 +249,9 @@ func NewGraphJinWithFS(conf *Config, db *sql.DB, fs FS, options ...Option) (g *G
 		g = nil
 		return
 	}
+
+	g.generateAllDiscovery()
+	g.fireAllSchemaCallbacks()
 	return
 }
 
@@ -657,7 +752,12 @@ func (g *GraphJin) Reload() error {
 	if pdb := gj.primaryDB(); pdb != nil {
 		db = pdb.db
 	}
-	return g.newGraphJin(gj.conf, db, nil, gj.fs, gj.opts...)
+	if err := g.newGraphJin(gj.conf, db, nil, gj.fs, gj.opts...); err != nil {
+		return err
+	}
+	g.generateAllDiscovery()
+	g.fireAllSchemaCallbacks()
+	return nil
 }
 
 // ReloadWithDB redoes database discover with a new primary DB connection.
@@ -803,6 +903,7 @@ type TableSchema struct {
 	Type          string       `json:"type"`
 	Comment       string       `json:"comment,omitempty"`
 	PrimaryKey    string       `json:"primary_key,omitempty"`
+	PrimaryKeys   []string     `json:"primary_keys,omitempty"`
 	Columns       []ColumnInfo `json:"columns"`
 	Relationships struct {
 		Outgoing []RelationInfo `json:"outgoing"` // Tables this table references
@@ -939,6 +1040,9 @@ func (gj *graphjinEngine) buildTableSchema(dbSchema *sdata.DBSchema, dbName, tab
 	if t.PrimaryCol.Name != "" {
 		schema.PrimaryKey = t.PrimaryCol.Name
 	}
+	if len(t.PrimaryCols) > 1 {
+		schema.PrimaryKeys = t.PKColNames()
+	}
 
 	// Add columns
 	for _, col := range t.Columns {
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+tests-large/02_adventureworks_data.sql filter=lfs diff=lfs merge=lfs -text`