add prefix to tablename in models

BK1031 · May 21, 2024 · 30e3340 · 30e3340
1 parent 874f956
commit 30e3340
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/model/route.go b/model/route.go
@@ -1,6 +1,9 @@
 package model
 
-import "time"
+import (
+	"rincon/config"
+	"time"
+)
 
 type Route struct {
 	Route       string    `json:"route" gorm:"primaryKey"`
@@ -9,7 +12,7 @@ type Route struct {
 }
 
 func (Route) TableName() string {
-	return "route"
+	return config.DatabaseTablePrefix + "route"
 }
 
 type RouteNode struct {

diff --git a/model/service.go b/model/service.go
@@ -1,6 +1,9 @@
 package model
 
-import "time"
+import (
+	"rincon/config"
+	"time"
+)
 
 type Service struct {
 	ID          int       `json:"id" gorm:"primaryKey"`
@@ -13,7 +16,7 @@ type Service struct {
 }
 
 func (Service) TableName() string {
-	return "service"
+	return config.DatabaseTablePrefix + "service"
 }
 
 type ServiceDependency struct {
@@ -25,5 +28,5 @@ type ServiceDependency struct {
 }
 
 func (ServiceDependency) TableName() string {
-	return "service_dependency"
+	return config.DatabaseTablePrefix + "service_dependency"
 }