Merge pull request #14 from mfuentesg/fix/unique-constraint

fix: handle unique constraint for domain + ip

server/grpc/grpc.go

@@ -61,7 +61,7 @@ func (srv *Server) PutRecord(ctx context.Context, r *pb.Record) (*pb.Record, err
 	})
 
 	if err != nil {
-		logEntry.Error("unable to store data")
+		logEntry.WithField("reason", err).Error("unable to store data")
 		return nil, err
 	}
 

storage/sqlite/sqlite.go

@@ -6,7 +6,18 @@ import (
 
 	"github.com/jmoiron/sqlx"
 	"github.com/mfuentesg/localdns/storage"
-	_ "modernc.org/sqlite" // sqlite driver
+	"modernc.org/sqlite"
+)
+
+var (
+	ErrRecordAlreadyExists = errors.New("record already exists")
+)
+
+// SQLite constraint codes
+// Reference: https://www.sqlite.org/rescode.html#pve
+
+const (
+	ErrCodeConstraintUnique = 2067
 )
 
 type SQLite struct {
@@ -23,12 +34,17 @@ func New(dsn string) (*SQLite, error) {
 }
 
 func (sq *SQLite) Put(r storage.Record) (string, error) {
-	query := `insert or replace into records(
+	query := `insert into records(
     	domain, ipv4, ipv6, ttl, type
 	) values(?, ?, ?, ?, ?) returning id`
 
 	var id string
 	err := sq.db.QueryRow(query, r.Domain, r.IPv4, r.IPv6, r.TTL, r.Type).Scan(&id)
+
+	errCode := err.(*sqlite.Error).Code()
+	if errCode == ErrCodeConstraintUnique {
+		return "", ErrRecordAlreadyExists
+	}
 	return id, err
 }