FilesExpand file tree

 master

/

sche.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

186 lines (166 loc) · 3.88 KB

 master

/

sche.lua

Copy path

Top

File metadata and controls

• Code
• Blame

186 lines (166 loc) · 3.88 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

38

39

40

41

42

43

44

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

70

71

72

73

74

75

76

77

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

108

109

110

111

112

113

114

115

116

117

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

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

local MinHeap = require "lua.minheap"

local LinkQue = require "lua.linkque"

local sche = {

ready_list = LinkQue.New(),

timer = MinHeap.New(),

allcos = {},

co_count = 0,

runningco = nil,

}

local stat_ready = 1

local stat_sleep = 2

local stat_yield = 3

local stat_dead = 4

local stat_block = 5

local stat_running = 6

local function add2Ready(co)

local status = co.status

if status == stat_ready or status == stat_dead or status == stat_running then

return

end

co.status = stat_ready

sche.timer:Remove(co)

sche.ready_list:Push(co)

end

local function _block(ms,stat)

local co = sche.runningco

if co.status ~= stat_running then

return

end

if ms and ms > 0 then

co.timeout = C.GetSysTick() + ms

if co.index == 0 then

sche.timer:Insert(co)

else

sche.timer:Change(co)

end

end

co.status = stat

coroutine.yield(co.coroutine)

if co.index ~= 0 then

co.timeout = 0

sche.timer:Change(co)

sche.timer:PopMin()

end

end

local function Sleep(ms)

local stat

if ms and ms > 0 then

stat = stat_sleep

else

stat = stat_yield

end

_block(ms,stat)

end

local function Yield()

Sleep(0)

end

local function Block(ms)

_block(ms,stat_block)

end

local function WakeUp(co)

add2Ready(co)

end

local function SwitchTo(co)

local pre_co = sche.runningco

sche.runningco = co

co.status = stat_running

coroutine.resume(co.coroutine,co)

sche.runningco = pre_co

return co.status

end

local function Schedule(co)

local readylist = sche.ready_list

if co then

local status = co.status

if status == stat_ready or status == stat_dead or status == stat_running then

return sche.ready_list:Len()

end

sche.timer:Remove(co)

if SwitchTo(co) == stat_yield then

add2Ready(co)

end

if sche.stop then

return -1

end

else

local yields = {}

co = readylist:Pop()

while co do

if SwitchTo(co) == stat_yield then

table.insert(yields,co)

end

if sche.stop then

return -1

end

co = readylist:Pop()

end

local now = C.GetSysTick()

local timer = sche.timer

while timer:Min() ~=0 and timer:Min() <= now do

co = timer:PopMin()

--if co.status == stat_block or co.status == stat_sleep then

add2Ready(co)

--end

end

for k,v in pairs(yields) do

add2Ready(v)

end

end

return sche.ready_list:Len()

end

local function Running()

return sche.runningco

end

local function GetCoByIdentity(identity)

return sche.allcos[identity]

end

local function start_fun(co)

local ret,err = pcall(co.start_func,table.unpack(co.args))

if not ret then

CLog.SysLog(CLog.LOG_ERROR,string.format("error on start_fun::%s",err))

end

sche.allcos[co.identity] = nil

sche.co_count = sche.co_count - 1

co.status = stat_dead

end

local g_counter = 0

local function gen_identity()

g_counter = bit32.band(g_counter + 1,0x000FFFFF)

return string.format("%d-%d",C.GetSysTick(),g_counter)

end

--产生一个coroutine在下次调用Schedule时执行

local function Spawn(func,...)

local co = {index=0,timeout=0,identity=gen_identity(),start_func = func,args={...}}

co.coroutine = coroutine.create(start_fun)

sche.allcos[co.identity] = co

sche.co_count = sche.co_count + 1

add2Ready(co)

return co

end

--产生一个coroutine立刻执行

local function SpawnAndRun(func,...)

local co = {index=0,timeout=0,identity=gen_identity(),start_func = func,args={...}}

co.coroutine = coroutine.create(start_fun)

sche.allcos[co.identity] = co

sche.co_count = sche.co_count + 1

Schedule(co)

return co

end

local function Exit()

sche.stop = true

Yield() --yield to scheduler

end

return {

Spawn = Spawn,

SpawnAndRun = SpawnAndRun,

Yield = Yield,

Sleep = Sleep,

Block = Block,

WakeUp = WakeUp,

Running = Running,

GetCoByIdentity = GetCoByIdentity,

Schedule = Schedule,

GetCoCount = function () return sche.co_count end,

Exit = Exit,

}