Blame view

src/services/dbseed.lua 6.83 KB
314bc5df   zhengshouren   提交服务器初始代码
1
2
3
4
5
6
  require "shared.init"
  require "utils.init"
  require "GlobalVar"
  require "RedisKeys"
  require "ProtocolCode"
  require "skynet.manager"
913e070e   测试   添加订单表,全局id定时回写数据库
7
  require "utils.MysqlUtil"
314bc5df   zhengshouren   提交服务器初始代码
8
9
10
11
  
  skynet = require "skynet"
  
  redisproxy = require("shared.redisproxy")
0de80321   测试   创建游戏数据库,role对应mys...
12
  mysqlproxy = require "shared.mysqlproxy"
314bc5df   zhengshouren   提交服务器初始代码
13
14
15
  
  SendPacket = function ( ... ) end
  
913e070e   测试   添加订单表,全局id定时回写数据库
16
  
314bc5df   zhengshouren   提交服务器初始代码
17
  local function initRedisDb( ... )
913e070e   测试   添加订单表,全局id定时回写数据库
18
19
20
21
22
23
24
25
26
27
28
  	local function initAutoIncrementUid(tbName, keyName, fieldName)
  		if not fieldName then fieldName = "value" end
  		local mysqlVal = getDbCfgVal(tbName, keyName, fieldName)
  		if not mysqlVal then 
  			skynet.error(string.format("get db cfg fail, table %s, key %s, field %s", tbName, keyName, fieldName))
  			return
  		end
  		local redisVal = tonum(redisproxy:hget(tbName, keyName))
  		if redisVal < mysqlVal then
  			redisproxy:hset(tbName, keyName, mysqlVal)
  		end
314bc5df   zhengshouren   提交服务器初始代码
29
  	end
913e070e   测试   添加订单表,全局id定时回写数据库
30
31
32
33
34
35
36
  
  	initAutoIncrementUid("autoincrement_set", "role")
  	initAutoIncrementUid("autoincrement_set", "union")
  	initAutoIncrementUid("autoincrement_set", "order")
  	initAutoIncrementUid("autoincrement_set", "email")
  	initAutoIncrementUid("autoincrement_set", "emailTimestamp")
  	initAutoIncrementUid("autoincrement_set", "delay_email")
6136eaca   测试   添加好友表
37
38
  	initAutoIncrementUid("autoincrement_set", "stopcreate")
  	initAutoIncrementUid("autoincrement_set", "maintain")
913e070e   测试   添加订单表,全局id定时回写数据库
39
40
41
42
  
  	--redisproxy:hsetnx("adv_season", "idx", 0)
  	--redisproxy:hsetnx("adv_season", "chapter", globalCsv.adv_endless_default_chapter)
  	--redisproxy:hsetnx("adv_season", "overTime", 0)
314bc5df   zhengshouren   提交服务器初始代码
43
44
  end
  
0de80321   测试   创建游戏数据库,role对应mys...
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
  -- 初始化服务器数据库以及服务器信息表
  local function initServerDatabase()
  	local servId = skynet.getenv("servId")
  	mysqlproxy:query(string.format("CREATE DATABASE IF NOT EXISTS server_%s DEFAULT CHARSET = utf8mb4 COLLATE utf8mb4_general_ci;", servId))
  	mysqlproxy:query(string.format("use server_%s", servId))
  
  	-- 服务器信息表 开服时间
  	mysqlproxy:query [[
  		CREATE TABLE IF NOT EXISTS `server_info` (
  			`key` varchar(45) NOT NULL,
  			`int_value` int(11) DEFAULT NULL,
  			`str_value` varchar(128) DEFAULT NULL,
  			PRIMARY KEY (`key`)
  			) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
  	]]
  
  	local res = mysqlproxy:query("SELECT * FROM `server_info` where `key` = 'server_start';")
  	if not next(res) then
  		mysqlproxy:query(string.format("INSERT INTO `server_info`(`key`, `str_value`) VALUES('server_start', '%s');",
  			os.date("%Y%m%d", skynet.timex())))
  	end
  end
  
  local function initAutoIncreUidTable()
  	mysqlproxy:query [[
913e070e   测试   添加订单表,全局id定时回写数据库
70
  		CREATE TABLE IF NOT EXISTS `autoincrement_set` (
0de80321   测试   创建游戏数据库,role对应mys...
71
72
73
74
75
76
77
  			`key` varchar(45) NOT NULL,
  			`value` int(11) DEFAULT NULL,
  			PRIMARY KEY (`key`)
  			) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
  	]]
  	local servId = tonumber(skynet.getenv("servId"))
  	if servId then
913e070e   测试   添加订单表,全局id定时回写数据库
78
  		local tpl = "INSERT INTO `autoincrement_set`(`key`, `value`) values('%s', %d)"
0de80321   测试   创建游戏数据库,role对应mys...
79
80
  		mysqlproxy:query(string.format(tpl, "role", servId * MAX_ROLE_NUM))
  		mysqlproxy:query(string.format(tpl, "union", servId * MAX_ROLE_NUM))
913e070e   测试   添加订单表,全局id定时回写数据库
81
  		mysqlproxy:query(string.format(tpl, "order", 0))
0de80321   测试   创建游戏数据库,role对应mys...
82
83
84
  		mysqlproxy:query(string.format(tpl, "email", 0))
  		mysqlproxy:query(string.format(tpl, "emailTimestamp", 0))
  		mysqlproxy:query(string.format(tpl, "delay_email", 0))
6136eaca   测试   添加好友表
85
86
  		mysqlproxy:query(string.format(tpl, "stopcreate", 0))
  		mysqlproxy:query(string.format(tpl, "maintain", 0))
0de80321   测试   创建游戏数据库,role对应mys...
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
  	end
  end
  
  local function initAdvSeasonTable()
  	mysqlproxy:query [[
  		CREATE TABLE IF NOT EXISTS `adv_season` (
  			`key` varchar(45) NOT NULL,
  			`value` int(11) DEFAULT NULL,
  			PRIMARY KEY (`key`)
  			) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
  	]]
  	local servId = tonumber(skynet.getenv("servId"))
  	if servId then
  		local tpl = "INSERT INTO `adv_season`(`key`, `value`) values('%s', %d)"
  
  		mysqlproxy:query(string.format(tpl, "idx", 0))
  		mysqlproxy:query(string.format(tpl, "chapter", globalCsv.adv_endless_default_chapter))
  		mysqlproxy:query(string.format(tpl, "overTime", 0))
  	end
  end
  
  local function checkRoleTables()
6136eaca   测试   添加好友表
109
  	local list = {"Role", "Daily", "Activity", "Diner", "Store", "Hero", "RoleIncre", "Rune", "Order", "Email", "Friend"}
fa992c94   测试   添加daily,diner,act...
110
111
112
113
114
115
  	for _, name in ipairs(list) do
  		local obj = require("models."..name).new({key = "key"})
  		print("check table [" .. name .. "] begin.")
  		obj:checkTableSchema()
  		print("check table [" .. name .. "] end.")
  	end
0de80321   测试   创建游戏数据库,role对应mys...
116
117
  end
  
6136eaca   测试   添加好友表
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
  local function createMysqlSp()
  	mysqlproxy:query "DROP PROCEDURE IF EXISTS `add_friends`"
  	mysqlproxy:query [[
  		CREATE PROCEDURE `add_friends`(IN role_id bigint, IN friend_id bigint, IN add_time int)
  		BEGIN
  			DECLARE t_error INTEGER DEFAULT 0;    
  			DECLARE CONTINUE HANDLER FOR SQLEXCEPTION SET t_error=1;    
      
  			START TRANSACTION;    
  				INSERT INTO `Friend`(`roleid`,`fid`,`addTime`) VALUES(role_id, friend_id, add_time);       
  				INSERT INTO `Friend`(`roleid`,`fid`,`addTime`) VALUES(friend_id, role_id, add_time);       
      
  			IF t_error = 1 THEN    
  				ROLLBACK;    
  			ELSE    
  				COMMIT;    
  			END IF;    
  			select t_error; 
  		END
  	]]
  
  	mysqlproxy:query "DROP PROCEDURE IF EXISTS `del_friends`"
  	mysqlproxy:query [[
  		CREATE PROCEDURE `del_friends`(IN role_id bigint, IN friend_id bigint)
  		BEGIN
  			DECLARE t_error INTEGER DEFAULT 0;    
  			DECLARE CONTINUE HANDLER FOR SQLEXCEPTION SET t_error=1;    
      
  			START TRANSACTION;    
  				DELETE FROM `Friend` WHERE `roleid` = role_id AND `fid` = friend_id;       
  				DELETE FROM `Friend` WHERE `roleid` = friend_id AND `fid` = role_id;       
      
  			IF t_error = 1 THEN    
  				ROLLBACK;    
  			ELSE    
  				COMMIT;    
  			END IF;    
  			select t_error; 
  		END
  	]]
  end
  
314bc5df   zhengshouren   提交服务器初始代码
160
161
  local steps = {
  	[1] = {
0de80321   测试   创建游戏数据库,role对应mys...
162
163
164
  		handler = initServerDatabase,
  		desc = "initialize server database "
  	},
913e070e   测试   添加订单表,全局id定时回写数据库
165
  	[2] = {
0de80321   测试   创建游戏数据库,role对应mys...
166
167
168
  		handler = initAutoIncreUidTable,
  		desc = "initialize auto_increment_uid table "
  	},
913e070e   测试   添加订单表,全局id定时回写数据库
169
  	[3] = {
0de80321   测试   创建游戏数据库,role对应mys...
170
171
172
  		handler = initAdvSeasonTable,
  		desc = "initialize adv_season table "
  	},
913e070e   测试   添加订单表,全局id定时回写数据库
173
  	[4] = {
0de80321   测试   创建游戏数据库,role对应mys...
174
175
  		handler = checkRoleTables,
  		desc = "check role tables "
6136eaca   测试   添加好友表
176
177
178
179
180
  	},
  	[5] = {
  		handler = createMysqlSp,
  		desc = "create mysql store procedure "
  	},
314bc5df   zhengshouren   提交服务器初始代码
181
182
  }
  
6136eaca   测试   添加好友表
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
  local function loadAllUserInfo()
  	local maxId = 0
  	local sql = "SELECT `id`, `uid`, `name` FROM `Role` WHERE `id` > %d ORDER BY `id` LIMIT 1000;"
  	while true do
  		local res = mysqlproxy:query(string.format(sql, maxId))
  		if not next(res) then
  			return
  		else
  			for _, info in ipairs(res) do
  				if info["id"] > maxId then
  					maxId = info["id"]
  				end
  				redisproxy:pipelining(function (red)
  					local dbName = string.upper(info["name"])
  					red:set(string.format("user:%s", dbName), info["id"])
  					red:set(string.format("uid:%s", info["uid"]), dbName)
  				end)
  			end
  		end
  	end
  end
  
314bc5df   zhengshouren   提交服务器初始代码
205
  skynet.start(function ()
0de80321   测试   创建游戏数据库,role对应mys...
206
207
208
209
210
211
  	--local new = redisproxy:hsetnx("autoincrement_set", "server_start", os.date("%Y%m%d", skynet.timex())) == 1
  	--if not new then
  	--	print("server has been initialized...")
  	--	skynet.exit()
  	--	return
  	--end
a5486ede   zhouhaihai   csvdata 修改为 share...
212
213
  	csvdb = require "shared.csvdata"
  	globalCsv = csvdb["GlobalDefineCsv"]
314bc5df   zhengshouren   提交服务器初始代码
214
215
216
217
218
219
  
  	for _, action in ipairs(steps) do
  		print(action.desc .. "start ...")
  		action.handler()
  		print(action.desc .. "finished ...")
  	end
913e070e   测试   添加订单表,全局id定时回写数据库
220
  	initRedisDb()
6136eaca   测试   添加好友表
221
  	loadAllUserInfo()
314bc5df   zhengshouren   提交服务器初始代码
222
223
  	skynet.exit()
  end)