update 04.03.24
This commit is contained in:
parent
fa863cbd12
commit
eab7fea353
5 changed files with 355 additions and 334 deletions
134
router/debt.go
Normal file
134
router/debt.go
Normal file
|
|
@ -0,0 +1,134 @@
|
|||
package router
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"gibb165lb2/util"
|
||||
"github.com/gorilla/mux"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
func GetDebts(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
personID := vars["id"]
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
|
||||
debts, err := util.GetDebtsForPerson(util.Client, personID)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(debts)
|
||||
}
|
||||
func CreateDebt(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var debt util.Debt
|
||||
personID := vars["id"]
|
||||
json.NewDecoder(r.Body).Decode(&debt)
|
||||
debt.PersonID, _ = primitive.ObjectIDFromHex(personID)
|
||||
collection := util.Client.Database("debtlist").Collection("debts")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
result, err := collection.InsertOne(ctx, debt)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(result)
|
||||
}
|
||||
func DeleteDebt(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var debt util.Debt
|
||||
debtId := vars["id"]
|
||||
json.NewDecoder(r.Body).Decode(&debt)
|
||||
debt.ID, _ = primitive.ObjectIDFromHex(debtId)
|
||||
collection := util.Client.Database("debtlist").Collection("debts")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
filter := bson.M{"_id": debt.ID}
|
||||
result, err := collection.DeleteOne(ctx, filter)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(result)
|
||||
}
|
||||
func GetAllDebts(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var debts []util.Debt
|
||||
collection := util.Client.Database("debtlist").Collection("debts")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
cursor, err := collection.Find(ctx, bson.M{})
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
defer cursor.Close(ctx)
|
||||
for cursor.Next(ctx) {
|
||||
var debt util.Debt
|
||||
cursor.Decode(&debt)
|
||||
debts = append(debts, debt)
|
||||
}
|
||||
if err := cursor.Err(); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(debts)
|
||||
}
|
||||
|
||||
func GetDebt(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
debtID := vars["id"]
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var debt util.Debt
|
||||
collection := util.Client.Database("debtlist").Collection("debts")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
did, err := primitive.ObjectIDFromHex(debtID)
|
||||
filter := bson.M{"_id": did}
|
||||
err = collection.FindOne(ctx, filter).Decode(&debt)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(debt)
|
||||
}
|
||||
func UpdateDebt(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var debt util.Debt
|
||||
vars := mux.Vars(r)
|
||||
debtID := vars["id"]
|
||||
debt.ID, _ = primitive.ObjectIDFromHex(debtID)
|
||||
json.NewDecoder(r.Body).Decode(&debt)
|
||||
collection := util.Client.Database("debtlist").Collection("debts")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
filter := bson.M{"_id": debt.ID}
|
||||
update := bson.M{
|
||||
"$set": bson.M{
|
||||
"fk_pid": debt.PersonID,
|
||||
"amount": debt.Amount,
|
||||
"description": debt.Description,
|
||||
"datetime": debt.Datetime,
|
||||
},
|
||||
}
|
||||
result, err := collection.UpdateOne(ctx, filter, update)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(result)
|
||||
}
|
||||
154
router/person.go
Normal file
154
router/person.go
Normal file
|
|
@ -0,0 +1,154 @@
|
|||
package router
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"gibb165lb2/util"
|
||||
"github.com/gorilla/mux"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
func CreatePerson(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var person util.Person
|
||||
json.NewDecoder(r.Body).Decode(&person)
|
||||
collection := util.Client.Database("debtlist").Collection("persons")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
result, err := collection.InsertOne(ctx, person)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
|
||||
}
|
||||
json.NewEncoder(w).Encode(result)
|
||||
}
|
||||
func UpdatePerson(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var person util.Person
|
||||
vars := mux.Vars(r)
|
||||
personID := vars["id"]
|
||||
person.ID, _ = primitive.ObjectIDFromHex(personID)
|
||||
json.NewDecoder(r.Body).Decode(&person)
|
||||
fmt.Println(person)
|
||||
|
||||
collection := util.Client.Database("debtlist").Collection("persons")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
filter := bson.M{"_id": person.ID}
|
||||
update := bson.M{
|
||||
"$set": bson.M{
|
||||
"name": person.Name,
|
||||
"age": person.Age,
|
||||
"email": person.Email,
|
||||
},
|
||||
}
|
||||
result, err := collection.UpdateOne(ctx, filter, update)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
|
||||
}
|
||||
json.NewEncoder(w).Encode(result)
|
||||
}
|
||||
func DeletePerson(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
vars := mux.Vars(r)
|
||||
personID := vars["id"]
|
||||
collection := util.Client.Database("debtlist").Collection("persons")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
newPersonID, err := primitive.ObjectIDFromHex(personID)
|
||||
filter := bson.M{"_id": newPersonID}
|
||||
result, err := collection.DeleteOne(ctx, filter)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(result)
|
||||
}
|
||||
func GetPerson(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
personID := vars["id"]
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var person util.Person
|
||||
collection := util.Client.Database("debtlist").Collection("persons")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
pid, err := primitive.ObjectIDFromHex(personID)
|
||||
filter := bson.M{"_id": pid}
|
||||
err = collection.FindOne(ctx, filter).Decode(&person)
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(person)
|
||||
}
|
||||
func GetAllPersons(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var persons []util.Person
|
||||
collection := util.Client.Database("debtlist").Collection("persons")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
cursor, err := collection.Find(ctx, bson.M{})
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
defer cursor.Close(ctx)
|
||||
for cursor.Next(ctx) {
|
||||
var person util.Person
|
||||
cursor.Decode(&person)
|
||||
persons = append(persons, person)
|
||||
}
|
||||
if err := cursor.Err(); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(persons)
|
||||
}
|
||||
func GetPersonsWithDebts(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
var personsWithDebts []util.PersonWithDebts
|
||||
var persons []util.Person
|
||||
collection := util.Client.Database("debtlist").Collection("persons")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
cursor, err := collection.Find(ctx, bson.M{})
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
defer cursor.Close(ctx)
|
||||
for cursor.Next(ctx) {
|
||||
var person util.Person
|
||||
cursor.Decode(&person)
|
||||
persons = append(persons, person)
|
||||
}
|
||||
if err := cursor.Err(); err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
for _, person := range persons {
|
||||
debts, err := util.GetDebtsForPerson(util.Client, person.ID.Hex())
|
||||
if err != nil {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
w.Write([]byte(`{"message": "` + err.Error() + `"}`))
|
||||
return
|
||||
}
|
||||
personsWithDebts = append(personsWithDebts, util.PersonWithDebts{Person: person, Debts: debts})
|
||||
}
|
||||
json.NewEncoder(w).Encode(personsWithDebts)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue