added rpc message to change step tracker

Author: gianmarcomennecozzi

database/query.go

@@ -28,8 +28,8 @@ var (
 		"step integer, " +
 		"solved_challenges text);"
 
-	AddTeamQuery = "INSERT INTO team (tag, event_id, email, name, password, created_at, last_access, solved_challenges)" +
-		"VALUES ($1, $2, $3, $4, $5, $6, $7, $8)"
+	AddTeamQuery = "INSERT INTO team (tag, event_id, email, name, password, created_at, last_access, step, solved_challenges)" +
+		"VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)"
 
 	AddEventQuery = "INSERT INTO event (tag, name, available, capacity, frontends, status, exercises, started_at, finish_expected, finished_at, createdby, onlyvpn)" +
 		"VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10,$11,$12)"
@@ -39,6 +39,7 @@ var (
 
 	UpdateEventLastaccessedDate = "UPDATE team SET last_access = $2 WHERE tag = $1"
 	UpdateTeamSolvedChl         = "UPDATE team SET solved_challenges = $2 WHERE tag = $1"
+	UpdateTeamStep              = "UPDATE team SET step = $2 WHERE tag = $1"
 
 	QuerySolvedChls = "SELECT solved_challenges FROM team WHERE tag=$1"
 	QueryEventTable = "SELECT * FROM event"

database/store.go

@@ -46,6 +46,7 @@ type Store interface {
 	GetEventStatus(*pb.GetEventStatusRequest) (int32, error)
 	SetEventStatus(*pb.SetEventStatusRequest) (int32, error)
 	UpdateTeamSolvedChallenge(*pb.UpdateTeamSolvedChallengeRequest) (string, error)
+	UpdateTeamStep(request *pb.UpdateTeamStepTrackerRequest) (string, error)
 	UpdateTeamLastAccess(*pb.UpdateTeamLastAccessRequest) (string, error)
 	UpdateCloseEvent(*pb.UpdateEventRequest) (string, error)
 }
@@ -243,6 +244,18 @@ func (s *store) UpdateTeamSolvedChallenge(in *pb.UpdateTeamSolvedChallengeReques
 	return OK, nil
 }
 
+func (s *store) UpdateTeamStep(in *pb.UpdateTeamStepTrackerRequest) (string, error) {
+	s.m.Lock()
+	defer s.m.Unlock()
+
+	_, err := s.db.Exec(UpdateEventLastaccessedDate, in.TeamId, in.Step)
+	if err != nil {
+		return "", err
+	}
+
+	return OK, nil
+}
+
 func (s *store) UpdateTeamLastAccess(in *pb.UpdateTeamLastAccessRequest) (string, error) {
 	s.m.Lock()
 	defer s.m.Unlock()

database/store_test.go

@@ -301,6 +301,24 @@ func TestTeamUpdateLastAccess(t *testing.T) {
 	}
 }
 
+func TestTeamUpdateStep(t *testing.T) {
+
+	conn, err := createTestClientConn()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer conn.Close()
+	c := pb.NewStoreClient(conn)
+
+	_, err = c.UpdateTeamStepTracker(context.Background(), &pb.UpdateTeamStepTrackerRequest{
+		TeamId: "team1",
+		Step:   1,
+	})
+	if err != nil {
+		t.Fatalf("Error updating team step: %s", err.Error())
+	}
+}
+
 func TestCloseEvent(t *testing.T) {
 
 	conn, err := createTestClientConn()