Blame view

src/shared/ModelBaseMysql.lua 16.3 KB
0de80321   liuzujun   创建游戏数据库,role对应mys...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
  local ModelBaseMysql = class("ModelBaseMysql")
  ModelBaseMysql.key = "key"
  ModelBaseMysql.schema = {}
  
  local string_format = string.format
  local table_insert = table.insert
  local table_unpack = table.unpack
  local assert = assert
  local next = next
  local ipairs = ipairs
  local pairs = pairs
  local tostring = tostring
  local tonumber = tonumber
  local mysqlproxy = mysqlproxy
  
  local function filterProperties(properties, filter)
      for i, field in ipairs(filter) do
          properties[field] = nil
      end
  end
  
  function ModelBaseMysql:ctor(properties)
      self.cacheFields = {} --缓存字段 不更新数据库的字段
  
      self[self.class.key .. "_"] = properties[self.class.key] --数据库key
      properties[self.class.key] = nil
  
      if not self:isValidKey() then
          print(string_format("%s [%s:key] should be give in new(ctor)", tostring(self), self.class.__cname))
          return
      end
  
      if type(properties) ~= "table" then properties = {} end
      self:loadProperties(properties)  --缺少的域将设置默认值
      self:getPriKey()
  end
  
0de80321   liuzujun   创建游戏数据库,role对应mys...
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
  --[[--
  
  返回对象的 ID 值。
  
  **Returns:**
  
  -   ID 
  
  ]]
  function ModelBaseMysql:getKey()
      local id = self[self.class.key .. "_"]
      assert(id ~= nil, string_format("%s [%s:getKey()] Invalid key", tostring(self), self.class.__cname))
      return id
  end
  
  function ModelBaseMysql:getPriKey()
      for k, v in pairs(self.class.schema) do
          local objType, def, keyType, length = table_unpack(v)
2ca93972   liuzujun   添加邮件表
56
          if keyType == "pri" or keyType == "pri_auto" then
0de80321   liuzujun   创建游戏数据库,role对应mys...
57
              self.pri_key = k
fa992c94   liuzujun   添加daily,diner,act...
58
              break
0de80321   liuzujun   创建游戏数据库,role对应mys...
59
60
61
62
63
64
65
66
67
          end
      end
  end
  
  function ModelBaseMysql:load(properties)
      if not self:isValidKey() then
          print(string_format("%s [%s:id] should be set before load", tostring(self), self.class.__cname))
          return false
      end
a12bfcce   liuzujun   添加英雄表
68
      local load = false
0de80321   liuzujun   创建游戏数据库,role对应mys...
69
      if not properties then
913e070e   liuzujun   添加订单表,全局id定时回写数据库
70
          properties = mysqlproxy:query(string_format("SELECT * from `%s` where `%s` = %s;", self.class.__cname, self.pri_key, self:getKey()))
a12bfcce   liuzujun   添加英雄表
71
          load = true
0de80321   liuzujun   创建游戏数据库,role对应mys...
72
73
74
      end
      if not next(properties) then return false end
  
a12bfcce   liuzujun   添加英雄表
75
76
77
      local data = load and properties[1] or properties
  
      self:loadProperties(data)
0de80321   liuzujun   创建游戏数据库,role对应mys...
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
  
      self:onLoad()
  
      return true
  end
  
  --创建model对应的redis数据, 必须已经设置了ID
  function ModelBaseMysql:create()
      if not self:isValidKey() then
          print(string_format("%s [%s:key] should be set before create", tostring(self), self.class.__cname))
          return nil
      end
  
      self:save()
      self:onCreate()
  
      return self
  end
  
  -- save 忽略 缓存配置
  function ModelBaseMysql:save()
      local redisProperties = self:getProperties()
  
      local params = {}
      for fieldName, value in pairs(redisProperties) do
          local propname = fieldName .. "_"
          if self.class.schema[fieldName][1] == "table" then
              if not next(self[propname]) then 
                  params[fieldName] = "NULL"
              else
6af09a92   liuzujun   mysql二进制数据加上quote...
108
109
110
                  local result = mysqlproxy:quote_sql_str(MsgPack.pack(self[propname]))
                  --params[fieldName] = "'" .. MsgPack.pack(self[propname]) .. "'"
                  params[fieldName] = result
0de80321   liuzujun   创建游戏数据库,role对应mys...
111
112
              end
          elseif self.class.schema[fieldName][1] == "string" then
6af09a92   liuzujun   mysql二进制数据加上quote...
113
114
115
              local result = mysqlproxy:quote_sql_str(self[propname])
              --params[fieldName] = "'" .. self[propname] .. "'"
              params[fieldName] = result
0de80321   liuzujun   创建游戏数据库,role对应mys...
116
117
118
119
120
121
          else
              params[fieldName] = self[propname]
          end
      end
      if next(params) then
          -- insert update
913e070e   liuzujun   添加订单表,全局id定时回写数据库
122
          local sql = "INSERT INTO `%s` (%s) VALUES (%s) ON DUPLICATE KEY UPDATE %s;"
0de80321   liuzujun   创建游戏数据库,role对应mys...
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
          local tbName = self.class.__cname
          local key_list = ""
          local value_list = ""
          local update_list = ""
          for k, v in pairs(params) do
              if key_list ~= "" then
                 key_list = key_list .. ","
              end
              if value_list ~= "" then
                 value_list = value_list .. ","
              end
              if update_list ~= "" then
                 update_list = update_list .. ","
              end
  
fa992c94   liuzujun   添加daily,diner,act...
138
              key_list = key_list .. "`" .. k .. "`"
0de80321   liuzujun   创建游戏数据库,role对应mys...
139
              value_list = value_list .. v
fa992c94   liuzujun   添加daily,diner,act...
140
              update_list = update_list .. "`" .. k .. "`=" .. v
0de80321   liuzujun   创建游戏数据库,role对应mys...
141
142
          end
          sql = string_format(sql, tbName, key_list, value_list, update_list)
6af09a92   liuzujun   mysql二进制数据加上quote...
143
          local res = mysqlproxy:query(sql)
6136eaca   liuzujun   添加好友表
144
145
          if res["errno"] then
              skynet.error(sql)
01be2d78   liuzujun   修改mysql链接断开重连的bug
146
              skynet.error(res["err"])
6136eaca   liuzujun   添加好友表
147
          end
0de80321   liuzujun   创建游戏数据库,role对应mys...
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
      end
  end
  
  --[[--
  
  确定对象是否设置了有效的 key
  
  ]]
  function ModelBaseMysql:isValidKey()
      local propname = self.class.key .. "_"
      local key = self[propname]
      return type(key) == "string" and key ~= ""
  end
  
  --[[--
  
  加载对象的属性进内存。
  NOTE: 如果properties缺少schema中的域, 将用默认值来填充
  
  **Parameters:**
  
  -   properties: 包含属性值的数组
  
  ]]
  function ModelBaseMysql:loadProperties(properties)
      assert(type(properties) == "table", "Invalid properties")
      for field, schema in pairs(self.class.schema) do
          local typ, def = table_unpack(schema)
          local propname = field .. "_"
  
          if typ == "table" and type(properties[field]) == "string" then
              properties[field] = MsgPack.unpack(properties[field])
          end
  
          local val = properties[field] or def
          if val ~= nil then
              if typ == "number" then val = tonumber(val) end
              assert(type(val) == typ,
                 string_format("%s [%s:loadProperties()] Type mismatch, %s expected %s, actual is %s",
                                   tostring(self), self.class.__cname, field, typ, type(val)))
              self[propname] = val
          end
      end
  end
  
  --[[--
  
  取得对象的属性值。
  
  **Parameters:**
  
  -   fields: 要取得哪些属性的值,如果未指定该参数,则返回 fields 中设定的属性
  -   filter: 要从结果中过滤掉哪些属性,如果未指定则不过滤
  
  **Returns:**
  
  -   包含属性值的数组
  
  ]]
  function ModelBaseMysql:getProperties(fields, filter)
      local schema = self.class.schema
      if type(fields) ~= "table" then fields = table.keys(self.class.schema) end
  
      local properties = {}
      for i, field in ipairs(fields) do
          local propname = field .. "_"
          local typ = schema[field][1]
          local val = self[propname]
          assert(type(val) == typ,
                 string_format("%s [%s:getProperties()] Type mismatch, %s expected %s, actual is %s",
                                   tostring(self), self.class.__cname, field, typ, type(val)))
          properties[field] = val
      end
  
      if type(filter) == "table" then
          filterProperties(properties, filter)
      end
  
      return properties
  end
  
  function ModelBaseMysql:getProperty(property)
      if type(property) ~= "string" then return nil end
      if not self.class.schema[property] then return nil end
      return self:getProperties({property})[property]
  end
  
6136eaca   liuzujun   添加好友表
235
  function ModelBaseMysql:setProperty(property, value, forceSave)
0de80321   liuzujun   创建游戏数据库,role对应mys...
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
      if not self.class.schema[property] then
          print(string_format("%s [%s:setProperty()] Invalid property : %s",
              tostring(self), self.class.__cname, property))
          return
      end
  
      local typ, def = table_unpack(self.class.schema[property])
      local propname = property .. "_"
  
      if typ == "number" then value = tonumber(value) end
      if typ == "table" and not value then
          value = self[propname] -- table 可以用自己的缓冲
      end
      assert(type(value) == typ,
         string_format("%s [%s:setProperties()] Type mismatch, %s expected %s, actual is %s",
           tostring(self), self.class.__cname, property, typ, type(value)))
0de80321   liuzujun   创建游戏数据库,role对应mys...
252
  
6136eaca   liuzujun   添加好友表
253
254
255
256
257
258
259
260
261
262
263
      if typ == "number" or typ == "string" then
          if self[propname] == value then
              return
          end
      end
      self[propname] = value
      --self:save()
      self.cacheFields[property] = self[propname]
      if forceSave then
          self:update()
      end
0de80321   liuzujun   创建游戏数据库,role对应mys...
264
265
  end
  
6136eaca   liuzujun   添加好友表
266
  function ModelBaseMysql:setProperties(fields, forceSave)
0de80321   liuzujun   创建游戏数据库,role对应mys...
267
268
269
270
271
272
273
274
275
276
277
278
279
280
      for property, value in pairs(fields) do
           if not self.class.schema[property] then
              print(string_format("%s [%s:setProperty()] Invalid property : %s",
                  tostring(self), self.class.__cname, property))
          else
              local typ, def = table_unpack(self.class.schema[property])
              local propname = property .. "_"
              if typ == "number" then value = tonumber(value) end
              if typ == "table" and not value then
                  value = self[propname] -- table 可以用自己的缓冲
              end
              assert(type(value) == typ,
                 string_format("%s [%s:setProperties()] Type mismatch, %s expected %s, actual is %s",
                   tostring(self), self.class.__cname, property, typ, type(value)))
0de80321   liuzujun   创建游戏数据库,role对应mys...
281
  
6136eaca   liuzujun   添加好友表
282
              if typ == "number" or typ == "string" then
ab57e0ab   liuzujun   修改数据存储bug
283
284
285
                  if self[propname] ~= value then
                      self[propname] = value
                      self.cacheFields[property] = self[propname]
6136eaca   liuzujun   添加好友表
286
                  end
0de80321   liuzujun   创建游戏数据库,role对应mys...
287
              end
0de80321   liuzujun   创建游戏数据库,role对应mys...
288
289
          end
      end
6136eaca   liuzujun   添加好友表
290
291
292
293
      if forceSave then
          self:update()
      end
      --self:save()
0de80321   liuzujun   创建游戏数据库,role对应mys...
294
295
  end
  
6136eaca   liuzujun   添加好友表
296
  function ModelBaseMysql:incrProperty(property, value, forceSave)
0de80321   liuzujun   创建游戏数据库,role对应mys...
297
298
299
300
301
302
303
304
305
      if not self.class.schema[property] then
          print(string_format("%s [%s:setProperty()] Invalid property : %s",
              tostring(self), self.class.__cname, property))
          return
      end
  
      local typ, def = table_unpack(self.class.schema[property])
      local propname = property .. "_"
  
6136eaca   liuzujun   添加好友表
306
      if typ == "table" or typ == "string" then return end
0de80321   liuzujun   创建游戏数据库,role对应mys...
307
308
      if typ == "number" then value = tonumber(value) end
  
6136eaca   liuzujun   添加好友表
309
      self:setProperty(property, self[propname] + value, forceSave)
0de80321   liuzujun   创建游戏数据库,role对应mys...
310
311
312
313
314
315
316
317
  end
  
  function ModelBaseMysql:onLoad()
  end
  
  function ModelBaseMysql:onCreate()
  end
  
fa992c94   liuzujun   添加daily,diner,act...
318
319
320
321
322
  function ModelBaseMysql:checkKeyExists(key)
      local res = mysqlproxy:query(string_format("SELECT * FROM `%s` WHERE `%s` = %s", self.class.__cname, self.pri_key, key))
      return next(res)
  end
  
0de80321   liuzujun   创建游戏数据库,role对应mys...
323
324
325
  function ModelBaseMysql:checkTableSchema()
      -- 1.检测是否表存在
      local typeMap = {
2ca93972   liuzujun   添加邮件表
326
          number = {"int", 0},
0de80321   liuzujun   创建游戏数据库,role对应mys...
327
          string = {"varchar", "", 128},
a12bfcce   liuzujun   添加英雄表
328
329
          table = {"blob", "NULL"},
          pri = {"bigint", 0},
0de80321   liuzujun   创建游戏数据库,role对应mys...
330
331
332
333
334
      }
      local tbName = self.class.__cname
      local create_sql = [[
  		CREATE TABLE IF NOT EXISTS `%s` (
              %s
a12bfcce   liuzujun   添加英雄表
335
              PRIMARY KEY (`%s`)%s
0de80321   liuzujun   创建游戏数据库,role对应mys...
336
337
  			) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci;
      ]]
a12bfcce   liuzujun   添加英雄表
338
339
      local index_tpl_str = [[,INDEX `%s_Index` (`%s`)]]
      
0de80321   liuzujun   创建游戏数据库,role对应mys...
340
341
342
343
344
345
      local alter_sql = [[
  		ALTER TABLE `%s` ADD COLUMN (
              %s
  		) ;
      ]]
      local field_tpl_str = "`%s` %s%s DEFAULT %s"
2ca93972   liuzujun   添加邮件表
346
      local auto_increment_str = "`%s` %s NOT NULL AUTO_INCREMENT"
0de80321   liuzujun   创建游戏数据库,role对应mys...
347
348
      local field_str = ""
  
2ca93972   liuzujun   添加邮件表
349
      local res = mysqlproxy:query("desc `".. tbName .. "`;")
a12bfcce   liuzujun   添加英雄表
350
      local keyList = {}
0de80321   liuzujun   创建游戏数据库,role对应mys...
351
352
353
354
      if res["err"] then -- 表不存在
          local schema = {}
          for k, v in pairs(self.class.schema) do
              local keyType = v[3]
2ca93972   liuzujun   添加邮件表
355
              if keyType == "pri" or keyType == "pri_auto" then
0de80321   liuzujun   创建游戏数据库,role对应mys...
356
357
358
                  self.pri_key = k
                  table_insert(schema, 1, {k, v})
              else
a12bfcce   liuzujun   添加英雄表
359
360
361
                  if keyType == "index" then
                      table_insert(keyList, k)
                  end
0de80321   liuzujun   创建游戏数据库,role对应mys...
362
363
364
365
366
367
                  table_insert(schema, {k, v})
              end
          end
          for _, tbl in ipairs(schema) do
              local k, v = tbl[1], tbl[2]
              local objType, def, keyType, length = table_unpack(v)
2ca93972   liuzujun   添加邮件表
368
              local isAutoPriKey = false
0de80321   liuzujun   创建游戏数据库,role对应mys...
369
              assert(typeMap[objType], string_format("schema invalid type, %s, %s", tbName, k))
a12bfcce   liuzujun   添加英雄表
370
371
              -- 主键使用bigint存储
              if keyType == "pri" then objType = "pri" end
2ca93972   liuzujun   添加邮件表
372
373
374
375
376
377
              if keyType == "pri" or keyType == "pri_auto" then
                  objType = "pri"
                  if keyType == "pri_auto" then
                      isAutoPriKey = true
                  end
              end
0de80321   liuzujun   创建游戏数据库,role对应mys...
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
  
              local info = typeMap[objType]
              local suffix = ""
              local fieldType = info[1]
              if objType == "table" or not def or def == "" then def = info[2] end
              if type(def) == "string" and def ~= "NULL" then def = "'" .. def .. "'" end
              if info[3] and not length then length = info[3] end
              -- 设置字段长度
              if info[3] then suffix = string.format("(%d)", length) end
              -- 很长的string使用blob
              if  keyType == "blob" then 
                  fieldType = keyType
                  suffix = ""
                  def = "NULL"
              end
  
2ca93972   liuzujun   添加邮件表
394
395
396
397
398
              if not isAutoPriKey then
                  field_str = field_str .. string.format(field_tpl_str..",", k, fieldType, suffix, def)
              else
                  field_str = field_str .. string.format(auto_increment_str..",", k, fieldType)
              end
0de80321   liuzujun   创建游戏数据库,role对应mys...
399
400
401
          end
  
          assert(self.pri_key, string_format("table not include primary key, [%s]", tbName))
a12bfcce   liuzujun   添加英雄表
402
403
404
405
          local index_key_str = ""
          for _, k in ipairs(keyList) do
              index_key_str = index_key_str .. string_format(index_tpl_str, k, k)
          end
0de80321   liuzujun   创建游戏数据库,role对应mys...
406
          -- 创建表格
2ca93972   liuzujun   添加邮件表
407
          print(string_format(create_sql, tbName, field_str, self.pri_key, index_key_str))
a12bfcce   liuzujun   添加英雄表
408
          mysqlproxy:query(string_format(create_sql, tbName, field_str, self.pri_key, index_key_str))
0de80321   liuzujun   创建游戏数据库,role对应mys...
409
410
411
412
413
414
415
416
      else    -- 检测是否有添加新字段
          local addCol = {}
          local curCols = {}
          for _, col in ipairs(res) do
              curCols[col["Field"]] = 1
          end
          for k, v in pairs(self.class.schema) do
              local objType, def, keyType, length = table_unpack(v)
0de80321   liuzujun   创建游戏数据库,role对应mys...
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
              if not curCols[k] then
                  print(string_format("table [%s] add new column [%s]", tbName, k))
                  assert(typeMap[objType], string_format("schema invalid type, [%s], [%s]", tbName, k))
  
                  local info = typeMap[objType]
                  local suffix = ""
                  local fieldType = info[1]
                  if objType == "table" or not def or def == "" then def = info[2] end
                  if type(def) == "string" and def ~= "NULL" then def = "'" .. def .. "'" end
                  if info[3] and not length then length = info[3] end
                  -- 设置字段长度
                  if info[3] then suffix = string.format("(%d)", length) end
                  -- 很长的string使用blob
                  if  keyType == "blob" then 
                      fieldType = keyType
                      suffix = ""
                      def = "NULL"
                  end
                  local sep = ","
                  if field_str == "" then
                      sep = ""
                  end
                  field_str = field_str .. string.format(sep..field_tpl_str, k, fieldType, suffix, def)
              end
          end
          -- 添加新列
          if field_str ~= "" then
              mysqlproxy:query(string_format(alter_sql, tbName, field_str))
          end
      end
  
  end
  
6136eaca   liuzujun   添加好友表
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
  function  ModelBaseMysql:loadFields(fields)
      if not self:isValidKey() then
          print(string_format("%s [%s:id] should be set before load", tostring(self), self.class.__cname))
          return
      end
      if not next(fields) then
          return
      end
      local final = {}
      for _, v in ipairs(fields) do
          table.insert(final, '`'..v..'`')
      end
      local field_list = table.concat(final, ",")
      local res = mysqlproxy:query(string_format("SELECT %s from `%s` where `%s` = %s;", field_list, self.class.__cname, self.pri_key, self:getKey()))
      if res["errno"] then
          return
      end
  
      return res[1]
  end
  
  function ModelBaseMysql:update()
      if next(self.cacheFields) then
          self:updateFields(self.cacheFields)
          self.cacheFields = {}
      end
  end
  
  function ModelBaseMysql:updateFields(fields)
      local params = {}
      for field, value in pairs(fields) do
          if self.class.schema[field][1] == "table" then
              if next(value) then 
                  local result = mysqlproxy:quote_sql_str(MsgPack.pack(value))
                  params[field] = result
              end
          elseif self.class.schema[field][1] == "string" then
              local result = mysqlproxy:quote_sql_str(value)
              params[field] = result
          else
              params[field] = value 
          end
      end
      if next(params) then
          local sql = "UPDATE `%s` SET %s WHERE `%s` = %s;"
          local tbName = self.class.__cname
          local tmp = {}
          for k, v in pairs(params) do
              table.insert(tmp, '`' .. k .. '` = ' .. v)
          end
          sql = string_format(sql, tbName, table.concat(tmp, ","), self.pri_key, self:getKey())
          local res = mysqlproxy:query(sql)
          if res["errno"] then
              skynet.error(sql)
01be2d78   liuzujun   修改mysql链接断开重连的bug
504
              skynet.error(res["err"])
6136eaca   liuzujun   添加好友表
505
506
507
508
509
510
              return false
          end
      end
      return true
  end
  
0de80321   liuzujun   创建游戏数据库,role对应mys...
511
  return ModelBaseMysql