Exemplo n.º 1
0
func dataCompletenessSummaryProto(r *http.Request, h http.Header, b *bytes.Buffer) *weft.Result {
	typeID := r.URL.Query().Get("typeID")

	var err error
	var rows *sql.Rows
	var expected int

	switch typeID {
	case "":
		rows, err = dbR.Query(`SELECT siteID, typeID, count, expected
		FROM data.completeness_summary
		JOIN data.site USING (sitePK)
		JOIN data.completeness_type USING (typePK)`)
	default:
		var typePK int
		if err = dbR.QueryRow(`SELECT typePK FROM data.completeness_type WHERE typeID = $1`,
			typeID).Scan(&typePK); err != nil {
			if err == sql.ErrNoRows {
				return &weft.NotFound
			}
			return weft.InternalServerError(err)
		}

		rows, err = dbR.Query(`SELECT siteID, typeID, count, expected
		FROM data.completeness_summary
		JOIN data.site USING (sitePK)
		JOIN data.completeness_type USING (typePK)
		WHERE typeID = $1;`, typeID)
	}

	if err != nil {
		return weft.InternalServerError(err)
	}

	defer rows.Close()

	var t time.Time
	var dcr mtrpb.DataCompletenessSummaryResult

	for rows.Next() {
		var count int
		var siteID string

		if err = rows.Scan(&siteID, &typeID, &count, &expected); err != nil {
			return weft.InternalServerError(err)
		}

		c := float32(count) / (float32(expected) / 288)
		dc := mtrpb.DataCompletenessSummary{TypeID: typeID, SiteID: siteID, Completeness: c, Seconds: t.Unix()}
		dcr.Result = append(dcr.Result, &dc)
	}

	var by []byte

	if by, err = proto.Marshal(&dcr); err != nil {
		return weft.InternalServerError(err)
	}

	b.Write(by)

	return &weft.StatusOK
}