gotodb.go

package main

import "fmt"
import "os"
import "net"

func main() {
    
    if len(os.Args) < 5 {
        fmt.Println("not enough arguments")
        fmt.Println("usage: go run gotodb.go <table> <file> <delimeter> <server>")
        return
    }
    
    table_name := os.Args[1]
    file_name := os.Args[2]
    delim := os.Args[3][0]
    server := os.Args[4]
    
    conn, e_conn := net.Dial("tcp", server)
    if e_conn != nil {
        fmt.Println("error connecting to database: ", e_conn.Error())
        return
    }
    defer conn.Close()

    rec := []byte("insert]" + table_name + "]")
    
    file_handle, e_file := os.Open(file_name)
    if e_file != nil {
        fmt.Println("error opening file: ", e_file.Error())
    }
    defer file_handle.Close()
    
    buffer := make([]byte, 1)
    _, e_file = file_handle.Read(buffer)
    for e_file == nil {
        if buffer[0] == delim {
            rec = append(rec, ']')
        } else if buffer[0] == '\n' {
            rec = append(rec, '\n')
            send(rec, conn)
            rec = rec[0:8+len(table_name)]
        } else {
            rec = append(rec, buffer[0])
        }
        
        _, e_file = file_handle.Read(buffer)
    }
}

func send(rec []byte, conn net.Conn) {
    
    _, e_conn := conn.Write(rec)
    if e_conn != nil {
        fmt.Println(e_conn.Error())
        return
    }
    
    response := make([]byte, 99)
    _, e_conn = conn.Read(response)
    if e_conn != nil {
        fmt.Println(e_conn.Error())
        return
    }
    
    fmt.Print(string(response))
}