feat: add GetApplicationRepository method to Container for lazy initi…

…alization of ApplicationRepository
cartesi · Jan 31, 2025 · 0bcdbb8 · 0bcdbb8
1 parent 300fc90
commit 0bcdbb8
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/pkg/convenience/container.go b/pkg/convenience/container.go
@@ -27,6 +27,7 @@ type Container struct {
 	AutoCount              bool
 	rawInputRefRepository  *repository.RawInputRefRepository
 	rawOutputRefRepository *repository.RawOutputRefRepository
+	appRepository          *repository.ApplicationRepository
 }
 
 func NewContainer(db sqlx.DB, autoCount bool) *Container {
@@ -36,6 +37,20 @@ func NewContainer(db sqlx.DB, autoCount bool) *Container {
 	}
 }
 
+func (c *Container) GetApplicationRepository() *repository.ApplicationRepository {
+	if c.appRepository != nil {
+		return c.appRepository
+	}
+	c.appRepository = &repository.ApplicationRepository{
+		Db: *c.db,
+	}
+	err := c.appRepository.CreateTables()
+	if err != nil {
+		panic(err)
+	}
+	return c.appRepository
+}
+
 func (c *Container) GetOutputDecoder() *decoder.OutputDecoder {
 	if c.outputDecoder != nil {
 		return c.outputDecoder