예제 #1
0
// Wrapper for inserting through a Database struct
func BulkInsert(db *sql.DB, analytics []database.Analytic) error {
	// Base query
	insertQuery := sq.
		Insert("visits").
		Columns("time", "event", "path", "ip", "platform", "refererDomain", "countryCode")

	// Add values for each analytic object
	for _, analytic := range analytics {
		insertQuery = insertQuery.Values(
			analytic.Time.Unix(),
			analytic.Event,
			analytic.Path,
			analytic.Ip,
			analytic.Platform,
			analytic.RefererDomain,
			analytic.CountryCode)
	}

	// Add database
	insertQuery = insertQuery.RunWith(db)

	_, err := insertQuery.Exec()
	if err != nil {
		return err
	}

	return nil
}
예제 #2
0
// CreateUser converts an incoming JSON body into a User struct
func registerUser(usr *gardenswap.User) error {

	err := usr.SetPassword(usr.Password)
	if err != nil {
		return err
	}

	err = usr.RotateJWT()
	if err != nil {
		return err
	}

	query, _, err := sq.Insert(UserType).
		Columns("id", "name", "email", "jwt", "password").
		Values(usr.ID, usr.Name, usr.Email, usr.JWT, usr.PasswordHash).
		ToSql()

	if err != nil {
		return fmt.Errorf("Error creating SQL: %s", err.Error())
	}

	_, err = db.Get().Exec(query)
	if err != nil {
		return fmt.Errorf("Error inserting new user: %s", err.Error())
	}

	return nil
}
func organizationPost(w http.ResponseWriter, r *http.Request) {
	err := r.ParseForm()
	errors.CheckHttpErr(err != nil, "Internal Error", 500)
	form := r.PostForm
	// TODO ensure all params are there
	// TODO confirm admin permissions

	name := form["name"][0]
	city := form["city"][0]
	latitude := form["latitude"][0]
	longitude := form["longitude"][0]

	orgStr := []string{name, city, latitude, longitude}
	orgVals := make([]interface{}, len(orgStr))
	for i, s := range orgStr {
		orgVals[i] = s
	}

	// TODO auto assign org_id
	orgQuery, orgArgs, err := sq.Insert("organization").
		Columns([]string{"name", "city", "latitude", "longitude"}...).
		Values(orgVals...).
		ToSql()
	errors.CheckHttpErr(err != nil, "Internal Error", 500)

	_, err = db.DB.Exec(orgQuery, orgArgs...)
	if mysqlError, ok := err.(*mysql.MySQLError); ok {
		errors.CheckHttpErr(mysqlError.Number == 1062, "Organization Already Exists", 409)
	}
	errors.CheckHttpErr(err != nil, "Internal Error", 500)

	w.Write([]byte("Organization created.\n"))
}
예제 #4
0
// insert処理、last inserted idを返却する
func (self *Handle) InsertX(obj Model, values map[string]interface{}) int {

	//values["created_at"] = sq.Expr("NOW()")
	//values["updated_at"] = sq.Expr("NOW()")

	b := sq.Insert(obj.GetTable()).SetMap(values)

	s, args, err := b.ToSql()
	if err != nil {
		log.WithFields(log.Fields{
			"table":  obj.GetTable(),
			"values": values,
			"err":    err,
		}).Panic("Fail To Build Insert SQL")
	}

	res, err := self.Conn.Exec(s, args...)
	if err != nil {
		log.WithFields(log.Fields{
			"table":  obj.GetTable(),
			"values": values,
			"err":    err,
		}).Panic("Fail To Execute Insert SQL")
	}

	lastID, err := res.LastInsertId()
	if err != nil {
		log.WithFields(log.Fields{
			"table":  obj.GetTable(),
			"values": values,
			"err":    err,
		}).Panic("Fail To get Last Insert ID")
	}
	return int(lastID)
}
예제 #5
0
func userPost(w http.ResponseWriter, r *http.Request) {
	err := r.ParseForm()
	errors.CheckHttpErr(err != nil, "Internal Error", 500)
	// TODO generate ids manually instead of sql auto_increment
	// TODO role assigned by server, not request
	form := r.PostForm
	// TODO ensure all params are there
	// TODO validation of email, password, role
	// TODO confirm admin permissions
	// TODO better error handling

	hashedPassword, err := bcrypt.GenerateFromPassword([]byte(form["password"][0]), bcrypt.DefaultCost)
	errors.CheckHttpErr(err != nil, "Internal Error", 500)

	name := form["name"][0]
	email := form["email"][0]
	password := string(hashedPassword)

	usrStr := []string{name, email, password}
	usrVals := make([]interface{}, len(usrStr))
	for i, s := range usrStr {
		usrVals[i] = s
	}

	query, args, err := sq.Insert("user").
		Columns([]string{"name", "email", "password"}...).
		Values(usrVals...).
		ToSql()
	errors.CheckHttpErr(err != nil, "Internal Error", 500)

	_, err = db.DB.Exec(query, args...)
	if mysqlError, ok := err.(*mysql.MySQLError); ok {
		errors.CheckHttpErr(mysqlError.Number == 1062, "User Already Exists", 409)
	}
	errors.CheckHttpErr(err != nil, "Internal Error", 500)

	w.Write([]byte("User created.\n"))
}