Files
jedi_repo/main.go
2024-05-31 15:15:54 -06:00

154 lines
3.8 KiB
Go

package main
import (
"database/sql"
"encoding/json"
"fmt"
"log"
"net/http"
_ "github.com/lib/pq"
)
const (
host = "45.79.210.141"
port = 5432
user = "postgres"
password = "shae3Hai.dot"
dbname = "diegodb"
)
type Jedi struct {
Name string `json:"name"`
Color string `json:"color"`
Era string `json:"era"`
}
func main() {
http.HandleFunc("/create", createJedi)
http.HandleFunc("/retrieve", retrieveJedi)
http.HandleFunc("/delete", deleteJedi)
http.HandleFunc("/list", listJedi)
log.Fatal(http.ListenAndServe(":8080", nil))
}
func createJedi(w http.ResponseWriter, r *http.Request) {
var jedi Jedi
err := json.NewDecoder(r.Body).Decode(&jedi)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
db, err := sql.Open("postgres", fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname))
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer db.Close()
sqlStatement := `
INSERT INTO sables_de_luz (jedi, color, era)
VALUES ($1, $2, $3)`
_, err = db.Exec(sqlStatement, jedi.Name, jedi.Color, jedi.Era)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.WriteHeader(http.StatusCreated)
json.NewEncoder(w).Encode(jedi)
}
func retrieveJedi(w http.ResponseWriter, r *http.Request) {
name := r.URL.Query().Get("name")
db, err := sql.Open("postgres", fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname))
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer db.Close()
sqlStatement := `
SELECT jedi, color, era
FROM sables_de_luz
WHERE jedi = $1`
var jedi Jedi
row := db.QueryRow(sqlStatement, name)
err = row.Scan(&jedi.Name, &jedi.Color, &jedi.Era)
if err != nil {
if err == sql.ErrNoRows {
http.NotFound(w, r)
} else {
http.Error(w, err.Error(), http.StatusInternalServerError)
}
return
}
json.NewEncoder(w).Encode(jedi)
}
func deleteJedi(w http.ResponseWriter, r *http.Request) {
name := r.URL.Query().Get("name")
db, err := sql.Open("postgres", fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname))
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer db.Close()
sqlStatement := `
DELETE FROM sables_de_luz
WHERE jedi = $1`
_, err = db.Exec(sqlStatement, name)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.WriteHeader(http.StatusNoContent)
}
func listJedi(w http.ResponseWriter, r *http.Request) {
db, err := sql.Open("postgres", fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, password, dbname))
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer db.Close()
rows, err := db.Query("SELECT jedi, color, era FROM sables_de_luz")
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer rows.Close()
var jediList []Jedi
for rows.Next() {
var jedi Jedi
if err := rows.Scan(&jedi.Name, &jedi.Color, &jedi.Era); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
jediList = append(jediList, jedi)
}
if err := rows.Err(); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
// Construir una cadena con los nombres, colores y eras de los Jedi con saltos de línea
var response string
for _, jedi := range jediList {
response += fmt.Sprintf("%s, Color %s, Era %s\n", jedi.Name, jedi.Color, jedi.Era)
}
// Escribir la respuesta en la respuesta HTTP
w.Header().Set("Content-Type", "text/plain")
fmt.Fprintf(w, response)
}