changes for cloud

db/db.go

@@ -5260,6 +5260,31 @@ func GetStationIdsFromDlsMsgs(tenantName string) ([]int, error) {
 	return stationIds, nil
 }
 
+func DeleteDlsMsgsByTenant(tenantName string) error {
+	ctx, cancelfunc := context.WithTimeout(context.Background(), DbOperationTimeout*time.Second)
+	defer cancelfunc()
+
+	conn, err := MetadataDbClient.Client.Acquire(ctx)
+	if err != nil {
+		return err
+	}
+	defer conn.Release()
+
+	removeUserQuery := `DELETE FROM dls_messages WHERE tenant_name = $1`
+
+	stmt, err := conn.Conn().Prepare(ctx, "remove_dls_msgs_by_tenant", removeUserQuery)
+	if err != nil {
+		return err
+	}
+	_, err = conn.Conn().Exec(ctx, stmt.Name, tenantName)
+	if err != nil {
+		return err
+	}
+
+	return nil
+
+}
+
 // Tenants functions
 func UpsertTenant(name string) (models.Tenant, error) {
 	ctx, cancelfunc := context.WithTimeout(context.Background(), DbOperationTimeout*time.Second)

server/background_tasks.go

@@ -266,7 +266,7 @@ func (s *Server) StartBackgroundTasks() error {
 		return err
 	}
 	go s.UploadTenantUsageToDB()
-	go s.ConnectToFirebaseFunction()
+	go s.RefreshFirebaseFunctionsKey()
 	return nil
 }
 

server/memphis_cloud.go

@@ -1418,13 +1418,6 @@ func (umh UserMgmtHandler) RemoveMyUser(c *gin.Context) {
 		return
 	}
 
-	err = db.DeleteUser(user.Username, user.TenantName)
-	if err != nil {
-		serv.Errorf("RemoveMyUser: User " + user.Username + ": " + err.Error())
-		c.AbortWithStatusJSON(500, gin.H{"message": "Server error"})
-		return
-	}
-
 	shouldSendAnalytics, _ := shouldSendAnalytics()
 	if shouldSendAnalytics {
 		analytics.SendEvent(user.Username, "user-remove-himself")
@@ -1434,5 +1427,5 @@ func (umh UserMgmtHandler) RemoveMyUser(c *gin.Context) {
 	c.IndentedJSON(200, gin.H{})
 }
 
-func (s *Server) ConnectToFirebaseFunction() {
+func (s *Server) RefreshFirebaseFunctionsKey() {
 }

server/memphis_handlers_user_mgmt.go

@@ -182,6 +182,21 @@ func removeTenantResources(tenantName string) error {
 		return err
 	}
 
+	err = db.DeleteUsersByTenant(tenantName)
+	if err != nil {
+		return err
+	}
+
+	err = db.DeleteDlsMsgsByTenant(tenantName)
+	if err != nil {
+		return err
+	}
+
+	err = db.RemoveTenant(tenantName)
+	if err != nil {
+		return err
+	}
+
 	for _, station := range stations {
 		stationName := strings.ToLower(station.Name)
 		sName, err := StationNameFromStr(stationName)