Migrate code from Go branch to new repo

database/postgresql/postgresql.go

@@ -0,0 +1,66 @@
+package postgresql
+
+import (
+	"database/sql"
+	"fmt"
+
+	"github.com/librespeed/speedtest/database/schema"
+
+	_ "github.com/lib/pq"
+	log "github.com/sirupsen/logrus"
+)
+
+const (
+	connectionStringTemplate = `postgres://%s:%s@%s/%s?sslmode=disable`
+)
+
+type PostgreSQL struct {
+	db *sql.DB
+}
+
+func Open(hostname, username, password, database string) *PostgreSQL {
+	connStr := fmt.Sprintf(connectionStringTemplate, username, password, hostname, database)
+	conn, err := sql.Open("postgres", connStr)
+	if err != nil {
+		log.Fatalf("Cannot open PostgreSQL database: %s", err)
+	}
+	return &PostgreSQL{db: conn}
+}
+
+func (p *PostgreSQL) Insert(data *schema.TelemetryData) error {
+	stmt := `INSERT INTO speedtest_users (ip, ispinfo, extra, ua, lang, dl, ul, ping, jitter, log, uuid) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11) RETURNING id;`
+	_, err := p.db.Exec(stmt, data.IPAddress, data.ISPInfo, data.Extra, data.UserAgent, data.Language, data.Download, data.Upload, data.Ping, data.Jitter, data.Log, data.UUID)
+	return err
+}
+
+func (p *PostgreSQL) FetchByUUID(uuid string) (*schema.TelemetryData, error) {
+	var record schema.TelemetryData
+	row := p.db.QueryRow(`SELECT * FROM speedtest_users WHERE uuid = $1`, uuid)
+	if row != nil {
+		var id string
+		if err := row.Scan(&id, &record.Timestamp, &record.IPAddress, &record.ISPInfo, &record.Extra, &record.UserAgent, &record.Language, &record.Download, &record.Upload, &record.Ping, &record.Jitter, &record.Log, &record.UUID); err != nil {
+			return nil, err
+		}
+	}
+	return &record, nil
+}
+
+func (p *PostgreSQL) FetchLast100() ([]schema.TelemetryData, error) {
+	var records []schema.TelemetryData
+	rows, err := p.db.Query(`SELECT * FROM speedtest_users ORDER BY "timestamp" DESC LIMIT 100;`)
+	if err != nil {
+		return nil, err
+	}
+	if rows != nil {
+		var id string
+
+		for rows.Next() {
+			var record schema.TelemetryData
+			if err := rows.Scan(&id, &record.Timestamp, &record.IPAddress, &record.ISPInfo, &record.Extra, &record.UserAgent, &record.Language, &record.Download, &record.Upload, &record.Ping, &record.Jitter, &record.Log, &record.UUID); err != nil {
+				return nil, err
+			}
+			records = append(records, record)
+		}
+	}
+	return records, nil
+}

database/postgresql/telemetry_postgresql.sql

@@ -0,0 +1,114 @@
+--
+-- PostgreSQL database dump
+--
+
+-- Dumped from database version 9.6.3
+-- Dumped by pg_dump version 9.6.5
+
+SET statement_timeout = 0;
+SET lock_timeout = 0;
+SET idle_in_transaction_session_timeout = 0;
+SET client_encoding = 'UTF8';
+SET standard_conforming_strings = on;
+SET check_function_bodies = false;
+SET client_min_messages = warning;
+SET row_security = off;
+
+--
+-- Name: plpgsql; Type: EXTENSION; Schema: -; Owner: 
+--
+
+CREATE EXTENSION IF NOT EXISTS plpgsql WITH SCHEMA pg_catalog;
+
+
+--
+-- Name: EXTENSION plpgsql; Type: COMMENT; Schema: -; Owner: 
+--
+
+COMMENT ON EXTENSION plpgsql IS 'PL/pgSQL procedural language';
+
+
+SET search_path = public, pg_catalog;
+
+SET default_tablespace = '';
+
+SET default_with_oids = false;
+
+--
+-- Name: speedtest_users; Type: TABLE; Schema: public; Owner: speedtest
+--
+
+CREATE TABLE speedtest_users (
+    id integer NOT NULL,
+    "timestamp" timestamp without time zone DEFAULT now() NOT NULL,
+    ip text NOT NULL,
+	ispinfo text,
+	extra text,
+    ua text NOT NULL,
+    lang text NOT NULL,
+    dl text,
+    ul text,
+    ping text,
+    jitter text,
+    log text,
+    uuid text
+);
+
+-- Commented out the following line because it assumes the user of the speedtest server, @bplower
+-- ALTER TABLE speedtest_users OWNER TO speedtest;
+
+--
+-- Name: speedtest_users_id_seq; Type: SEQUENCE; Schema: public; Owner: speedtest
+--
+
+CREATE SEQUENCE speedtest_users_id_seq
+    START WITH 1
+    INCREMENT BY 1
+    NO MINVALUE
+    NO MAXVALUE
+    CACHE 1;
+
+-- Commented out the following line because it assumes the user of the speedtest server, @bplower
+-- ALTER TABLE speedtest_users_id_seq OWNER TO speedtest; 
+
+--
+-- Name: speedtest_users_id_seq; Type: SEQUENCE OWNED BY; Schema: public; Owner: speedtest
+--
+
+ALTER SEQUENCE speedtest_users_id_seq OWNED BY speedtest_users.id;
+
+
+--
+-- Name: speedtest_users id; Type: DEFAULT; Schema: public; Owner: speedtest
+--
+
+ALTER TABLE ONLY speedtest_users ALTER COLUMN id SET DEFAULT nextval('speedtest_users_id_seq'::regclass);
+
+
+--
+-- Data for Name: speedtest_users; Type: TABLE DATA; Schema: public; Owner: speedtest
+--
+
+COPY speedtest_users (id, "timestamp", ip, ua, lang, dl, ul, ping, jitter, log, uuid) FROM stdin;
+\.
+
+
+--
+-- Name: speedtest_users_id_seq; Type: SEQUENCE SET; Schema: public; Owner: speedtest
+--
+
+SELECT pg_catalog.setval('speedtest_users_id_seq', 1, true);
+
+
+--
+-- Name: speedtest_users speedtest_users_pkey; Type: CONSTRAINT; Schema: public; Owner: speedtest
+--
+
+ALTER TABLE ONLY speedtest_users
+    ADD CONSTRAINT speedtest_users_pkey PRIMARY KEY (id);
+
+
+--
+-- PostgreSQL database dump complete
+--
+