Golang Site.RemoveContactFromSite方法代码示例

func removeContactFromSite(controller *contactsController, contactID int64, siteID int64) error {
	var site database.Site
	err := site.GetSite(controller.DB, siteID)
	if err != nil {
		return err
	}
	err = site.RemoveContactFromSite(controller.DB, contactID)
	if err != nil {
		return err
	}
	return nil
}

//.........这里部分代码省略.........
	}
	//Verify the saved site is same as the input.
	if !database.CompareSites(siteUpdated, sUpdate) {
		t.Error("Updated site saved not equal to input:\n", siteUpdated, sUpdate)
	}

	// Create first contact - ContactID is for referencing the contact get test
	c := database.Contact{Name: "Joe Contact", EmailAddress: "[email protected]", SmsNumber: "5125551212",
		SmsActive: false, EmailActive: false, ContactID: 1}
	err = c.CreateContact(db)
	if err != nil {
		t.Fatal("Failed to create new contact:", err)
	}
	// Associate to the site ID
	err = site.AddContactToSite(db, c.ContactID)
	if err != nil {
		t.Fatal("Failed to associate contact with site:", err)
	}

	// Create second contact
	c2 := database.Contact{Name: "Jill Contact", EmailAddress: "[email protected]", SmsNumber: "5125551213",
		SmsActive: false, EmailActive: false}
	err = c2.CreateContact(db)
	if err != nil {
		t.Fatal("Failed to create new site:", err)
	}
	// Associate the contact to the site
	err = site.AddContactToSite(db, c2.ContactID)
	if err != nil {
		t.Error("Failed to associate contact2 with site:", err)
	}

	//Get the saved site contacts
	err = site.GetSiteContacts(db, site.SiteID)
	if err != nil {
		t.Error("Failed to retrieve site contacts:", err)
	}

	// Verify the first contact was Loaded as the last in list by sort order
	if !reflect.DeepEqual(c, site.Contacts[1]) {
		t.Error("New contact saved not equal to input:\n", site.Contacts[1], c)
	}
	// Verify the second contact was Loaded as the first in list by sort order
	if !reflect.DeepEqual(c2, site.Contacts[0]) {
		t.Error("New contact saved not equal to input:\n", site.Contacts[0], c2)
	}

	// Remove second contact from site.
	err = site.RemoveContactFromSite(db, c2.ContactID)
	if err != nil {
		t.Fatal("Failed to remove contact2 from site:", err)
	}

	//Get the saved site contacts again
	err = site.GetSiteContacts(db, site.SiteID)
	if err != nil {
		t.Fatal("Failed to retrieve site contacts:", err)
	}

	if len(site.Contacts) != 1 {
		t.Fatal("Site should have only one contact after removal")
	}

	// Get the first contact via the GetContact method
	c1Get := database.Contact{}
	err = c1Get.GetContact(db, c.ContactID)
	if err != nil {
		t.Error("Failed to retrieve the first contact.")
	}

	// Verify the first contact was retrieved OK
	if !reflect.DeepEqual(c, c1Get) {
		t.Error("Retrieved contact saved not equal to input:\n", c1Get, c)
	}

	// Update the first contact.
	c1Update := database.Contact{Name: "Jane Contact", EmailAddress: "[email protected]", SmsNumber: "5125551313",
		SmsActive: true, EmailActive: true, ContactID: 1}
	c1Get.Name = c1Update.Name
	c1Get.EmailAddress = c1Update.EmailAddress
	c1Get.SmsNumber = c1Update.SmsNumber
	c1Get.EmailActive = c1Update.EmailActive
	c1Get.SmsActive = c1Update.SmsActive
	err = c1Get.UpdateContact(db)
	if err != nil {
		t.Error("Failed to update the first contact.")
	}

	// Get the first contact again after update
	c1Get2 := database.Contact{}
	err = c1Get2.GetContact(db, c1Update.ContactID)
	if err != nil {
		t.Error("Failed to retrieve the first contact.")
	}

	// Verify the first contact was retrieved OK
	if !reflect.DeepEqual(c1Update, c1Get2) {
		t.Error("Retrieved updated contact saved not equal to input:\n", c1Get2, c1Update)
	}
}

func (controller *sitesController) editPost(rw http.ResponseWriter, req *http.Request) (int, error) {
	err := req.ParseForm()
	if err != nil {
		return http.StatusInternalServerError, err
	}

	decoder := schema.NewDecoder()
	// Ignore unknown keys to prevent errors from the CSRF token.
	decoder.IgnoreUnknownKeys(true)
	formSite := new(viewmodels.SitesEditViewModel)
	err = decoder.Decode(formSite, req.PostForm)
	if err != nil {
		return http.StatusInternalServerError, err
	}

	valErrors := validateSiteForm(formSite)
	if len(valErrors) > 0 {
		isAuthenticated, user := getCurrentUser(rw, req, controller.authorizer)
		var contacts database.Contacts
		err = contacts.GetContacts(controller.DB)
		if err != nil {
			return http.StatusInternalServerError, err
		}
		vm := viewmodels.EditSiteViewModel(formSite, contacts, isAuthenticated, user, valErrors)
		vm.CsrfField = csrf.TemplateField(req)
		return http.StatusOK, controller.editTemplate.Execute(rw, vm)
	}

	// Get the site to edit
	site := new(database.Site)
	err = site.GetSite(controller.DB, formSite.SiteID)
	if err != nil {
		return http.StatusInternalServerError, err
	}

	err = viewmodels.MapSiteVMtoDB(formSite, site)
	if err != nil {
		return http.StatusInternalServerError, err
	}
	err = site.UpdateSite(controller.DB)
	if err != nil {
		return http.StatusInternalServerError, err
	}

	//Loop selected ones first and if it's not already in the site then add it.
	for _, contactSelID := range formSite.SelectedContacts {
		if !int64InSlice(int64(contactSelID), formSite.SiteContacts) {
			err = site.AddContactToSite(controller.DB, contactSelID)
			if err != nil {
				return http.StatusInternalServerError, err
			}
		}
	}

	// Loop existing site contacts and if it's not in the selected items then remove it.
	for _, contactSiteID := range formSite.SiteContacts {
		if !int64InSlice(int64(contactSiteID), formSite.SelectedContacts) {
			err = site.RemoveContactFromSite(controller.DB, contactSiteID)
			if err != nil {
				return http.StatusInternalServerError, err
			}
		}
	}

	// Refresh the pinger with the changes.
	err = controller.pinger.UpdateSiteSettings()
	if err != nil {
		return http.StatusInternalServerError, err
	}

	http.Redirect(rw, req, "/settings", http.StatusSeeOther)
	return http.StatusSeeOther, nil
}