@@ -88,6 +88,7 @@ func New() database.Store {
88
88
customRoles : make ([]database.CustomRole , 0 ),
89
89
locks : map [int64 ]struct {}{},
90
90
runtimeConfig : map [string ]string {},
91
+ userStatusChanges : make ([]database.UserStatusChange , 0 ),
91
92
},
92
93
}
93
94
// Always start with a default org. Matching migration 198.
@@ -256,6 +257,7 @@ type data struct {
256
257
lastLicenseID int32
257
258
defaultProxyDisplayName string
258
259
defaultProxyIconURL string
260
+ userStatusChanges []database.UserStatusChange
259
261
}
260
262
261
263
func tryPercentile (fs []float64 , p float64 ) float64 {
@@ -5673,7 +5675,21 @@ func (q *FakeQuerier) GetUserStatusCountsByDay(_ context.Context, arg database.G
5673
5675
return nil , err
5674
5676
}
5675
5677
5676
- panic ("not implemented" )
5678
+ result := make ([]database.GetUserStatusCountsByDayRow , 0 )
5679
+ for _ , change := range q .userStatusChanges {
5680
+ if change .ChangedAt .Before (arg .StartTime ) || change .ChangedAt .After (arg .EndTime ) {
5681
+ continue
5682
+ }
5683
+ result = append (result , database.GetUserStatusCountsByDayRow {
5684
+ Status : database.NullUserStatus {
5685
+ UserStatus : change .NewStatus ,
5686
+ Valid : true ,
5687
+ },
5688
+ Count : 1 ,
5689
+ })
5690
+ }
5691
+
5692
+ return result , nil
5677
5693
}
5678
5694
5679
5695
func (q * FakeQuerier ) GetUserWorkspaceBuildParameters (_ context.Context , params database.GetUserWorkspaceBuildParametersParams ) ([]database.GetUserWorkspaceBuildParametersRow , error ) {
@@ -8028,6 +8044,12 @@ func (q *FakeQuerier) InsertUser(_ context.Context, arg database.InsertUserParam
8028
8044
sort .Slice (q .users , func (i , j int ) bool {
8029
8045
return q .users [i ].CreatedAt .Before (q .users [j ].CreatedAt )
8030
8046
})
8047
+
8048
+ q .userStatusChanges = append (q .userStatusChanges , database.UserStatusChange {
8049
+ UserID : user .ID ,
8050
+ NewStatus : user .Status ,
8051
+ ChangedAt : user .UpdatedAt ,
8052
+ })
8031
8053
return user , nil
8032
8054
}
8033
8055
@@ -9064,12 +9086,18 @@ func (q *FakeQuerier) UpdateInactiveUsersToDormant(_ context.Context, params dat
9064
9086
Username : user .Username ,
9065
9087
LastSeenAt : user .LastSeenAt ,
9066
9088
})
9089
+ q .userStatusChanges = append (q .userStatusChanges , database.UserStatusChange {
9090
+ UserID : user .ID ,
9091
+ NewStatus : database .UserStatusDormant ,
9092
+ ChangedAt : params .UpdatedAt ,
9093
+ })
9067
9094
}
9068
9095
}
9069
9096
9070
9097
if len (updated ) == 0 {
9071
9098
return nil , sql .ErrNoRows
9072
9099
}
9100
+
9073
9101
return updated , nil
9074
9102
}
9075
9103
@@ -9870,6 +9898,12 @@ func (q *FakeQuerier) UpdateUserStatus(_ context.Context, arg database.UpdateUse
9870
9898
user .Status = arg .Status
9871
9899
user .UpdatedAt = arg .UpdatedAt
9872
9900
q .users [index ] = user
9901
+
9902
+ q .userStatusChanges = append (q .userStatusChanges , database.UserStatusChange {
9903
+ UserID : user .ID ,
9904
+ NewStatus : user .Status ,
9905
+ ChangedAt : user .UpdatedAt ,
9906
+ })
9873
9907
return user , nil
9874
9908
}
9875
9909
return database.User {}, sql .ErrNoRows
0 commit comments