Generate interfaces for tables

RadhiFadlillah · Aug 11, 2021 · 9fe1309 · 9fe1309
1 parent c8b7c5e
commit 9fe1309
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/internal/templates/interfaces.txt b/internal/templates/interfaces.txt
@@ -10,6 +10,23 @@ import (
 	{{end -}}
 )
 
+{{if gt (len .ddlQueries) 0}}
+// Interfaces for generated tables.
+
+	{{range $query := .ddlQueries -}}
+		{{if gt (len $query.Columns) 0}}
+			{{$interfaceName := (print "Itf" (camel $query.TableName)) -}}
+			// {{$interfaceName}} is interface for table "{{$query.TableName}}".
+			type {{$interfaceName}} interface {
+				{{range $col := $query.Columns -}}
+					{{$funcName := print "Get" (camel $col.Name) -}}
+					{{$funcName}}() interface{}
+				{{end -}}
+			}
+		{{end}}
+	{{end}}
+{{end}}
+
 {{if gt (len .selectQueries) 0}}
 // Interfaces for SELECT queries.