| /** |
| Licensed to the Apache Software Foundation (ASF) under one |
| or more contributor license agreements. See the NOTICE file |
| distributed with this work for additional information |
| regarding copyright ownership. The ASF licenses this file |
| to you under the Apache License, Version 2.0 (the |
| "License"); you may not use this file except in compliance |
| with the License. You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, |
| software distributed under the License is distributed on an |
| "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| KIND, either express or implied. See the License for the |
| specific language governing permissions and limitations |
| under the License. |
| **/ |
| |
| package main |
| |
| import ( |
| "database/sql" |
| "fmt" |
| _ "github.com/go-sql-driver/mysql" |
| ) |
| |
| func main() { |
| host := "127.0.0.1" |
| port := 9030 |
| user := "root" |
| password := "" |
| dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/", user, password, host, port) |
| |
| //connect to doris |
| driver, err := sql.Open("mysql", dsn) |
| if err != nil { |
| fmt.Printf("open mysql driver failed, error[%v]\n", err) |
| return |
| } |
| if err := driver.Ping(); err != nil { |
| fmt.Printf("ping doris failed, error[%v]\n", err) |
| return |
| } |
| fmt.Printf("connect to doris successfully\n") |
| |
| //create database |
| if _, err := driver.Exec("CREATE DATABASE IF NOT EXISTS db_test"); err != nil { |
| fmt.Printf("create database failed, error[%v]\n", err) |
| return |
| } |
| fmt.Printf("create database successfully\n") |
| |
| //set db context |
| if _, err := driver.Exec("USE db_test"); err != nil { |
| fmt.Printf("set db context failed, error[%v]\n", err) |
| return |
| } |
| fmt.Printf("set db context successfully\n") |
| |
| //create table |
| SQL := "CREATE TABLE IF NOT EXISTS table_test(siteid INT, citycode SMALLINT, pv BIGINT SUM) " + |
| "AGGREGATE KEY(siteid, citycode) " + |
| "DISTRIBUTED BY HASH(siteid) BUCKETS 10 " + |
| "PROPERTIES(\"replication_num\" = \"1\")" |
| if _, err := driver.Exec(SQL); err != nil { |
| fmt.Printf("create table failed, error[%v]\n", err) |
| return |
| } |
| fmt.Printf("create table successfully\n") |
| |
| //insert data |
| SQL = "INSERT INTO table_test values(1, 2, 3), (4, 5, 6), (1, 2, 4)" |
| if _, err := driver.Exec(SQL); err != nil { |
| fmt.Printf("insert data failed, error[%v]\n", err) |
| return |
| } |
| fmt.Printf("insert data successfully\n") |
| |
| //query data |
| rows, err := driver.Query("SELECT * FROM table_test") |
| if err != nil { |
| fmt.Printf("query data from doris failed, error[%v]\n", err) |
| return |
| } |
| for rows.Next() { |
| var siteId, cityCode int |
| var pv int64 |
| if err := rows.Scan(&siteId, &cityCode, &pv); err != nil { |
| fmt.Printf("scan columns failed, error[%v]\n", err) |
| return |
| } |
| fmt.Printf("%d\t%d\t%d\n", siteId, cityCode, pv) |
| } |
| fmt.Printf("query data successfully\n") |
| |
| //drop database |
| if _, err := driver.Exec("DROP DATABASE IF EXISTS db_test"); err != nil { |
| fmt.Printf("drop database failed, error[%v]\n", err) |
| return |
| } |
| fmt.Printf("drop database successfully\n") |
| } |