stream

Author: mstgnz

examples/main.go

@@ -11,6 +11,7 @@ import (
 	"github.com/mstgnz/sqlmapper/postgres"
 	"github.com/mstgnz/sqlmapper/sqlite"
 	"github.com/mstgnz/sqlmapper/sqlserver"
+	"github.com/mstgnz/sqlmapper/stream"
 )
 
 // DatabaseType represents supported database types
@@ -92,7 +93,55 @@ func (c *Converter) Convert(sourceType, targetType DatabaseType, content string)
 	return result, nil
 }
 
+// StreamExample demonstrates stream parsing functionality
+func StreamExample() {
+	// Create MySQL stream parser
+	parser := mysql.NewMySQLStreamParser()
+
+	// Open source file
+	file, err := os.Open("examples/files/mysql.sql")
+	if err != nil {
+		log.Fatalf("Failed to open file: %v", err)
+	}
+	defer file.Close()
+
+	// Parse stream and handle objects
+	err = parser.ParseStream(file, func(obj stream.SchemaObject) error {
+		switch obj.Type {
+		case stream.TableObject:
+			if table, ok := obj.Data.(*sqlmapper.Table); ok {
+				fmt.Printf("Found table: %s\n", table.Name)
+			}
+		case stream.ViewObject:
+			if view, ok := obj.Data.(*sqlmapper.View); ok {
+				fmt.Printf("Found view: %s\n", view.Name)
+			}
+		case stream.FunctionObject:
+			if function, ok := obj.Data.(*sqlmapper.Function); ok {
+				fmt.Printf("Found function: %s\n", function.Name)
+			}
+		case stream.ProcedureObject:
+			if procedure, ok := obj.Data.(*sqlmapper.Procedure); ok {
+				fmt.Printf("Found procedure: %s\n", procedure.Name)
+			}
+		case stream.TriggerObject:
+			if trigger, ok := obj.Data.(*sqlmapper.Trigger); ok {
+				fmt.Printf("Found trigger: %s\n", trigger.Name)
+			}
+		}
+		return nil
+	})
+
+	if err != nil {
+		log.Fatalf("Stream parsing error: %v", err)
+	}
+}
+
 func main() {
+	// Stream parsing example
+	fmt.Println("\n=== Stream Parsing Example ===")
+	StreamExample()
+
 	converter := NewConverter()
 
 	// Example conversions

mysql/mysql.go

@@ -20,7 +20,7 @@ type MySQL struct {
 
 // NewMySQL creates and initializes a new MySQL parser instance.
 // It returns a parser that can handle MySQL specific SQL syntax and schema structures.
-func NewMySQL() *MySQL {
+func NewMySQL() sqlmapper.Database {
 	return &MySQL{
 		schema: &sqlmapper.Schema{},
 	}

mysql/mysql_stream.go

@@ -18,7 +18,7 @@ type MySQLStreamParser struct {
 // NewMySQLStreamParser creates a new MySQL stream parser
 func NewMySQLStreamParser() *MySQLStreamParser {
 	return &MySQLStreamParser{
-		mysql: NewMySQL(),
+		mysql: NewMySQL().(*MySQL),
 	}
 }
 

oracle/oracle.go

@@ -20,7 +20,7 @@ type Oracle struct {
 
 // NewOracle creates and initializes a new Oracle parser instance.
 // It returns a parser that can handle Oracle specific SQL syntax and schema structures.
-func NewOracle() *Oracle {
+func NewOracle() sqlmapper.Database {
 	return &Oracle{
 		schema: &sqlmapper.Schema{},
 	}

oracle/oracle_stream.go

@@ -18,7 +18,7 @@ type OracleStreamParser struct {
 // NewOracleStreamParser creates a new Oracle stream parser
 func NewOracleStreamParser() *OracleStreamParser {
 	return &OracleStreamParser{
-		oracle: NewOracle(),
+		oracle: NewOracle().(*Oracle),
 	}
 }
 

postgres/postgres.go

@@ -20,7 +20,7 @@ type PostgreSQL struct {
 
 // NewPostgreSQL creates and initializes a new PostgreSQL parser instance.
 // It returns a parser that can handle PostgreSQL specific SQL syntax and schema structures.
-func NewPostgreSQL() *PostgreSQL {
+func NewPostgreSQL() sqlmapper.Database {
 	return &PostgreSQL{
 		schema: &sqlmapper.Schema{},
 	}

postgres/postgres_stream.go

@@ -18,7 +18,7 @@ type PostgreSQLStreamParser struct {
 // NewPostgreSQLStreamParser creates a new PostgreSQL stream parser
 func NewPostgreSQLStreamParser() *PostgreSQLStreamParser {
 	return &PostgreSQLStreamParser{
-		postgres: NewPostgreSQL(),
+		postgres: NewPostgreSQL().(*PostgreSQL),
 	}
 }
 

schema.go

@@ -3,6 +3,12 @@ package sqlmapper
 // DatabaseType represents the supported database types
 type DatabaseType string
 
+// Database represents the common interface for all database implementations
+type Database interface {
+	Parse(content string) (*Schema, error)
+	Generate(schema *Schema) (string, error)
+}
+
 const (
 	MySQL      DatabaseType = "mysql"
 	PostgreSQL DatabaseType = "postgresql"

sqlite/sqlite.go

@@ -21,7 +21,7 @@ type SQLite struct {
 
 // NewSQLite creates and initializes a new SQLite parser instance.
 // It returns a parser that can handle SQLite specific SQL syntax and schema structures.
-func NewSQLite() *SQLite {
+func NewSQLite() sqlmapper.Database {
 	return &SQLite{
 		schema: &sqlmapper.Schema{},
 		buf:    &bytes.Buffer{},

sqlite/sqlite_stream.go

@@ -18,7 +18,7 @@ type SQLiteStreamParser struct {
 // NewSQLiteStreamParser creates a new SQLite stream parser
 func NewSQLiteStreamParser() *SQLiteStreamParser {
 	return &SQLiteStreamParser{
-		sqlite: NewSQLite(),
+		sqlite: NewSQLite().(*SQLite),
 	}
 }