Function initQueryHandler

main.go:70–216

(dbPath string, queryString string, serverPort uint)

}

func initQueryHandler(dbPath string, queryString string, serverPort uint) (func(w http.ResponseWriter, r *http.Request), error) {
	// Init db and query

	if dbPath == "" {
		return nil, fmt.Errorf("Must provide --db param")
	}
	if queryString == "" {
		return nil, fmt.Errorf("Must provide --query param")
	}
	if _, err := os.Stat(dbPath); os.IsNotExist(err) {
		return nil, fmt.Errorf("Database file '%s' doesn't exist", dbPath)
	}

	db, err := sql.Open("sqlite3", fmt.Sprintf("file:%s?mode=rw&cache=shared&_journal_mode=WAL", dbPath))
	if err != nil {
		return nil, err
	}

	db.SetMaxOpenConns(1)

	queryStmt, err := db.Prepare(queryString)
	if err != nil {
		db.Close()
		return nil, err
	}

	helpMessage := buildHelpMessage("", queryString, queryStmt, serverPort)

	return func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Server", "SQLiteQueryServer v"+version)
		w.Header().Set("Access-Control-Allow-Origin", "*")
		w.Header().Set("X-Content-Type-Options", "nosniff")

		if r.URL.Path != "/query" {
			http.Error(w, helpMessage, http.StatusNotFound)
			return
		}
		if r.Method != "POST" && r.Method != "GET" {
			http.Error(w, helpMessage, http.StatusMethodNotAllowed)
			return
		}

		// Init fullResponse
		fullResponse := []queryResult{}

		var reqCsvReader *csv.Reader
		if r.Method == "GET" {
			// Static query
			reqCsvReader = csv.NewReader(strings.NewReader(""))
		} else {
			// Parameterized query
			reqCsvReader = csv.NewReader(r.Body)
		}
		reqCsvReader.FieldsPerRecord = -1

		// Iterate over each query
		for {
			csvRecord, err := reqCsvReader.Read()