From 72df840b09a902577709c35d39aa2dc6ec3fc91a Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Tue, 5 Nov 2024 15:06:14 +0800
Subject: [PATCH] 1

---
 server/wheelserver/app/app.go   | 9 +++++++++
 server/wheelserver/mt/export.go | 4 ++--
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/server/wheelserver/app/app.go b/server/wheelserver/app/app.go
index 9bbb9a96..7c7d6434 100644
--- a/server/wheelserver/app/app.go
+++ b/server/wheelserver/app/app.go
@@ -3,6 +3,7 @@ package app
 import (
 	"f5"
 	"main/mt"
+	"main/constant"
 )
 
 type app struct {
@@ -38,6 +39,14 @@ func (this *app) Update() {
 }
 
 func (this *app) registerDataSources() {
+	f5.GetApp().RegisterOrmDb(
+		constant.WHEEL_DB,
+		mt.Table.WheelDb.GetById(0).GetHost(),
+		mt.Table.WheelDb.GetById(0).GetPort(),
+		mt.Table.WheelDb.GetById(0).GetUser(),
+		mt.Table.WheelDb.GetById(0).GetPasswd(),
+		mt.Table.WheelDb.GetById(0).GetDatabase(),
+	)
 }
 
 func (this *app) HasTask() bool {
diff --git a/server/wheelserver/mt/export.go b/server/wheelserver/mt/export.go
index 593b4cd6..20cb8952 100644
--- a/server/wheelserver/mt/export.go
+++ b/server/wheelserver/mt/export.go
@@ -8,7 +8,7 @@ type table struct {
 	WheelServerCluster *WheelServerClusterTable
 	Config      *ConfigTable
 	Web3ServiceCluster *Web3ServiceClusterTable
-	WheelDbCluster *WheelDbTable
+	WheelDb *WheelDbTable
 }
 
 var Table = f5.New(func(this *table) {
@@ -27,7 +27,7 @@ var Table = f5.New(func(this *table) {
 		this.PrimKey = ""
 	})
 
-	this.WheelDbCluster = f5.New(func(this *WheelDbTable) {
+	this.WheelDb = f5.New(func(this *WheelDbTable) {
 		this.FileName = "../config/wheeldb.json"
 		this.PrimKey = ""
 	})