Make constructor return concrete type (#44)

sunboyy · Feb 27, 2024 · 7bb44ba · 7bb44ba
1 parent 4992f43
commit 7bb44ba
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/internal/mongo/generator.go b/internal/mongo/generator.go
@@ -76,7 +76,9 @@ func (g RepositoryGenerator) GenerateConstructor() (codegen.FunctionBuilder, err
 			},
 		},
 		Returns: []code.Type{
-			code.SimpleType(g.InterfaceName),
+			code.PointerType{
+				ContainedType: code.SimpleType(g.repoImplStructName()),
+			},
 		},
 		Body: codegen.FunctionBody{
 			codegen.ReturnStatement{

diff --git a/test/generator_test_expected.txt b/test/generator_test_expected.txt
@@ -10,7 +10,7 @@ import (
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
-func NewUserRepository(collection *mongo.Collection) UserRepository {
+func NewUserRepository(collection *mongo.Collection) *UserRepositoryMongo {
 	return &UserRepositoryMongo{
 		collection: collection,
 	}