a65e546..master · sc-build

src/Control/Task/Scheduler/Query.hs+3

since splitsnapshots

unifiedsplit

@@ -41,6 +41,7 @@ scheduleTaskQuery task = do

41 :* Set (param @2) `as` #payload

42 :* Default `as` #creation_time

43 :* Default `as` #started

44+ :* Default `as` #failures

44 )

45 )

45 ( OnConflict

46 ( OnConflict

46 (OnConstraint #pk_task_payload)

47 (OnConstraint #pk_task_payload)

@@ -87,6 +88,7 @@ rescheduleTaskQuery payload = do

87 #tasks

88 #tasks

88 ( Default `as` #creation_time

89 ( Default `as` #creation_time

89 :* Default `as` #started

90 :* Default `as` #started

91+ :* Set (#failures + 1) `as` #failures

90 )

92 )

91 (param @1 .== #task .&& param @2 .== #payload)

93 (param @1 .== #task .&& param @2 .== #payload)

92 dbWrite (const $ return ()) $

94 dbWrite (const $ return ()) $

@@ -112,6 +114,7 @@ pickTasksQuery tasks limitCount = do

112 ( from (table #tasks)

114 ( from (table #tasks)

113 & where_ (#task `in_` (literal <$> tasks))

115 & where_ (#task `in_` (literal <$> tasks))

114 & where_ (not_ . notNull $ #started)

116 & where_ (not_ . notNull $ #started)

117+ & where_ (#failures .< 20)

115 & orderBy [#creation_time & Asc]

118 & orderBy [#creation_time & Asc]

116 & limit limitCount

119 & limit limitCount

117 & orderBy [#creation_time & Desc]

120 & orderBy [#creation_time & Desc]

src/Server/Schema.hs+3−1

@@ -6,13 +6,15 @@ module Server.Schema

6where

7

8import Server.Schema.V1

9+import Server.Schema.V2

9import Squeal.PostgreSQL

10import Squeal.PostgreSQL

10

11

11type StaticPQ m = MonadPQ Schema m

12type StaticPQ m = MonadPQ Schema m

12

13

13−type Schema = SchemaV1

14+type Schema = SchemaV2

14

15

15migration :: AlignedList (Migration (Terminally PQ IO)) (Public '[]) Schema

16migration :: AlignedList (Migration (Terminally PQ IO)) (Public '[]) Schema

16migration =

17migration =

17 pureMigration schemaMigrationV1

18 pureMigration schemaMigrationV1

19+ :>> pureMigration schemaMigrationV2

18 :>> Done

20 :>> Done

src/Server/Schema/V2.hsnew file+46

@@ -0,0 +1,46 @@

1+module Server.Schema.V2

2+ ( SchemaV2,

3+ schemaMigrationV2,

4+ )

5+where

6+

7+import Data.Aeson

8+import Data.Int

9+import Data.Submission

10+import Data.Text (Text)

11+import Data.Time

12+import Server.Schema.V1

13+import Squeal.PostgreSQL

14+

15+type SchemaV2 =

16+ Public

17+ '[ "tasks"

18+ ::: 'Table

19+ ( '[ "pk_task_payload" :=> 'PrimaryKey '["task", "payload"]

20+ ]

21+ :=> '[ "task" ::: 'NoDef :=> 'NotNull (PG Text),

22+ "payload" ::: 'NoDef :=> 'NotNull (PG (Jsonb Value)),

23+ "creation_time" ::: 'Def :=> 'NotNull (PG UTCTime),

24+ "started" ::: 'Def :=> 'NotNull (PG Bool),

25+ "failures" ::: 'Def :=> 'NotNull (PG Int32)

26+ ]

27+ ),

28+ "submissions"

29+ ::: 'Table

30+ ( '["pk_submission_repo_full_name_sha" :=> 'PrimaryKey '["repo_full_name", "sha"]]

31+ :=> '[ "user_name" ::: 'NoDef :=> 'NotNull (PG String),

32+ "repo_full_name" ::: 'NoDef :=> 'NotNull (PG String),

33+ "problem" ::: 'NoDef :=> 'NotNull (PG String),

34+ "sha" ::: 'NoDef :=> 'NotNull (PG String),

35+ "status" ::: 'NoDef :=> 'NotNull (PG (Jsonb SubmissionStatus))

36+ ]

37+ )

38+ ]

39+

40+schemaMigrationV2 :: Migration Definition SchemaV1 SchemaV2

41+schemaMigrationV2 =

42+ Migration

43+ { name = "v2",

44+ up = alterTable #tasks (addColumn #failures (default_ 0 $ notNullable int)),

45+ down = alterTable #tasks (dropColumn #failures)

46+ }