func (i *Impl) CreateUser(w rest.ResponseWriter, r *rest.Request) {
var user UserPayload
err := r.DecodeJsonPayload(&user)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
// returns -1 if username is unable to be registered, ie, already taken
uid, err := i.DB.CreateUser(user.Username, user.Name)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
if uid == -1 {
w.WriteJson(map[string]string{
"uid": strconv.Itoa(uid),
"user": "******",
})
return
}
userOut, err := i.DB.GetUser(uid)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.WriteJson(&userOut)
}
func (i *Impl) UnfollowUser(w rest.ResponseWriter, r *rest.Request) {
v, err := url.ParseQuery(r.URL.RawQuery)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
uid, err := strconv.Atoi(v.Get("uid"))
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
otherId, err := strconv.Atoi(v.Get("otherId"))
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
res, err := i.DB.Unfollow(uid, otherId)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
if res == true {
w.WriteJson(map[string]string{"following": v.Get("otherId"),
"follower": v.Get("uid"), "unfollowed": "true"})
} else {
w.WriteJson(map[string]string{"following": v.Get("otherId"),
"follower": v.Get("uid"), "unfollowed": "false"})
}
}
func (i *Impl) GetUser(w rest.ResponseWriter, r *rest.Request) {
v, err := url.ParseQuery(r.URL.RawQuery)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
uid, err := strconv.Atoi(v.Get("uid"))
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
usr, err := i.DB.GetUser(uid)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.WriteJson(&usr)
}
/*
* consumes JSON of the form
{
"uid": <user id>
"msg": <text string containing message>
}
*/
func (i *Impl) PostStatus(w rest.ResponseWriter, r *rest.Request) {
var status StatusPayload
if err := r.DecodeJsonPayload(&status); err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
sid, err := i.DB.PostStatus(status.Uid, status.Msg)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
post, err := i.DB.GetStatus(sid)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.WriteJson(&post)
}
func (i *Impl) GetTimeline(w rest.ResponseWriter, r *rest.Request) {
v, err := url.ParseQuery(r.URL.RawQuery)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
uid, err := strconv.Atoi(v.Get("uid"))
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
page, err := strconv.Atoi(v.Get("page"))
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
res, err := i.DB.GetUserTimeline(uid, page, 30)
if err != nil {
rest.Error(w, err.Error(), http.StatusInternalServerError)
return
}
output := new(TimelineResponse)
output.Posts = make([]rdb.Status, len(res))
output.Uid = uid
output.Page = page
outputIndex := 0
// channel to send/recieve status structs
statuses := make(chan rdb.Status)
defer close(statuses)
var pst int
for j := 0; j < len(res); j++ {
pst = res[j]
// anon goroutine to get a status in timeline page
// this means that all statuses are fetched concurrently
go func(post int) {
status, err := i.DB.GetStatus(post)
if err != nil {
log.Printf("error getting post %d, %v\n", post, err)
return
}
// pipe the fetched status into the channel previously made
statuses <- status
}(pst)
}
// this code is blocking
Loop:
for outputIndex < len(res) {
select {
case sts := <-statuses:
output.Posts[outputIndex] = sts
outputIndex++
case <-time.After(time.Second * 1):
log.Printf("timeout getting timeline:%d page:%d\n", uid,
page)
break Loop
}
}
// because the statuses were retrieved concurrently we can't be sure
// of what order they will appear in output.Posts, so we must sort them
// if we want them to appear from newest to oldest
sort.Sort(output.Posts)
w.WriteJson(&output)
}