it-ebooks - Go Mega 教程

... // Follow func // follow someone usr_id other.id follow_id u.id func ( u * User ) Follow ( username string ) error { other , err := GetUserByUsername (username) if err != nil { return err} return db. Model (other). Association ( " Followers " ). Append (u). Error } // Unfollow func func ( u * User ) Unfollow ( username string ) error { other , err := GetUserByUsername (username) if err != nil { return err} return db. Model (other). Association ( " Followers " ). Delete (u). Error } // FollowSelf func func ( u * User ) FollowSelf () error { return db. Model (u). Association ( " Followers " ). Append (u). Error } // FollowersCount func func ( u * User ) FollowersCount () int { return db. Model (u). Association ( " Followers " ). Count ()} // FollowingIDs func func ( u * User ) FollowingIDs () [] int { var ids [] int rows , err := db. Table ( " follower " ). Where ( " follower_id = ? " , u. ID ). Select ( " user_id, follower_id " ). Rows () if err != nil {log. Println ( " Counting Following error: " , err) return ids} defer rows. Close () for rows. Next () { var id , followerID int rows. Scan (&id, &followerID)ids = append (ids, id)} return ids} // FollowingCount func func ( u * User ) FollowingCount () int { ids := u. FollowingIDs () return len (ids)} // FollowingPosts func func ( u * User ) FollowingPosts () (*[] Post , error ) { var posts []Post ids := u. FollowingIDs () if err := db. Preload ( " User " ). Order ( " timestamp desc " ). Where ( " user_id in (?) " , ids). Find (&posts). Error ; err != nil { return nil , err} return &posts, nil } // IsFollowedByUser func func ( u * User ) IsFollowedByUser ( username string ) bool { user , _ := GetUserByUsername (username) ids := user. FollowingIDs () for _ , id := range ids { if u. ID == id { return true }} return false } // CreatePost func func ( u * User ) CreatePost ( body string ) error { post := Post{Body: body, UserID: u. ID } return db. Create (&post). Error }...

// AddUser func func AddUser ( username , password , email string ) error { user := User{Username: username, Email: email}user. SetPassword (password)user. SetAvatar (email) if err := db. Create (&user). Error ; err != nil { return err} return user. FollowSelf ()}

package main import ( " log " " github.com/bonfy/go-mega-code/model " _ " github.com/jinzhu/gorm/dialects/mysql " ) func main () {log. Println ( " DB Init ... " ) db := model. ConnectToDB () defer db. Close ()model. SetDB (db)db. DropTableIfExists (model. User {}, model. Post {}, " follower " )db. CreateTable (model. User {}, model. Post {})model. AddUser ( " bonfy " , " abc123 " , " i@bonfy.im " )model. AddUser ( " rene " , " abc123 " , " rene@test.com " ) u1 , _ := model. GetUserByUsername ( " bonfy " )u1. CreatePost ( " Beautiful day in Portland! " )model. UpdateAboutMe (u1. Username , ` I'm the author of Go-Mega Tutorial you are reading now! ` ) u2 , _ := model. GetUserByUsername ( " rene " )u2. CreatePost ( " The Avengers movie was so cool! " )u2. CreatePost ( " Sun shine is beautiful " )u1. Follow (u2. Username )}

$ go run cmd/db_init/main.go

... func ( h home ) registerRoutes () { r := mux. NewRouter ()r. HandleFunc ( " /logout " , middleAuth (logoutHandler))r. HandleFunc ( " /login " , loginHandler)r. HandleFunc ( " /register " , registerHandler)r. HandleFunc ( " /user/{username} " , middleAuth (profileHandler))r. HandleFunc ( " /follow/{username} " , middleAuth (followHandler))r. HandleFunc ( " /unfollow/{username} " , middleAuth (unFollowHandler))r. HandleFunc ( " /profile_edit " , middleAuth (profileEditHandler))r. HandleFunc ( " / " , middleAuth (indexHandler))http. Handle ( " / " , r)}... func followHandler ( w http . ResponseWriter , r * http . Request ) { vars := mux. Vars (r) pUser := vars[ " username " ] sUser , _ := getSessionUser (r) err := vm. Follow (sUser, pUser) if err != nil {log. Println ( " Follow error: " , err)w. Write ([] byte ( " Error in Follow " )) return }http. Redirect (w, r, fmt. Sprintf ( " /user/ %s " , pUser), http. StatusSeeOther )} func unFollowHandler ( w http . ResponseWriter , r * http . Request ) { vars := mux. Vars (r) pUser := vars[ " username " ] sUser , _ := getSessionUser (r) err := vm. UnFollow (sUser, pUser) if err != nil {log. Println ( " UnFollow error: " , err)w. Write ([] byte ( " Error in UnFollow " )) return }http. Redirect (w, r, fmt. Sprintf ( " /user/ %s " , pUser), http. StatusSeeOther )}

package vm import " github.com/bonfy/go-mega-code/model " // ProfileViewModel struct type ProfileViewModel struct { BaseViewModel Posts []model. Post Editable bool IsFollow bool FollowersCount int FollowingCount int ProfileUser model. User } // ProfileViewModelOp struct type ProfileViewModelOp struct {} // GetVM func func ( ProfileViewModelOp ) GetVM ( sUser , pUser string ) ( ProfileViewModel , error ) { v := ProfileViewModel{}v. SetTitle ( " Profile " ) u , err := model. GetUserByUsername (pUser) if err != nil { return v, err} posts , _ := model. GetPostsByUserID (u. ID )v. ProfileUser = *uv. Editable = (sUser == pUser) if !v. Editable {v. IsFollow = u. IsFollowedByUser (sUser)}v. FollowersCount = u. FollowersCount ()v. FollowingCount = u. FollowingCount ()v. Posts = *postsv. SetCurrentUser (sUser) return v, nil } // Follow func : A follow B func Follow ( a , b string ) error { u , err := model. GetUserByUsername (a) if err != nil { return err} return u. Follow (b)} // UnFollow func : A unfollow B func UnFollow ( a , b string ) error { u , err := model. GetUserByUsername (a) if err != nil { return err} return u. Unfollow (b)}